1

我一直在搞乱如何从保存的模型中进行转移。我的情况是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()

如果您能给我提供有关如何从保存的模型创建新模型的任何帮助,我将不胜感激。

4

0 回答 0