正在加载的整体keras模型给出ValueError:无效的input_shape参数(None,224、224、3):模型具有0个张量输入

问题描述 投票:1回答:1

我的模型是2个不同的keras模型的集合,这些模型连接到相同的输入层,合并后具有2个输出层。两种模型都是经过预训练的,我正在尝试创建并行架构。我的架构是:`

model_input = Input(shape=(224,224,3), name="model_input")
gender_model_copy.layers.pop(0)
color_model_copy.layers.pop(0)
color_model_ens1 = color_model_copy(model_input)
gender_model_ens1 = gender_model_copy(model_input)
model_f = Model(input=[model_input], output=[color_model_ens1,gender_model_ens1])
model_f.save('path')

`

模型被编译,我也可以做出预测,但是当我保存并尝试重新加载它时,我得到了:

ValueError: Invalid input_shape argument (None, 224, 224, 3): model has 0 tensor inputs.完整跟踪:Github gist link

我有一个自定义图层,正在使用custom_objects={'Scale':Scale()}添加keras.models.load_model中的参数我的keras版本是2.2.5,而tensorflow版本是1.15

EDIT:我意识到问题是我使图层无法通过layer.trainable=False进行训练,而没有这样做就能够加载没有错误的模型。我仍然想知道为什么会这样。

python tensorflow keras keras-layer keras-2
1个回答
0
投票

@ SajinGohil,请问我遇到类似错误的代码示例解释了您的解决方案

© www.soinside.com 2019 - 2024. All rights reserved.