keras中加载了两个模型来测试网络,但只有第一个模型起作用。后来加载的那个不能正常工作

2024-10-08 20:24:19 发布

您现在位置:Python中文网/ 问答频道 /正文


我使用keras独立地训练了两个网络模型。在测试阶段,我需要使用这两个经过训练的模型。然而,问题来了。无论我如何改变模型的语句顺序,总是只有第一个声明的模型运行良好,而第一个模型之后的模型都不能正常工作。所有模型的图层名称都是唯一的。我的部分代码:

input_shape = (tsz[0], tsz[1], 3)
s_input_shape = (s_tsz[0], s_tsz[1], 3)        
model_full = o2o_test(input_shape)
model = Model(inputs=model_full.input, outputs=model_full.get_layer('get_vgg16_output').output)
model.load_weights('./checkpoints/weights.0076-0.3648-0.1801-0.0507-0.0498-2.0425.hdf5', by_name=True)
s_model_full = o2o_test_s(input_shape, s_input_shape)
s_model = Model(inputs=s_model_full.input, outputs=s_model_full.get_layer('get_vgg16_output_s').output)
s_model.load_weights('./checkpoints_coarse_search/weights.0025-0.1547-0.0884-0.4746.hdf5', by_name=True)

这个问题困扰了我两天。希望有人能帮我找出原因


Tags: 模型testlayerinputoutputgetmodeloutputs

热门问题