在预测中使用 TensorFlow 对象检测 API 模型

Question

我使用 TensorFlow object detection API 从头开始​​训练 SSD Inception 模型。评估脚本表明该模型已经学到了一些东西，现在我想使用该模型。

我查看了对象检测ipynb，它可以将单个图像提供给经过训练的模型。但是，这适用于带有 MobileNet 的 SSD。我使用以下行（加载元图后）打印我训练的 TensorFlow 模型的张量名称。

print([str(op.name) for op in tf.get_default_graph().get_operations()] )

但它不包含与 ipynb 中相同的输入或输出张量名称。我也搜索了代码，但是许多函数相互指向，很难找到我要找的东西。

如何找到我需要的张量名称？还是有其他我不知道的方法？

Answer 1

要使用图表，您需要使用this provided script 冻结/导出它。生成的.pb 文件将包含您需要的节点。我不知道为什么它是这样组织的，但确实如此。