加载没有top的本地h5模型

Question

因此，有一种方法可以在tensorflow中加载内置模型，而不使用top进行迁移学习。例如：

tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE,
                                               include_top=False,
                                               weights='imagenet')

但是如何对保存在我pc中的h5模型执行相同的操作，因为tensorflow.keras.models.load_model没有include_top参数？

Answer 1

刚刚在github上找到了一个解决方案

model = load_model("model.h5")
model._layers.pop(0)

Answer 2

所有图层均可在中访问

all_layers = model.layers

您可以弹出最后一层，并使用剩余的层进行预测。

model = load_model('model.h5')
model.layers.pop()
model.predict(inp)