tensorflow使用freeze_graph.py将ckpt转为pb文件的方法


Posted in Python onApril 22, 2020

废话少说直接上代码样例如下

import tensorflow as tf
import os
from tensorflow.python.tools import freeze_graph
# 本来这个model本无需解释太多,但是这么多人不能耐下心来看,那么我简单的说一下吧
# network是你们自己定义的模型结构而已
# ps:
# def network(input):
# return tf.layers.max_pooling2d(input, 2, 2)
from model import network


os.environ['CUDA_VISIBLE_DEVICES']='2' #设置GPU


model_path = "path to /model.ckpt-0000" #设置model的路径,因新版tensorflow会生成三个文件,只需写到数字前


def main():

 tf.reset_default_graph()

 input_node = tf.placeholder(tf.float32, shape=(228, 304, 3)) #这个是你送入网络的图片大小,如果你是其他的大小自行修改
 input_node = tf.expand_dims(input_node, 0)
 flow = network(input_node)
 flow = tf.cast(flow, tf.uint8, 'out') #设置输出类型以及输出的接口名字,为了之后的调用pb的时候使用

 saver = tf.train.Saver()
 with tf.Session() as sess:

  saver.restore(sess, model_path)

  #保存图
  tf.train.write_graph(sess.graph_def, 'output_model/pb_model', 'model.pb')
  #把图和参数结构一起
  freeze_graph.freeze_graph('output_model/pb_model/model.pb', '', False, model_path, 'out','save/restore_all', 'save/Const:0', 'output_model/pb_model/frozen_model.pb', False, "")

 print("done")

if __name__ == '__main__':
 main()

这节是关于tensorflow的Freezing,字面意思是冷冻,可理解为整合合并;整合什么呢,就是将模型文件和权重文件整合合并为一个文件,主要用途是便于发布。

官方解释可参考:https://www.tensorflow.org/extend/tool_developers/#freezing 

这里我按我的理解翻译下,不对的地方请指正:
有一点令我们为比较困惑的是,tensorflow在训练过程中,通常不会将权重数据保存的格式文件里(这里我理解是模型文件),反而是分开保存在一个叫checkpoint的检查点文件里,当初始化时,再通过模型文件里的变量Op节点来从checkoupoint文件读取数据并初始化变量。这种模型和权重数据分开保存的情况,使得发布产品时不是那么方便,所以便有了freeze_graph.py脚本文件用来将这两文件整合合并成一个文件。
freeze_graph.py是怎么做的呢?首行它先加载模型文件,再从checkpoint文件读取权重数据初始化到模型里的权重变量,再将权重变量转换成权重 常量 (因为 常量 能随模型一起保存在同一个文件里),然后再通过指定的输出节点将没用于输出推理的Op节点从图中剥离掉,再重新保存到指定的文件里(用write_graphdef或Saver)

文件目录:tensorflow/python/tools/free_graph.py
测试文件:tensorflow/python/tools/free_graph_test.py 这个测试文件很有学习价值

参数:

总共有11个参数,一个个介绍下(必选: 表示必须有值;可选: 表示可以为空):
1、input_graph:(必选)模型文件,可以是二进制的pb文件,或文本的meta文件,用input_binary来指定区分(见下面说明)
2、input_saver:(可选)Saver解析器。保存模型和权限时,Saver也可以自身序列化保存,以便在加载时应用合适的版本。主要用于版本不兼容时使用。可以为空,为空时用当前版本的Saver。
3、input_binary:(可选)配合input_graph用,为true时,input_graph为二进制,为false时,input_graph为文件。默认False
4、input_checkpoint:(必选)检查点数据文件。训练时,给Saver用于保存权重、偏置等变量值。这时用于模型恢复变量值。
5、output_node_names:(必选)输出节点的名字,有多个时用逗号分开。用于指定输出节点,将没有在输出线上的其它节点剔除。
6、restore_op_name:(可选)从模型恢复节点的名字。升级版中已弃用。默认:save/restore_all
7、filename_tensor_name:(可选)已弃用。默认:save/Const:0
8、output_graph:(必选)用来保存整合后的模型输出文件。
9、clear_devices:(可选),默认True。指定是否清除训练时节点指定的运算设备(如cpu、gpu、tpu。cpu是默认)
10、initializer_nodes:(可选)默认空。权限加载后,可通过此参数来指定需要初始化的节点,用逗号分隔多个节点名字。
11、variable_names_blacklist:(可先)默认空。变量黑名单,用于指定不用恢复值的变量,用逗号分隔多个变量名字。

用法:

例:python tensorflow/python/tools/free_graph.py \
?input_graph=some_graph_def.pb \ 注意:这里的pb文件是用tf.train.write_graph方法保存的
?input_checkpoint=model.ckpt.1001 \ 注意:这里若是r12以上的版本,只需给.data-00000….前面的文件名,如:model.ckpt.1001.data-00000-of-00001,只需写model.ckpt.1001
?output_graph=/tmp/frozen_graph.pb
?output_node_names=softmax

另外,如果模型文件是.meta格式的,也就是说用saver.Save方法和checkpoint一起生成的元模型文件,free_graph.py不适用,但可以改造下:
1、copy free_graph.py为free_graph_meta.py
2、修改free_graph.py,导入meta_graph:from tensorflow.python.framework import meta_graph
3、将91行到97行换成:input_graph_def = meta_graph.read_meta_graph_file(input_graph).graph_def

这样改即可加载meta文件

到此这篇关于tensorflow使用freeze_graph.py将ckpt转为pb文件的方法的文章就介绍到这了,更多相关tensorflow ckpt转为pb文件内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python中IPYTHON入门实例
May 11 Python
在Django的模型和公用函数中使用惰性翻译对象
Jul 27 Python
Python 3.6 性能测试框架Locust安装及使用方法(详解)
Oct 11 Python
Python自然语言处理 NLTK 库用法入门教程【经典】
Jun 26 Python
python中csv文件的若干读写方法小结
Jul 04 Python
PyCharm鼠标右键不显示Run unittest的解决方法
Nov 30 Python
python+pyqt5编写md5生成器
Mar 18 Python
使用Python实现 学生学籍管理系统
Nov 26 Python
python deque模块简单使用代码实例
Mar 12 Python
解决python多线程报错:AttributeError: Can't pickle local object问题
Apr 08 Python
Python模拟登入的N种方式(建议收藏)
May 31 Python
python读取图像矩阵文件并转换为向量实例
Jun 18 Python
tensorflow实现将ckpt转pb文件的方法
Apr 22 #Python
jupyter lab文件导出/下载方式
Apr 22 #Python
python模拟实现分发扑克牌
Apr 22 #Python
tensorflow模型文件(ckpt)转pb文件的方法(不知道输出节点名)
Apr 22 #Python
有趣的Python图片制作之如何用QQ好友头像拼接出里昂
Apr 22 #Python
python模拟斗地主发牌
Apr 22 #Python
matlab 计算灰度图像的一阶矩,二阶矩,三阶矩实例
Apr 22 #Python
You might like
PHP运行出现Notice : Use of undefined constant 的完美解决方案分享
2012/03/05 PHP
PHPThumb图片处理实例
2014/05/03 PHP
php基于curl实现随机ip地址抓取内容的方法
2016/10/11 PHP
浅谈PHP中pack、unpack的详细用法
2018/03/12 PHP
thinkphp框架类库扩展操作示例
2019/11/26 PHP
浅谈JavaScript中面向对象技术的模拟
2006/09/25 Javascript
js实现图片在未加载完成前显示加载中字样
2014/09/03 Javascript
基于jQuery通过jQuery.form.js插件实现异步上传
2015/12/13 Javascript
javascript的几种继承方法介绍
2016/03/22 Javascript
全面解析bootstrap格子布局
2016/05/22 Javascript
利用D3.js实现最简单的柱状图示例代码
2016/12/09 Javascript
JS搜狐面试题分析
2016/12/16 Javascript
javascript实现右下角广告框效果
2017/02/01 Javascript
bootstrap paginator分页插件的两种使用方式实例详解
2017/11/14 Javascript
JavaScript获取移动设备型号的实现代码(JS获取手机型号和系统)
2018/03/10 Javascript
原生实现一个react-redux的代码示例
2018/06/08 Javascript
微信小程序调用摄像头隐藏式拍照功能
2018/08/22 Javascript
layui使用数据表格实现购物车功能
2019/07/26 Javascript
微信小程序中为什么使用var that=this
2019/08/27 Javascript
vue中对象数组去重的实现
2020/02/06 Javascript
Vue中keep-alive 实现后退不刷新并保持滚动位置
2020/03/17 Javascript
浅谈nuxtjs校验登录中间件和混入(mixin)
2020/11/06 Javascript
[51:44]2018DOTA2亚洲邀请赛 4.3 突围赛 Optic vs iG 第二场
2018/04/04 DOTA
python3.6的venv模块使用详解
2018/08/01 Python
Selenium定时刷新网页的实现代码
2018/10/31 Python
python面向对象实现名片管理系统文件版
2019/04/26 Python
使用pyshp包进行shapefile文件修改的例子
2019/12/06 Python
简单了解django文件下载方式
2020/02/10 Python
解决Django提交表单报错:CSRF token missing or incorrect的问题
2020/03/13 Python
如何使用PyCharm将代码上传到GitHub上(图文详解)
2020/04/27 Python
浅谈HTML5新增和废弃的标签
2019/04/28 HTML / CSS
班组长工作职责
2013/12/25 职场文书
学习两会精神心得范文
2014/03/17 职场文书
校优秀毕业生主要事迹
2014/05/26 职场文书
大学生考试作弊检讨书
2014/09/21 职场文书
小程序实现悬浮按钮的全过程记录
2021/10/16 HTML / CSS