tensorflow将.h5模型转换成.pb

最新推荐文章于 2025-07-16 13:43:08 发布

喝粥也会胖的唐僧

最新推荐文章于 2025-07-16 13:43:08 发布

阅读量6.2k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：机器学习文章标签： tensorflow keras tensorflow-js tensorflow-lite tensorflow-serving

本文链接：https://blog.csdn.net/zhou_438/article/details/103391384

机器学习专栏收录该内容

32 篇文章

订阅专栏

本文介绍如何将Keras训练的.h5模型转换为TensorFlow的.pb通用格式，以便于使用tensorflow-js、tensorflow-lite或tensorflow-serving进行部署。通过Python代码演示了设置学习阶段、加载模型、定义输出路径及使用tf.saved_model简单保存模型的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

keras使用起来十分简洁，我们经常会拿来训练数据，然后保存为模型.h5，然而对于tensorflow来说，.pb才是它的通用格式，因为tensorflow-js，tensorflow-lite,tensorflow-serving都会使用到，因此转换是具有很强的实际意义的。

import tensorflow as tf

tf.keras.backend.set_learning_phase(0) 
model = tf.keras.models.load_model('./mmm.h5')#加载h5模型
export_path = './pb_model'

with tf.keras.backend.get_session() as sess:
    tf.saved_model.simple_save(
        sess,
        export_path,
        inputs={'input_image': model.input},
        outputs={t.name: t for t in model.outputs})

测试一下：

D:\AN\lib\site-packages\numpy\core\__init__.py:29: UserWarning: loaded more than 1 DLL from .libs:
D:\AN\lib\site-packages\numpy\.libs\libopenblas.PYQHXLVVQ7VESDPUVUADXEVJOBGHJPAY.gfortran-win_amd64.dll
D:\AN\lib\site-packages\numpy\.libs\libopenblas.TXA6YQSD3GCQQC22GEQ54J2UDCXDXHWN.gfortran-win_amd64.dll
  stacklevel=1)

运行后悔产生一个文件夹，里面就是我们需要的.pb啦