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语言实现获取主机名根据端口杀死进程
Mar 31 Python
Python数据类型详解(二)列表
May 08 Python
bat和python批量重命名文件的实现代码
May 19 Python
python中numpy.zeros(np.zeros)的使用方法
Nov 07 Python
Python 实现数据结构-堆栈和队列的操作方法
Jul 17 Python
PyTorch中Tensor的拼接与拆分的实现
Aug 18 Python
详解如何用python实现一个简单下载器的服务端和客户端
Oct 28 Python
在python中logger setlevel没有生效的解决
Feb 21 Python
python实现简单的购物程序代码实例
Mar 03 Python
Python读取Excel数据并生成图表过程解析
Jun 18 Python
python使用建议与技巧分享(二)
Aug 17 Python
pycharm部署django项目到云服务器的详细流程
Jun 29 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
PHP中header和session_start前不能有输出原因分析
2013/01/11 PHP
PHP跳转页面的几种实现方法详解
2013/06/08 PHP
浅析php变量修饰符static的使用
2013/06/28 PHP
laravel安装和配置教程
2014/10/29 PHP
使用PHP生成PDF方法详解
2015/01/23 PHP
php+ajax注册实时验证功能
2016/07/20 PHP
golang与PHP输出excel示例
2016/07/22 PHP
php strftime函数获取日期时间(switch用法)
2018/05/16 PHP
用js实现的页面关键字密度查询代码
2007/12/27 Javascript
Bootstrap每天必学之导航
2015/11/26 Javascript
BootStrap初学者对弹出框和进度条的使用感觉
2016/06/27 Javascript
Angular.js中处理页面闪烁的方法详解
2017/03/09 Javascript
Angular2 自定义validators的实现方法
2017/07/05 Javascript
React-router4路由监听的实现
2018/08/07 Javascript
解决layui 复选框等内置控件不显示的问题
2018/08/14 Javascript
node删除、复制文件或文件夹示例代码
2019/08/13 Javascript
浅析微信小程序modal弹窗关闭默认会执行cancel问题
2019/10/14 Javascript
webpack DllPlugin xxx is not defined解决办法
2019/12/13 Javascript
javascript如何使用函数random来实现课堂随机点名方法详解
2020/07/28 Javascript
在Python的Django框架中显示对象子集的方法
2015/07/21 Python
python selenium UI自动化解决验证码的4种方法
2018/01/05 Python
Python cookbook(字符串与文本)针对任意多的分隔符拆分字符串操作示例
2018/04/19 Python
Python实现求解一元二次方程的方法示例
2018/06/20 Python
解决python中使用plot画图,图不显示的问题
2018/07/04 Python
对python:threading.Thread类的使用方法详解
2019/01/31 Python
Python实现计算长方形面积(带参数函数demo)
2020/01/18 Python
css3利用transform变形结合事件完成扇形导航
2020/10/26 HTML / CSS
HTML5实现锚点时请使用id取代name
2013/09/06 HTML / CSS
Expedia马来西亚旅游网站:廉价酒店,度假村和航班预订
2016/07/26 全球购物
音乐教学随笔感言
2014/02/19 职场文书
简历中个人自我评价分享
2014/03/15 职场文书
给校长的建议书300字
2014/05/16 职场文书
技术入股合作协议书
2014/10/07 职场文书
2016年教师政治思想表现评语
2015/12/02 职场文书
Java9新特性之Module模块化编程示例演绎
2022/03/16 Java/Android
pycharm安装深度学习pytorch的d2l包失败问题解决
2022/03/25 Python