如何找到输出节点名来构建冻结图?

我建议使用Tensorboard来可视化图形结构,而不是使用带有节点的文本文件。你可以找到更多的细节 here .

然而,图本身没有输入或输出的概念。可以将没有输入连接的节点视为输入节点的良好候选节点,特别是占位符节点。连接到损失函数的节点是很好的输出节点候选者。

总结:一般来说,您需要通过分析网络体系结构来猜测哪些节点是输入,哪些是输出。

至于您引用的存储库,您可以阅读 eval.py

input_x = graph.get_operation_by_name("input_x").outputs[0]
# input_y = graph.get_operation_by_name("input_y").outputs[0]
dropout_keep_prob = graph.get_operation_by_name("dropout_keep_prob").outputs[0]

# Tensors we want to evaluate
predictions = graph.get_operation_by_name("output/predictions").outputs[0]

所以很可能输入节点是“input_x”,输出节点是“output/predictions”。