首先,根据您发布的代码,您 没有 名称属性为“ predictions”的图层,因此此错误与您的图层
Dense图层无关
prediction:即:
prediction = Dense(1, activation='sigmoid', name='main_output')(combineFeatureLayer)
该
VGG16模型的
Dense图层为
name
predictions。特别是这一行:
x = Dense(classes, activation='softmax', name='predictions')(x)
而且由于您使用了其中两个模型,所以您的图层具有重复的名称。
您可以做的是将第二个模型中的图层重命名为除预测之外的其他名称,也许
predictions_1是这样的:
model2 = keras.applications.vgg16.VGG16(include_top=True, weights='imagenet', input_tensor=None, input_shape=None, pooling=None, classes=1000)# now change the name of the layer inplace.model2.get_layer(name='predictions').name='predictions_1'
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)