如何查找冻结模型的输入和输出节点

Question

我想在模型动物园的冻结模型上使用tensorflow的child-src脚本：optimize_for_inference.py。

如何查找/确定模型的输入和输出名称？

Here is a link to the graph generated by tensorboard

这个问题可能有所帮助：Given a tensor flow model graph, how to find the input node and output node names（对我来说没有）

Answer 1

我们现在已经添加了一些有关此流程的文档，请参阅此处：

https://www.tensorflow.org/mobile/prepare_models

如果您查找summarize_graph，您将看到如何找出正确的输入和输出节点的示例。