由于项目需要，需要将TensorFlow保存的模型从ckpt文件转换为pb文件。
import os
from tensorflow.python import pywrap_tensorflow
from net2use import inception_resnet_v2_small#这里使用自己定义的模型函数即可
import tensorflow as tf
if __name__=='__main__':
pb_file = "./model/output.pb"
ckpt_file = "./model/model.ckpt-652900"
'''
这里的节点名字可能
2022-11-02 15:59:21
37KB
1
（亲测可用）tensorflow训练模型进行调参，生成mobilenet_v1.pb模型文件用于预测
2021-10-19 13:41:16
16.29MB
1
网上关于tensorflow模型文件ckpt格式转pb文件的帖子很多，本人几乎尝试了所有方法，最后终于成功了，现总结如下。方法无外乎下面两种：
使用tensorflow.python.tools.freeze_graph.freeze_graph
使用graph_util.convert_variables_to_constants
1、tensorflow模型的文件解读
使用tensorflow训练好的模型会自动保存为四个文件，如下
checkpoint：记录近几次训练好的模型结果（名称）。
xxx.data-00000-of-00001: 模型的所有变量的值(weights, b
2021-09-17 11:47:47
68KB
1
把tensorflow保存的checkpoint类型模型冻结，转化为.pb模型输出。调用方法参见我的博文https://blog.csdn.net/weixin_41864878/article/details/84957681
2021-09-17 11:45:25
1KB
1