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 21 Python
Python和php通信乱码问题解决方法
Apr 15 Python
利用python程序生成word和PDF文档的方法
Feb 14 Python
Python模块结构与布局操作方法实例分析
Jul 24 Python
Python实现生成随机日期字符串的方法示例
Dec 25 Python
Pytorch中Tensor与各种图像格式的相互转化详解
Dec 26 Python
Python面向对象之继承原理与用法案例分析
Dec 31 Python
Python如何读取文件中图片格式
Jan 13 Python
python uuid生成唯一id或str的最简单案例
Jan 13 Python
Pytorch 图像变换函数集合小结
Feb 01 Python
pandas按条件筛选数据的实现
Feb 20 Python
python使用PySimpleGUI设置进度条及控件使用
Jun 10 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
神族 Protoss 历史背景
2020/03/14 星际争霸
PHP写的求多项式导数的函数代码
2012/07/04 PHP
PHP HTML JavaScript MySQL代码如何互相传值的方法分享
2012/09/30 PHP
PHP使用GIFEncoder类处理gif图片实例
2014/07/01 PHP
php检查字符串中是否包含7位GSM字符的方法
2015/03/17 PHP
laravel利用中间件防止未登录用户直接访问后台的方法
2019/09/30 PHP
js检测客户端不是firefox则提示下载
2007/04/07 Javascript
javascript getElementsByClassName 和js取地址栏参数
2010/01/02 Javascript
jQuery数组处理方法汇总
2011/06/20 Javascript
javascript实现动态侧边栏代码
2014/02/19 Javascript
js实现按钮控制图片360度翻转特效的方法
2015/02/17 Javascript
JavaScript返回当前会话cookie全部键值对照的方法
2015/04/03 Javascript
你所不了解的javascript操作DOM的细节知识点(一)
2015/06/17 Javascript
js实现基于正则表达式的轻量提示插件
2015/08/29 Javascript
JS实现登录页面记住密码和enter键登录方法推荐
2016/05/10 Javascript
JQuery控制图片由中心点逐渐放大效果
2016/06/26 Javascript
基于SpringMVC+Bootstrap+DataTables实现表格服务端分页、模糊查询
2016/10/30 Javascript
animate.css在vue项目中的使用教程
2018/08/05 Javascript
[01:06:07]2014 DOTA2国际邀请赛中国区预选赛5.21 DT VS CIS
2014/05/22 DOTA
python循环监控远程端口的方法
2015/03/14 Python
使用C语言扩展Python程序的简单入门指引
2015/04/14 Python
Django视图之ORM数据库查询操作API的实例
2017/10/27 Python
python逐行读写txt文件的实例讲解
2018/04/03 Python
Tensorflow获取张量Tensor的具体维数实例
2020/01/19 Python
利用Python如何实时检测自身内存占用
2020/05/09 Python
Python Celery异步任务队列使用方法解析
2020/08/10 Python
jupyter notebook 写代码自动补全的实现
2020/11/02 Python
HTML5语音识别标签写法附图
2013/11/18 HTML / CSS
学术会议邀请函范文
2014/01/22 职场文书
保险专业大学生职业规划书
2014/03/03 职场文书
产品质量承诺书范文
2014/03/27 职场文书
党员群众路线教育实践活动学习笔记
2014/11/05 职场文书
学生检讨书
2015/01/27 职场文书
个人收入证明格式
2015/06/24 职场文书
研究生毕业登记表的自我鉴定范文
2019/07/15 职场文书
Nginx反向代理至go-fastdfs案例讲解
2021/08/02 Servers