tensorflow ckpt模型和pb模型获取节点名称,及ckpt转pb模型实例


Posted in Python onJanuary 21, 2020

ckpt

from tensorflow.python import pywrap_tensorflow 
checkpoint_path = 'model.ckpt-8000' 
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) 
var_to_shape_map = reader.get_variable_to_shape_map() 
for key in var_to_shape_map: 
 print("tensor_name: ", key)

pb

import tensorflow as tf
import os

model_name = './mobilenet_v2_140_inf_graph.pb'

def create_graph():
 with tf.gfile.FastGFile(model_name, 'rb') as f:
  graph_def = tf.GraphDef()
  graph_def.ParseFromString(f.read())
  tf.import_graph_def(graph_def, name='')

create_graph()
tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]
for tensor_name in tensor_name_list:
 print(tensor_name,'\n')

ckpt转pb

def freeze_graph(input_checkpoint,output_graph):
 '''
 :param input_checkpoint:
 :param output_graph: PB模型保存路径
 :return:
 '''
 output_node_names = "xxx"
 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())

以上这篇tensorflow ckpt模型和pb模型获取节点名称,及ckpt转pb模型实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python分割文件的常用方法
Nov 01 Python
python如何实现excel数据添加到mongodb
Jul 30 Python
利用Hyperic调用Python实现进程守护
Jan 02 Python
python+matplotlib绘制3D条形图实例代码
Jan 17 Python
python如何发布自已pip项目的方法步骤
Oct 09 Python
python 实现将txt文件多行合并为一行并将中间的空格去掉方法
Dec 20 Python
python 含子图的gif生成时内存溢出的方法
Jul 07 Python
Python+OpenCv制作证件图片生成器的操作方法
Aug 21 Python
Python合并2个字典成1个新字典的方法(9种)
Dec 19 Python
pycharm无法导入本地模块的解决方式
Feb 12 Python
Pyspark获取并处理RDD数据代码实例
Mar 27 Python
pytorch 一行代码查看网络参数总量的实现
May 12 Python
tensorflow查看ckpt各节点名称实例
Jan 21 #Python
python同义词替换的实现(jieba分词)
Jan 21 #Python
tensorflow模型保存、加载之变量重命名实例
Jan 21 #Python
tensorflow实现测试时读取任意指定的check point的网络参数
Jan 21 #Python
tensorflow如何继续训练之前保存的模型实例
Jan 21 #Python
在tensorflow中设置保存checkpoint的最大数量实例
Jan 21 #Python
TensorFlow——Checkpoint为模型添加检查点的实例
Jan 21 #Python
You might like
锁定年轻人的双倍活力 星巴克推出星倍醇即饮浓咖啡
2021/03/03 咖啡文化
CodeIgniter删除和设置Cookie的方法
2015/04/07 PHP
PHP的消息通信机制测试实例
2016/11/10 PHP
JQuery 绑定select标签的onchange事件,弹出选择的值,并实现跳转、传参
2011/01/06 Javascript
使用js检测浏览器的实现代码
2013/05/14 Javascript
图片Slider 带左右按钮的js示例
2013/08/30 Javascript
jquery text(),val(),html()方法区别总结
2013/11/04 Javascript
一个支付页面DEMO附截图
2014/07/22 Javascript
js实现文本框只允许输入数字并限制数字大小的方法
2015/08/19 Javascript
JS基于Ajax实现的网页Loading效果代码
2015/10/27 Javascript
jQuery中选择器的基础使用教程
2016/05/23 Javascript
深入nodejs中流(stream)的理解
2017/03/27 NodeJs
jquery平滑滚动到顶部插件使用详解
2017/05/08 jQuery
vue的token刷新处理的方法
2018/07/17 Javascript
微信小程序自定义对话框弹出和隐藏动画
2018/07/19 Javascript
[02:44]重置世界,颠覆未来——DOTA2 7.23版本震撼上线
2019/12/01 DOTA
在Python中操作时间之tzset()方法的使用教程
2015/05/22 Python
python搭建服务器实现两个Android客户端间收发消息
2018/04/12 Python
python实现kmp算法的实例代码
2019/04/03 Python
nginx黑名单和django限速,最简单的防恶意请求方法分享
2019/08/09 Python
Tensorflow中的降维函数tf.reduce_*使用总结
2020/04/20 Python
基于python生成英文版词云图代码实例
2020/05/16 Python
Numpy 多维数据数组的实现
2020/06/18 Python
基于Python中Remove函数的用法讨论
2020/12/11 Python
美国大码时尚女装购物网站:ELOQUII
2017/12/28 全球购物
文秘专业毕业生就业推荐信
2013/11/08 职场文书
护士辞职信模板
2014/01/20 职场文书
ktv中秋节活动方案
2014/01/30 职场文书
师德师风演讲稿
2014/05/05 职场文书
大学毕业生求职自荐书
2014/06/05 职场文书
老人再婚离婚协议书范本
2014/10/27 职场文书
2015年小学生暑假总结
2015/07/13 职场文书
用Python创建简易网站图文教程
2021/06/11 Python
详解Python+OpenCV绘制灰度直方图
2022/03/22 Python
升级 Win11 还是坚守 Win10?微软 Win11 新系统缺失功能大盘点
2022/04/05 数码科技
Nginx如何获取自定义请求header头和URL参数详解
2022/07/23 Servers