我已经使用TensorFlow object detection API从头开始训练SSD Inception模型。评估脚本显示模型已经学到了什么,现在我想使用该模型。
我已经查看了可以将单个图像提供给训练模型的对象检测ipynb。但是,这适用于带有MobileNet的SSD。我使用了以下行(在加载元图之后)来打印我训练过的TensorFlow模型的张量名称。
print([str(op.name) for op in tf.get_default_graph().get_operations()] )
但它不包含与ipynb中相同的输入或输出张量名称。我也搜索过代码,但很多功能都指向对方,很难找到我要找的东西。
如何找到我需要的张量名称?还是有另一种我不知道的方法?