python - 如何从 Tensorflow 中的卡住模型(pb 文件)中找到 output_node_names？

Question

我正在尝试将 freeze_model.pb 转换为 TensorFlow JS 兼容 (.pb) 文件，该文件基于 Tensorflow 的 SSD Mobilenet V2 COCO 预训练模型。我陷入了如何获取使用tensorflowjs_converter 时所需的output_node_names 参数的困境。如何知道输出节点名称？

我尝试使用下面的 Python 脚本获取操作名称，但无法理解哪个是输出节点。

def load_graph(model_file):
  graph = tf.Graph()
  graph_def = tf.GraphDef()

  with open(model_file, "rb") as f:
    graph_def.ParseFromString(f.read())
  with graph.as_default():
    tf.import_graph_def(graph_def)

  return graph

graph = load_graph('frozen_model.pb')
ops = graph.get_operations()

Answer 1

首先，您可以检查 graph_def 中的所有节点，如下所示:

for node in graph_def.node
    print(node.name)

或者，如果您想直观地查看图形并确定将哪个节点用作输出，则可以使用 TensorBoard。有一个工具叫import_pb_to_tensorboard 。它本质上是使用几行代码将图形写入 log_dir，您可以将张量板指向该目录。您只需将这些行复制到您自己的脚本中即可实现相同的效果，而无需从 TensorFlow 存储库进行构建。

第三，还有一个工具叫 summarize_graph tool :

bazel build tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=/path/to/your/graph.pb