Подключите предварительно обученную модель с керасом без верха к пользовательской топ-модели

Question

Я делаю обучение с использованием керас.вот что я сделал:

• загрузка предварительно обученной модели (Mobilnet) без верха.
• сборка модели из нескольких слоев, ввод вывод mobilnet ,и вывод - softmax (классификационная миссия)
• теперь я тренирую верхнюю модель на изображениях шеи ботеля как ввод (после прохождения через mobilnet)
• наконец я хочу подключить топ-модель и мобилнет кполная модель, которая получает изображения и предсказывает классификацию.

псевдокод:

mnet=MobileNet(include_top=False,pooling='max',
weights='imagenet',input_shape=(224,224,3))
my_net = bottle_neck = Input(shape=(1024,)) 
some_layer = Dense(100 ,activation='relu')(bottle_neck)  
...
final_layer=Dense(6,activation='softmax')(prev_layers)

Моя цель - снова подключить мобилнет и мою сеть без поезда, спасибо,

Answer 1

Это описано в документации Keras, см. https://keras.io/applications/#fine-tune-inceptionv3-on-a-new-set-of-classes

. Ключевым моментом является использование вывода базовой модели в качестве входных данных вашей модели (mnet.output) и установка расширенного слоябазовая модель как trainable = False