当前位置: 首页 > 编程笔记 >

TensorFlow查看输入节点和输出节点名称方式

祁建业
2023-03-14
本文向大家介绍TensorFlow查看输入节点和输出节点名称方式,包括了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查看输入节点和输出节点名称方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持小牛知识库。

 类似资料:
  • 我用Tensorflow培训了一个模型(更快的rcnn\U resnet101\U coco\U 2018\U 01\U 28)。我有一个“.pb”图形。 要制作一个冻结图,我需要输入和输出节点。 如何在图表中找到它? 我这里有完整的节点列表。 没有任何节点像Softmax,占位符,因为它在其他帖子中建议。

  • 4.4 ROS节点名称重名 场景:ROS 中创建的节点是有名称的,C++初始化节点时通过API:ros::init(argc,argv,"xxxx");来定义节点名称,在Python中初始化节点则通过 rospy.init_node("yyyy") 来定义节点名称。在ROS的网络拓扑中,是不可以出现重名的节点的,因为假设可以重名存在,那么调用时会产生混淆,这也就意味着,不可以启动重名节点或者同一个

  • 新的节点红色,只是第一次尝试这个节点。 我希望将时间戳用作文件名,但似乎无法弄清楚如何使AWS节点将属性用作文件名。 根据文件“Amazon S3 out节点。将内容上载到Amazon S3 bucket。bucket名称可以在node bucket属性或msg.bucket属性中指定。Amazon S3上的文件名取自node filename属性或msg.filename属性。内容取自node属

  • 我有一个这样的结构` ...等等,在

  • 我在Hadoop权威指南中讨论了高可用性, 在这种情况下,若要从失败的namenode中恢复,管理员将使用其中一个文件系统元数据副本启动一个新的主namenode,并配置Datanode和客户端以使用这个新的namenode。新的namenode无法为请求提供服务,除非它 (i)将其名称空间映像加载到存储器中, 以下是我的问题:- a.)“文件系统元数据副本”是指在NFS中备份文件系统或以辅助名称

  • 我可能在这里滥用了Freemarker,但我希望能够使用它从XML文档的外部剥离包装元素,类似于: 我想删除外部的元素。很容易通过以下内容选择内部文档: 有什么想法吗?