2

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

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

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

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

如何找到我需要的张量名称?还是有另一种我不知道的方法?

4

1 回答 1

2

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

于 2017-10-18T12:18:13.030 回答