Удалите последний слой и вставьте три слоя Conv2D в Keras

Question

У меня есть модель в Керасе для классификации, которую я обучил на некотором наборе данных.Назовите эту модель "ification_model ".Эта модель сохранена в "ification.h5 ".Модель для обнаружения та же, за исключением того, что мы удаляем последний сверточный слой и добавляем три Conv2D слоя размером (3,3).Итак, наша модель для обнаружения «response_model» должна выглядеть следующим образом:

detection_model =ification_model [: last_conv_index] + Conv2d + Conv2d + Conv2d.

Как мы можем реализовать это в Керасе?

Answer 1

Ну, загрузите свою классификационную модель и используйте Функциональный API Keras , чтобы создать новую модель:

model = load_model("classification.h5")

last_conv_layer_output = model.layers[last_conv_index].output
conv = Conv2D(...)(last_conv_layer_output)
conv = Conv2D(...)(conv)
output = Conv2D(...)(conv)

new_model = Model(model.inputs, output)

# compile the new model and save it ...