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之禅》中对于Python编程过程中的一些建议
Apr 03 Python
Python实现以时间换空间的缓存替换算法
Feb 19 Python
浅谈用VSCode写python的正确姿势
Dec 16 Python
python PyTorch参数初始化和Finetune
Feb 11 Python
Python 中的Selenium异常处理实例代码
May 03 Python
python利用smtplib实现QQ邮箱发送邮件
May 20 Python
Python3匿名函数用法示例
Jul 25 Python
Python正则表达式实现简易计算器功能示例
May 07 Python
详解PyCharm安装MicroPython插件的教程
Jun 24 Python
Python3 pywin32模块安装的详细步骤
May 26 Python
python中_del_还原数据的方法
Dec 09 Python
Pytest allure 命令行参数的使用
Apr 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
全国FM电台频率大全 - 20 广西省
2020/03/11 无线电
PHP表单递交控件名称含有点号(.)会被转化为下划线(_)的处理方法
2013/01/06 PHP
PHP中的socket_read和socket_recv区别详解
2015/02/09 PHP
CI映射(加载)数据到view层的方法
2016/03/28 PHP
一段实用的php验证码函数
2016/05/19 PHP
PHP+Ajax实现验证码的实时验证
2016/07/20 PHP
PHP环形链表实现方法示例
2017/09/15 PHP
Laravel框架查询构造器简单示例
2019/05/08 PHP
js中判断Object、Array、Function等引用类型对象是否相等
2012/08/29 Javascript
js页面跳转的常用方法整理
2013/10/18 Javascript
JavaScript插件化开发教程 (一)
2015/01/27 Javascript
js实现模拟银行卡账号输入显示效果
2015/11/18 Javascript
全面解析Bootstrap表单使用方法(表单控件)
2015/11/24 Javascript
js将json格式的对象拼接成复杂的url参数方法
2016/05/25 Javascript
JS针对Array的各种操作汇总
2016/11/29 Javascript
简单实现js菜单栏切换效果
2017/03/04 Javascript
微信小程序开发之从相册获取图片 使用相机拍照 本地图片上传
2017/04/18 Javascript
Vue 父子组件的数据传递、修改和更新方法
2018/03/01 Javascript
vue实现tab切换外加样式切换方法
2018/03/16 Javascript
微信小程序通过保存图片分享到朋友圈功能
2018/05/24 Javascript
基于Vue 2.0 监听文本框内容变化及ref的使用说明介绍
2018/08/24 Javascript
Nuxt使用Vuex的方法示例
2019/09/06 Javascript
Python3基础之函数用法
2014/08/13 Python
pygame学习笔记(5):游戏精灵
2015/04/15 Python
在CentOS上配置Nginx+Gunicorn+Python+Flask环境的教程
2016/06/07 Python
python爬取NUS-WIDE数据库图片
2016/10/05 Python
python thrift搭建服务端和客户端测试程序
2018/01/17 Python
Python3实现配置文件差异对比脚本
2019/11/18 Python
python3中pip3安装出错,找不到SSL的解决方式
2019/12/12 Python
Python调用ffmpeg开源视频处理库,批量处理视频
2020/11/16 Python
如何用Matlab和Python读取Netcdf文件
2021/02/19 Python
使用CSS3创建动态菜单效果
2015/07/10 HTML / CSS
HTML5样式控制示例代码
2013/11/27 HTML / CSS
意大利比基尼品牌:MISS BIKINI
2019/11/02 全球购物
python爬不同图片分别保存在不同文件夹中的实现
2021/04/02 Python
使用pandas模块实现数据的标准化操作
2021/05/14 Python