加载没有顶部的本地 h5 模型答案

【问题标题】：Loading a local h5 model without top加载没有顶部的本地 h5 模型
【发布时间】：2021-06-04 11:02:09
【问题描述】：

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

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

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

【问题讨论】：

标签： python tensorflow machine-learning keras tensorflow2.0

【解决方案1】：

刚刚在github找到了解决方案：

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

【讨论】：

【解决方案2】：

所有层都可以在

中访问

all_layers = model.layers

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

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

【讨论】：