Tensorflow 如何从keras mobilenet保存创建和保存子模型？

我已经在keras中创建了mobilenet v2模型，以及lambda函数，该函数提供倒数第二层的输出。我如何保存这个特定的子模型（我想用tfjs转换器转换它，并在tensorflow js中使用它）

---

您可以创建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')

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

custom_model.save('model.h5')