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中使用urllib2获取http请求状态码的代码例子
Jul 07 Python
教你用Python写安卓游戏外挂
Jan 11 Python
python生成圆形图片的方法
Mar 25 Python
分析python动态规划的递归、非递归实现
Mar 04 Python
python自动化报告的输出用例详解
May 30 Python
python3中函数参数的四种简单用法
Jul 09 Python
对python生成业务报表的实例详解
Feb 03 Python
Python变量类型知识点总结
Feb 18 Python
python 微信好友特征数据分析及可视化
Jan 07 Python
jupyter note 实现将数据保存为word
Apr 14 Python
numpy的Fancy Indexing和array比较详解
Jun 11 Python
Python趣味挑战之教你用pygame画进度条
May 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
?繁体转换的class
2006/10/09 PHP
PHP控制网页过期时间的代码
2008/09/28 PHP
在php中判断一个请求是ajax请求还是普通请求的方法
2011/06/28 PHP
php绘图中显示不出图片的原因及解决
2014/03/05 PHP
教你识别简单的免查杀PHP后门
2015/09/13 PHP
PHP 接入支付宝即时到账功能
2016/09/18 PHP
PHP编写daemon process详解及实例代码
2016/09/30 PHP
PHP Mysqli 常用代码集合
2016/11/12 PHP
PHP实现的装箱算法示例
2018/06/23 PHP
疯掉了,尽然有js写的操作系统
2007/04/23 Javascript
javascript 建设银行登陆键盘
2008/06/10 Javascript
Jquery+WebService 校验账号是否已被注册的代码
2010/07/12 Javascript
40个有创意的jQuery图片和内容滑动及弹出插件收藏集之三
2012/01/03 Javascript
html组件不可输入(只读)同时任何组件都有效
2013/04/01 Javascript
Jquery实现侧边栏跟随滚动条固定(兼容IE6)
2014/04/02 Javascript
jQuery中复合属性选择器用法实例
2014/12/31 Javascript
js获取字符串字节数方法小结
2015/06/09 Javascript
jquery实现简单文字提示效果
2015/12/02 Javascript
AngularJS学习第一篇 AngularJS基础知识
2017/02/13 Javascript
js实现功能比较全面的全选和多选
2017/03/02 Javascript
Python简单实现enum功能的方法
2016/04/25 Python
Python 多线程实例详解
2017/03/25 Python
完美解决在oj中Python的循环输入问题
2018/06/25 Python
python实战教程之自动扫雷
2018/07/13 Python
python实现彩票系统
2020/06/28 Python
linux中如何使用python3获取ip地址
2019/07/15 Python
详解python实现小波变换的一个简单例子
2019/07/18 Python
pytorch 在网络中添加可训练参数,修改预训练权重文件的方法
2019/08/17 Python
Python 实现自动完成A4标签排版打印功能
2020/04/09 Python
pycharm 快速解决python代码冲突的问题
2021/01/15 Python
TripAdvisor越南:全球领先的旅游网站
2017/09/21 全球购物
经贸韩语专业大学生职业规划
2014/02/14 职场文书
丧事主持词大全
2014/04/02 职场文书
企业安全生产责任书
2014/04/14 职场文书
机关门卫的岗位职责
2014/04/29 职场文书
单位法定代表人授权委托书
2014/09/20 职场文书