TensorFlow获取加载模型中的全部张量名称代码


Posted in Python onFebruary 11, 2020

核心代码如下:

[tensor.name for tensor in tf.get_default_graph().as_graph_def().node]

实例代码:(加载了Inceptino_v3的模型,并获取该模型所有节点的名称)

# -*- coding: utf-8 -*-
 
import tensorflow as tf
import os
 
model_dir = 'C:/Inception_v3'
model_name = 'output_graph.pb'
 
# 读取并创建一个图graph来存放训练好的 Inception_v3模型(函数)
def create_graph():
 with tf.gfile.FastGFile(os.path.join(
   model_dir, model_name), 'rb') as f:
  # 使用tf.GraphDef()定义一个空的Graph
  graph_def = tf.GraphDef()
  graph_def.ParseFromString(f.read())
  # Imports the graph from graph_def into the current default Graph.
  tf.import_graph_def(graph_def, name='')
 
# 创建graph
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')

输出结果:

mixed_8/tower/conv_1/batchnorm/moving_variance 

mixed_8/tower/conv_1/batchnorm 

r_1/mixed/conv_1/batchnorm 

.

.

.

mixed_10/tower_1/mixed/conv_1/CheckNumerics 

mixed_10/tower_1/mixed/conv_1/control_dependency 

mixed_10/tower_1/mixed/conv_1 

pool_3 

pool_3/_reshape/shape 

pool_3/_reshape 

input/BottleneckInputPlaceholder 
.
.
.
.
final_training_ops/weights/final_weights 

final_training_ops/weights/final_weights/read 

final_training_ops/biases/final_biases 

final_training_ops/biases/final_biases/read 

final_training_ops/Wx_plus_b/MatMul 

final_training_ops/Wx_plus_b/add 

final_result

由于结果太长了,就省略了一些。

如果不想这样print输出也可以将其写入txt 查看。

写入txt代码如下:

tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]
 
txt_path = './txt/节点名称'
full_path = txt_path+ '.txt'
 
for tensor_name in tensor_name_list:
 name = tensor_name + '\n'
 file = open(full_path,'a+')
file.write(name)
file.close()

以上这篇TensorFlow获取加载模型中的全部张量名称代码就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python的Flask框架中实现登录用户的个人资料和头像的教程
Apr 20 Python
Python实现基于权重的随机数2种方法
Apr 28 Python
Python实现简单的文件传输与MySQL备份的脚本分享
Jan 03 Python
Python 多线程Threading初学教程
Aug 22 Python
python timestamp和datetime之间转换详解
Dec 11 Python
Python最火、R极具潜力 2017机器学习调查报告
Dec 11 Python
python多线程下信号处理程序示例
May 31 Python
Flask框架学习笔记之使用Flask实现表单开发详解
Aug 12 Python
djano一对一、多对多、分页实例代码
Aug 16 Python
Python3视频转字符动画的实例代码
Aug 29 Python
python常见字符串处理函数与用法汇总
Oct 30 Python
pytorch 归一化与反归一化实例
Dec 31 Python
tensorflow 获取checkpoint中的变量列表实例
Feb 11 #Python
python使用正则表达式去除中文文本多余空格,保留英文之间空格方法详解
Feb 11 #Python
python 函数中的参数类型
Feb 11 #Python
python正则过滤字母、中文、数字及特殊字符方法详解
Feb 11 #Python
python3正则模块re的使用方法详解
Feb 11 #Python
Python版中国省市经纬度
Feb 11 #Python
Python终端输出彩色字符方法详解
Feb 11 #Python
You might like
通过具体程序来理解PHP里面的抽象类
2010/01/28 PHP
详解PHP序列化反序列化的方法
2015/10/27 PHP
在WordPress中实现发送http请求的相关函数解析
2015/12/29 PHP
php表单文件iframe异步上传实例讲解
2017/07/26 PHP
PHP信号处理机制的操作代码讲解
2019/04/19 PHP
PHP常见的几种攻击方式实例小结
2019/04/29 PHP
PHP实现一个限制实例化次数的类示例
2019/09/16 PHP
基于jquery的一个拖拽到指定区域内的效果
2011/09/21 Javascript
超简单JS二级、多级联动的简单实例
2014/02/18 Javascript
JavaScript实现的图像模糊算法代码分享
2014/04/22 Javascript
jQuery中index()方法用法实例
2014/12/27 Javascript
Node.js node-schedule定时任务隔多少分钟执行一次的方法
2015/02/10 Javascript
jquery实现的动态回到顶部特效代码
2015/10/28 Javascript
BootStrap点击保存后实现模态框自动关闭的思路(模态框)
2017/09/26 Javascript
微信小程序 上传头像的实例详解
2017/10/27 Javascript
vue-cli项目优化方法- 缩短首屏加载时间
2018/04/01 Javascript
vue 注册组件的使用详解
2018/05/05 Javascript
ES6 系列之 WeakMap的使用示例
2018/08/06 Javascript
浅入深出Vue之组件使用
2019/07/11 Javascript
Nuxt v-bind绑定img src不显示的解决
2019/12/05 Javascript
python数据封装json格式数据
2018/03/04 Python
python XlsxWriter模块创建aexcel表格的实例讲解
2018/05/03 Python
快速解决PyCharm无法引用matplotlib的问题
2018/05/24 Python
Pandas 按索引合并数据集的方法
2018/11/15 Python
Python面向对象之类和对象实例详解
2018/12/10 Python
Python跳出多重循环的方法示例
2019/07/03 Python
python 字典有序并写入json文件过程解析
2019/09/30 Python
Python趣味入门教程之循环语句while
2020/08/26 Python
使用BeautifulSoup4解析XML的方法小结
2020/12/07 Python
python读取excel数据并且画图的实现示例
2021/02/08 Python
CSS教程:CSS3圆角属性
2009/04/02 HTML / CSS
岗位职责范本
2013/11/23 职场文书
办公室文员工作自我评价
2013/12/01 职场文书
市场开发与营销专业求职信
2013/12/31 职场文书
二年级数学教学反思
2014/01/21 职场文书
Python爬虫入门案例之爬取二手房源数据
2021/10/16 Python