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