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 相关文章推荐
自动化Nginx服务器的反向代理的配置方法
Jun 28 Python
Python中内建函数的简单用法说明
May 05 Python
使用python实现生成用户信息
Mar 20 Python
利用标准库fractions模块让Python支持分数类型的方法详解
Aug 11 Python
Python实现的递归神经网络简单示例
Aug 11 Python
python 自动去除空行的实例
Jul 24 Python
Python 取numpy数组的某几行某几列方法
Oct 24 Python
Python如何实现动态数组
Nov 02 Python
Python 支持向量机分类器的实现
Jan 15 Python
python中sympy库求常微分方程的用法
Apr 28 Python
python绘制雷达图实例讲解
Jan 03 Python
Python运算符+与+=的方法实例
Feb 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 mysql数据库操作类
2008/06/04 PHP
php中ob_get_length缓冲与获取缓冲长度实例
2014/11/20 PHP
PHP中调用C/C++制作的动态链接库的教程
2016/03/10 PHP
php微信公众平台开发(四)回复功能开发
2016/12/06 PHP
thinkphp框架无限级栏目的排序功能实现方法示例
2020/03/29 PHP
Thinkphp 框架配置操作之配置加载与读取配置实例分析
2020/05/15 PHP
多个iframe自动调整大小的问题
2006/09/18 Javascript
Jquery右下角抖动、浮动 实例代码(兼容ie6、FF)
2013/08/15 Javascript
简单的js图片轮换代码(js图片轮播)
2014/05/06 Javascript
浅析JavaScript访问对象属性和方法及区别
2015/11/16 Javascript
浅谈struts1 & jquery form 文件异步上传
2017/05/25 jQuery
详解用webpack2搭建angular2的项目
2017/06/22 Javascript
React Native之ListView实现九宫格效果的示例
2017/08/02 Javascript
JS实现身份证输入框的输入效果
2017/08/21 Javascript
javascript 中模板方法单例的实现方法
2017/10/17 Javascript
利用adb shell和node.js实现抖音自动抢红包功能(推荐)
2018/02/22 Javascript
详解Webpack多环境代码打包的方法
2018/08/03 Javascript
全面解析vue router 基本使用(动态路由,嵌套路由)
2018/09/02 Javascript
4个顶级开源JavaScript图表库
2018/09/29 Javascript
Angular resolve基础用法详解
2018/10/03 Javascript
Element实现表格分页数据选择+全选所有完善批量操作
2019/06/07 Javascript
vue项目中极验验证的使用代码示例
2019/12/03 Javascript
[49:21]2018DOTA2亚洲邀请赛3月30日 小组赛B组 Effect VS iG
2018/03/31 DOTA
Python 实现引用其他.py文件中的类和类的方法
2018/04/29 Python
python树莓派红外反射传感器
2019/01/21 Python
Python 函数绘图及函数图像微分与积分
2019/11/20 Python
计算Python Numpy向量之间的欧氏距离实例
2020/05/22 Python
python安装第三方库如xlrd的方法
2020/10/31 Python
纯CSS3实现的井字棋游戏
2020/11/25 HTML / CSS
HTML5 DeviceOrientation实现手机网站摇一摇功能代码实例
2015/04/24 HTML / CSS
浅谈Html5页面打开app的一些思考
2020/03/30 HTML / CSS
春秋航空官方网站:Spring Airlines
2017/09/27 全球购物
马来西亚太阳镜、眼镜和隐形眼镜网上商店:Focus Point
2018/12/13 全球购物
美国名表在线商城:Ashford(支持中文)
2019/09/24 全球购物
争先创优公开承诺书
2014/08/30 职场文书
零基础学java之方法的定义与调用详解
2022/04/10 Java/Android