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网络编程之读取网站根目录实例
Sep 30 Python
Python中给List添加元素的4种方法分享
Nov 28 Python
python批量导入数据进Elasticsearch的实例
May 30 Python
pandas pivot_table() 按日期分多列数据的方法
Nov 16 Python
python中break、continue 、exit() 、pass终止循环的区别详解
Jul 08 Python
django基于存储在前端的token用户认证解析
Aug 06 Python
Django项目创建到启动详解(最全最详细)
Sep 07 Python
python使用 __init__初始化操作简单示例
Sep 26 Python
pyCharm 实现关闭代码检查
Jun 09 Python
Python3开发环境搭建详细教程
Jun 18 Python
pycharm2020.1.2永久破解激活教程,实测有效
Oct 29 Python
python中Tkinter 窗口之输入框和文本框的实现
Apr 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
ThinkPHP整合百度Ueditor图文教程
2014/10/21 PHP
常用PHP数组排序函数归纳
2016/08/08 PHP
自写的利用PDO对mysql数据库增删改查操作类
2018/02/19 PHP
PHP操作Redis常用命令的实例详解
2020/12/23 PHP
插件:检测javascript的内存泄漏
2007/03/04 Javascript
通过js获取div的background-image属性
2013/10/15 Javascript
javascript中的if语句使用介绍
2013/11/20 Javascript
使用原生js封装webapp滑动效果(惯性滑动、滑动回弹)
2014/05/06 Javascript
JavaScript获取路径设计源码
2014/05/22 Javascript
Visual Studio中js调试的方法图解
2014/06/30 Javascript
在jQuery中处理XML数据的大致方法
2015/08/14 Javascript
jQuery实现表格奇偶行显示不同背景色 就这么简单
2017/03/13 Javascript
angular2 ng2 @input和@output理解及示例
2017/10/10 Javascript
Vue 基于 vuedraggable 实现选中、拖拽、排序效果
2020/05/18 Javascript
Vue使用Three.js加载glTF模型的方法详解
2020/06/14 Javascript
jQuery 实现扁平式小清新导航
2020/07/07 jQuery
用Python制作简单的朴素基数估计器的教程
2015/04/01 Python
Python中使用partial改变方法默认参数实例
2015/04/28 Python
在MAC上搭建python数据分析开发环境
2016/01/26 Python
使用Python的Scrapy框架十分钟爬取美女图
2016/12/26 Python
Python中index()和seek()的用法(详解)
2017/04/27 Python
Python cookbook(数据结构与算法)从字典中提取子集的方法示例
2018/03/22 Python
Python 创建新文件时避免覆盖已有的同名文件的解决方法
2018/11/16 Python
Python读取xlsx文件的实现方法
2019/07/04 Python
CSS3实现大小不一的粒子旋转加载动画
2016/04/21 HTML / CSS
一张图片能隐含千言万语之隐藏你的程序代码
2012/12/13 HTML / CSS
英国知名的皮手套品牌:Dents
2016/11/13 全球购物
Bealls Florida百货商店:生活服饰、家居装饰和鞋子
2018/02/23 全球购物
乌克兰时尚鞋子和衣服购物网站:Born2be
2018/05/24 全球购物
提高EJB性能都有哪些技巧
2012/03/25 面试题
大学生简短的自我评价
2014/09/12 职场文书
市场部岗位职责
2015/02/12 职场文书
党员年终个人总结
2015/02/14 职场文书
2015年餐厅服务员工作总结
2015/04/23 职场文书
2016年三八节红领巾广播稿
2015/12/17 职场文书
解决Golang中ResponseWriter的一个坑
2021/04/27 Golang