如何在tensorflow中查找给定.ckpt.meta文件的输出节点名称

Question

到目前为止，我可以获得给定任何.ckpt.meta文件的所有节点名称的列表，但我想知道是否有系统的方法可以从列表中找到输出节点名称。

import tensorflow as tf
tf.reset_default_graph()
with tf.Session() as sess:
saver = tf.train.import_meta_graph('mymodel.ckpt.meta')
graph_def = tf.get_default_graph().as_graph_def()
node_list=[n.name for n in graph_def.node]

Answer 1

您可以尝试：

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

Answer 2

这对我有效：

import tensorflow as tf
def get_node_name():
tf.reset_default_graph()
with tf.Session() as sess:
saver = tf.train.import_meta_graph(meta_file)
graph_def = tf.get_default_graph().as_graph_def()
name_list = []


for node in graph_def.node:
name_list.append(node.name)

outputs = set(name_list)

for index, output in enumerate(outputs):
print('Node Name: ', output)

然后运行：

get_node_name()