Блог

Я возился с тем, как выполнить передачу из сохраненной модели. Мой случай — 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()
  

Если вы можете оказать мне какую-либо помощь в том, как мне создать новую модель из сохраненной, я буду признателен.