我已使用Tensorflow训练了模型(faster_rcnn_resnet101_coco_2018_01_28)。我有一个“。pb” 图。
要制作冻结图,我需要输入和输出节点。
如何在图形中找到它?
我有节点here的完整列表。
没有其他文章中建议的“ Softmax”,“占位符”之类的节点。
答案 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