ckpt 转 pb

原创已于 2022-03-04 09:38:31 修改 · 1.5k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#p2p #linq #c#

于 2022-03-04 08:48:49 首次发布

部署专栏收录该内容

5 篇文章

订阅专栏

该博客介绍了如何将TensorFlow 1.x的模型权重转换为.pb格式，包括使用1.py获取输出节点，编辑.con.....pb.py文件并运行以完成转换。同时警告使用TF2.0可能会导致错误，并提供了一个简书链接作为参考。

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

链接：https://pan.baidu.com/s/1001xctKvVC4RRke3wKreUg
提取码：8y8o

正常保存的ckpt 有5个文件

第三个是权重（其他的去网上查下）

这是我的环境

absl-py (1.0.0)
astor (0.8.1)
certifi (2016.2.28)
gast (0.5.3)
grpcio (1.12.1)
h5py (2.7.0)
Keras-Applications (1.0.8)
Keras-Preprocessing (1.1.2)
Markdown (2.6.9)
numpy (1.13.3)
pip (21.3.1)
protobuf (3.7.1)
scipy (0.19.1)
setuptools (36.4.0)
six (1.10.0)
tensorboard (1.12.0)
tensorflow (1.12.0)
termcolor (1.1.0)
Werkzeug (0.12.2)
wheel (0.29.0)
wincertstore (0.2)

安装环境TXT pip install packages.txt