Как сохранить создать и сохранить подмодель из мобильной сети keras?

Question

Я создал модель mobilenet v2 в керасе вместе с лямбда-функцией, которая выводит меня из предпоследнего слоя.Как я могу сохранить эту конкретную подмодель (я хочу преобразовать ее с помощью tfjs-преобразователя и использовать в tenorflow-js)?

from keras.applications.mobilenet_v2 import MobileNetV2
model = MobileNetV2(input_shape=None, alpha=0.35, depth_multiplier=1, include_top=True, weights='imagenet', input_tensor=None, pooling=None, classes=1000)
get_3rd_layer_output = K.function([model.layers[0].input],
                                      [model.layers[-2].output])

Answer 1

Вы можете создать модель Keras вместо функции Keras и сохранить ее с помощью метода model.save(filepath):

custom_model = keras.models.Model(inputs=model.layers[0].input, 
                                  outputs=model.layers[-2].output)

custom_model.save('model.h5')