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


Posted in Python onApril 22, 2020

网上关于tensorflow模型文件ckpt格式转pb文件的帖子很多,本人几乎尝试了所有方法,最后终于成功了,现总结如下。方法无外乎下面两种:

  • 使用tensorflow.python.tools.freeze_graph.freeze_graph
  • 使用graph_util.convert_variables_to_constants

1、tensorflow模型的文件解读

使用tensorflow训练好的模型会自动保存为四个文件,如下

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

checkpoint:记录近几次训练好的模型结果(名称)。

xxx.data-00000-of-00001: 模型的所有变量的值(weights, biases, placeholders,gradients, hyper-parameters etc),也就是模型训练好参数和其他值。

xxx.index :模型的元数据,二进制或者其他格式,不可直接查看 。是一个不可变得字符串表,每一个键都是张量的名称,它的值是一个序列化的BundleEntryProto。 每个BundleEntryProto描述张量的元数据:“数据”文件中的哪个文件包含张量的内容,该文件的偏移量,校验和一些辅助数据等。

xxx.meta:模型的meta数据 ,二进制或者其他格式,不可直接查看,保存了TensorFlow计算图的结构信息,通俗地讲就是神经网络的网络结构。

2、最常见的ckpt转pb文件的方法

2、ckpt转pb文件(freeze_graph.freeze_graph)

此种方法尝试成功,虽然不知道输出节点名,但是只要模型代码还在就可以操作,直接上代码。

import tensorflow as tf
import os
from tensorflow.python.tools import freeze_graph
from model import network # network是你们自己定义的模型结构(代码结构)
# egs:
# def network(input):
# return tf.layers.softmax(input)
 
model_path = "model.ckpt-0000" #设置model的路径,因新版tensorflow会生成三个文件,只需写到数字前
 
def main():
 tf.reset_default_graph()
 # 设置输入网络的数据维度,根据训练时的模型输入数据的维度自行修改
 input_node = tf.placeholder(tf.float32, shape=(None, None, 200)) 
 output_node = network(input_node) # 神经网络的输出
 # 设置输出数据类型(特别注意,这里必须要跟输出网络参数的数据格式保持一致,不然会导致模型预测  精度或者预测能力的丢失)以及重新定义输出节点的名字(这样在后面保存pb文件以及之后使用pb文件时直接使用重新定义的节点名字即可)
 flow = tf.cast(output_node , tf.float16, 'the_outputs') 
 saver = tf.train.Saver()
 with tf.Session() as sess:
 saver.restore(sess, model_path)
 #保存模型图(结构),为一个json文件
 tf.train.write_graph(sess.graph_def, 'output_model/pb_model', 'model.pb')
 #将模型参数与模型图结合,并保存为pb文件
 freeze_graph.freeze_graph('output_model/pb_model/model.pb', '', False, model_path, 'the_outputs','save/restore_all', 'save/Const:0', 'output_model/pb_model/frozen_model.pb', False, "")
 print("done")
if __name__ == '__main__':
 main()

2、ckpt转pb文件(graph_util.convert_variables_to_constants)

没有成功,因为不知道输出节点的名字,使用该方法保存后的pb文件只有几十k,无法使用,写在这里主要是为了总结。直接上代码,代码里面没有的库(函数),按提示自行import。

def freeze_graph(input_checkpoint,output_graph):
 '''
 :param input_checkpoint:
 :param output_graph: PB模型保存路径
 :return:
 '''
 # checkpoint = tf.train.get_checkpoint_state(model_folder) #检查目录下ckpt文件状态是否可用
 # input_checkpoint = checkpoint.model_checkpoint_path #得ckpt文件路径
 
 # 指定输出的节点名称,该节点名称必须是原模型中存在的节点
 output_node_names = "InceptionV3/Logits/SpatialSqueeze"
 saver = tf.train.import_meta_graph(input_checkpoint + '.meta', clear_devices=True)
 graph = tf.get_default_graph() # 获得默认的图
 input_graph_def = graph.as_graph_def() # 返回一个序列化的图代表当前的图
 
 with tf.Session() as sess:
 saver.restore(sess, input_checkpoint) #恢复图并得到数据
 output_graph_def = graph_util.convert_variables_to_constants( # 模型持久化,将变量值固定
  sess=sess,
  input_graph_def=input_graph_def,# 等于:sess.graph_def
  output_node_names=output_node_names.split(","))# 如果有多个输出节点,以逗号隔开
 
 with tf.gfile.GFile(output_graph, "wb") as f: #保存模型
  f.write(output_graph_def.SerializeToString()) #序列化输出
 print("%d ops in the final graph." % len(output_graph_def.node)) #得到当前图有几个操作节点
 
 # for op in graph.get_operations():
 # print(op.name, op.values())
 
if __name__ == '__main__':
 # 输入ckpt模型路径
 input_checkpoint='models/model.ckpt-10000'
 # 输出pb模型的路径
 out_pb_path="models/pb/frozen_model.pb"
 # 调用freeze_graph将ckpt转为pb
 freeze_graph(input_checkpoint,out_pb_path)

参考链接:

到此这篇关于tensorflow模型文件(ckpt)转pb文件(不知道输出节点名)的文章就介绍到这了,更多相关tensorflow ckpt转pb文件内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
提升Python程序运行效率的6个方法
Mar 31 Python
详解Python中的array数组模块相关使用
Jul 05 Python
HTML中使用python屏蔽一些基本功能的方法
Jul 07 Python
python定向爬取淘宝商品价格
Feb 27 Python
对tensorflow 的模型保存和调用实例讲解
Jul 28 Python
对python GUI实现完美进度条的示例详解
Dec 13 Python
Python3.7 dataclass使用指南小结
Feb 22 Python
python实现合并多个list及合并多个django QuerySet的方法示例
Jun 11 Python
在python中实现同行输入/接收多个数据的示例
Jul 20 Python
python openCV获取人脸部分并存储功能
Aug 28 Python
python 在sql语句中使用%s,%d,%f说明
Jun 06 Python
python 实现单例模式的5种方法
Sep 23 Python
有趣的Python图片制作之如何用QQ好友头像拼接出里昂
Apr 22 #Python
python模拟斗地主发牌
Apr 22 #Python
matlab 计算灰度图像的一阶矩,二阶矩,三阶矩实例
Apr 22 #Python
python根据完整路径获得盘名/路径名/文件名/文件扩展名的方法
Apr 22 #Python
matlab中二维插值函数interp2的使用详解
Apr 22 #Python
python 一维二维插值实例
Apr 22 #Python
Numpy一维线性插值函数的用法
Apr 22 #Python
You might like
用php实现让页面只能被百度gogole蜘蛛访问的方法
2009/12/29 PHP
关于laravel 子查询 & join的使用
2019/10/16 PHP
使用自定义setTimeout和setInterval使之可以传递参数和对象参数
2009/04/24 Javascript
从阶乘函数对比Javascript和C#的异同
2012/05/31 Javascript
解决js正则匹配换行问题实现代码
2012/12/10 Javascript
JavaScript获取多个数组的交集简单实例
2013/11/11 Javascript
js 获取元素下面所有li的两种方法
2014/04/14 Javascript
jQuery.holdReady()方法用法实例
2014/12/27 Javascript
jQuery使用$.ajax进行即时验证实例详解
2015/12/11 Javascript
jQuery图片轮播插件——前端开发必看
2016/05/31 Javascript
AngularJS延迟加载html template
2016/07/27 Javascript
深入学习 JavaScript中的函数调用
2017/03/23 Javascript
jQuery手风琴的简单制作
2017/05/12 jQuery
原生js获取left值和top值的三种方法
2017/08/02 Javascript
JS实现的简单标签点击切换功能示例
2017/09/21 Javascript
深入理解ES6之数据解构的用法
2018/01/13 Javascript
VUE + UEditor 单图片跨域上传功能的实现方法
2018/02/08 Javascript
使用vue-infinite-scroll实现无限滚动效果
2018/06/22 Javascript
Vue 之孙组件向爷组件通信的实现
2019/04/23 Javascript
在Vue项目中,防止页面被缩放和放大示例
2019/10/28 Javascript
Node.js API详解之 net模块实例分析
2020/05/18 Javascript
[07:37]DOTA2-DPC中国联赛2月2日Recap集锦
2021/03/11 DOTA
解决Python传递中文参数的问题
2015/08/04 Python
bpython 功能强大的Python shell
2016/02/16 Python
新西兰杂志订阅:isubscribe
2019/08/26 全球购物
娇韵诗俄罗斯官方网站:Clarins俄罗斯
2020/10/03 全球购物
用C#语言写出与SQLSERVER访问时的具体过程
2013/04/16 面试题
安全资料员岗位职责
2013/12/14 职场文书
工作疏忽、懈怠的检讨书
2014/09/11 职场文书
2015高三毕业寄语赠言
2015/02/27 职场文书
法院个人总结
2015/03/03 职场文书
2015年小学数学教研组工作总结
2015/05/21 职场文书
被告代理词范文
2015/05/25 职场文书
2016年万圣节活动个人总结
2016/04/05 职场文书
2019年教师节:送给所有老师的祝福语
2019/09/05 职场文书
无线电知识基础入门篇
2022/02/18 无线电