TensorFlow查看输入节点和输出节点名称方式
作者:醉雨轩Y 时间:2022-11-27 04:50:11
TensorFlow 定义输入节点名称input_name:
with tf.name_scope('input'):
bottleneck_input = tf.placeholder_with_default(
bottleneck_tensor,
shape=[batch_size, bottleneck_tensor_size],
name='Mul')
TensorFlow查看pb数据库里面的输入节点和输出节点:
import tensorflow as tf
import os
model_dir = './tmp/'
model_name = 'output_graph.pb'
def create_graph():
with tf.gfile.FastGFile(os.path.join(
model_dir, 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')
来源:https://blog.csdn.net/miao0967020148/article/details/89434253
标签:TensorFlow,输入,节点,输出


猜你喜欢
JS 调试中常见的报错问题解决方法
2023-07-16 07:42:13
关于Go 是传值还是传引用?
2024-04-26 17:26:38

python 管理系统实现mysql交互的示例代码
2024-01-28 16:30:52

Python中字符串List按照长度排序
2023-11-28 21:43:02
瀑布流布局浅析
2011-09-16 20:18:09
SQL Server表中添加新列并添加描述
2024-01-23 08:10:17

Python的Django应用程序解决AJAX跨域访问问题的方法
2023-08-07 16:00:24
Python3基于sax解析xml操作示例
2022-06-07 10:18:52

python 类相关概念理解
2023-02-17 21:16:47
css基础教程属性篇之二
2008-07-25 19:23:00

如何实现一个简易版的vuex持久化工具
2024-04-30 10:34:29
针对SQL Server中业务规则链接的分析
2009-01-20 11:43:00
聚合函数和group by的关系详解
2024-01-20 15:19:53

Python 在OpenCV里实现仿射变换—坐标变换效果
2022-06-19 15:13:40

Spring Boot Mysql 数据库操作示例
2024-01-19 10:22:28

debug case: onsubmit=return false依然能提交
2009-01-18 13:07:00
js获取地址栏中传递的参数(两种方法)
2024-04-17 09:54:30
Windows服务器MySQL中文乱码的解决方法
2024-01-12 16:46:51

Jupyter安装链接aconda实现过程图解
2022-02-04 21:33:57

MYSQL慢查询和日志实例讲解
2024-01-26 12:20:35