【问题标题】:how to convert .h5 file to .pb file?如何将 .h5 文件转换为 .pb 文件?
【发布时间】:2020-02-11 10:46:24
【问题描述】:

如何将 .h5 权重文件转换为 .pb 文件?我训练了 keras 预训练模型并将文件保存为 something.h5 。如何将其转换为 .pb 文件以用于 tensorflow 服务?

P.S:不要贬低这个问题,我在网上找不到任何解决方案。请务必提及您降低问题的原因。否则,请帮助解决问题。

【问题讨论】:

标签: python tensorflow keras tensorflow-serving


【解决方案1】:
import tensorflow as tf
model = tf.keras.models.load_model(keras_model_path)
tf.saved_model.save(model, saved_model_path)

宾果游戏。

【讨论】:

  • ImportError: DLL load failed: 找不到指定的模块。
  • 信息太少,我无法为您提供帮助。听起来你没有安装任何东西。但我不知道是什么。关于加载和保存的基础知识,我建议您参考官方文档:tensorflow.org/tutorials/keras/save_and_load。本文档涵盖 h5 和 pb 文件格式。
  • 代替tf.saved_model.save(model, saved_model_path),也可以使用model.save(saved_model_path)
  • 如果不包含扩展名,则默认为 savedmodel 格式
猜你喜欢
  • 2019-04-14
  • 2020-12-15
  • 1970-01-01
  • 1970-01-01
  • 2020-06-09
  • 2021-08-05
  • 2018-01-11
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多