分类：tensorflow模型文件

2020
09-27

tensorflow模型文件(ckpt)转pb文件的方法（不知道输出节点名）

网上关于tensorflow模型文件ckpt格式转pb文件的帖子很多，本人几乎尝试了所有方法，最后终于成功了，现总结如下。方法无外乎下面两种：使用tensorflow.python.tools.freeze_graph.freeze_graph使用graph_util.convert_variables_to_constants1、tensorflow模型的文件解读使用tensorflow训练好的模型会自动保存为四个文件，如下checkpoint：记录近几次训练好的模型结果（名称）。xxx.data-00000-of-00001:模型的所有变量的值(w... 继续阅读 >