Tensorflow:如何从模型图获取输入和输出节点?

时间:2019-05-21 12:19:32

标签: tensorflow

我已使用Tensorflow训练了模型(faster_rcnn_resnet101_coco_2018_01_28)。我有一个“。pb” 图。

要制作冻结图,我需要输入和输出节点

如何在图形中找到它?

我有节点here的完整列表。

没有其他文章中建议的“ Softmax”,“占位符”之类的节点。

1 个答案:

答案 0 :(得分:0)

使用C ++构建的摘要图。我的建议是您可以使用Docker运行摘要图并获取经过训练的模型的输入和输出名称。请参见此链接

GITHUB LINK FOR GRAPH TRANSFORMATION TOOL(SUMMARIZE GRAPH)

bazel build tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=tensorflow_inception_graph.pb