分类：ckpt转pb文件

2020
09-27

tensorflow实现将ckpt转pb文件的方法

   本博客实现将自己训练保存的ckpt模型转换为pb文件，该方法适用于任何ckpt模型，当然你需要确定ckpt模型输入/输出的节点名称。   使用tf.train.saver()保存模型时会产生多个文件，会把计算图的结构和图上参数取值分成了不同的文件存储。这种方法是在TensorFlow中是最常用的保存方式。    例如：下面的代码运行后，会在save目录下保存了四个文件：importtensorflowastf#... 继续阅读 >

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... 继续阅读 >