我一直在搞乱如何从保存的模型中进行转移。我的情况是resnet50_coco_best_v2.1.0.h5,加载后就像
然后我尝试将它添加到我的新模型中
resnet50Coco = models.load_model(url_model, backbone_name='resnet101')
new_resnet50Coco = resnet50Coco.get_layer('filtered_detections')
inputs = Input(shape=(256,256,3))
SalidaResnet50Coco = Model(inputs = inputs, outputs = new_resnet50Coco.output)
model = Sequential()
model.add(SalidaResnet50Coco )
model.add(Conv2D(512, 5, strides = (1), padding='same', activation='relu'))
model.add(Conv2D(1024, 5, strides = (1) , padding='same', activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2),strides=2))
model.add(Flatten())
model.add(Dense(10, activation='relu'))
model.add(Dense(8, activation='softmax'))
model.layers[0].trainable=False
model.summary()
如果您能给我提供有关如何从保存的模型创建新模型的任何帮助,我将不胜感激。