如何解决加载没有顶部的本地 h5 模型
所以有一种方法可以在 tensorflow 中加载内置模型，而无需 top 进行迁移学习。例如：
tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE,include_top=False,weights='imagenet')
但是如何对保存在我电脑中的 h5 模型做同样的事情，因为 tensorflow.keras.models.load_model 没有 include_top 参数？
解决方法
刚刚在 github 上找到了解决方案：
model = load_model("model.h5")
model._layers.pop(0)
所有图层都可以在
中访问all_layers = model.layers
您可以弹出最后一层并使用其余层进行预测。
model = load_model('model.h5')
model.layers.pop()
model.predict(inp)
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
