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

时间:2017-10-18 11:51:50

标签: tensorflow deep-learning tensorflow-slim

我已经使用TensorFlow object detection API从头开始训练SSD Inception模型。评估脚本显示模型已经学到了什么,现在我想使用该模型。

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

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

但它不包含与ipynb中相同的输入或输出张量名称。我也搜索过代码,但很多功能都指向对方,很难找到我要找的东西。

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

1 个答案:

答案 0 :(得分:2)

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