TensorFlow查看输入节点和输出节点名称方式
更新时间：2020年01月04日 10:12:42 作者：醉雨轩Y
今天小编就为大家分享一篇TensorFlow查看输入节点和输出节点名称方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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')
以上这篇TensorFlow查看输入节点和输出节点名称方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Scipy是基于Numpy的科学计算工具库,方便、易于使用、专为科学和工程设计,是一个用于数学、科学、工程领域的常用软件包,这篇文章主要给大家介绍了关于Python常见报错解决之SciPy和NumPy版本冲突的相关资料,需要的朋友可以参考下2024-03-03
今天小编就为大家分享一篇python多线程共享变量的使用和效率方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
这篇文章主要为大家详细介绍了python+rsync精确同步指定格式文件，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-08-08
这篇文章主要介绍了Python中异常处理的一些避坑指南，文中的示例代码讲解详细，对我们学习Python有一定帮助，感兴趣的小伙伴可以学习一下2022-03-03
这篇文章主要是想和大家来聊一聊Python如何利用自动操作 GUI 神器——PyAutoGUI在桌面实现自动化操作，感兴趣的小伙伴可以了解一下2022-06-06
这篇文章主要介绍了最新Python idle下载、安装与使用教程图文详解,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
反射就是通过字符串的形式，导入模块；通过字符串的形式，去模块寻找指定函数，并执行。利用字符串的形式去对象（模块）中操作（查找/获取/删除/添加）成员，一种基于字符串的事件驱动2022-09-09
这篇文章主要介绍了python异常处理之try finally不报错的原因，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-05-05
这篇文章主要介绍了用Python进行行为驱动开发的入门教程,本文也对BDD的概念做了详细的解释,需要的朋友可以参考下2015-04-04
这篇文章主要为大家详细介绍了python实现简易学生信息管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-07-07
最新评论