如何在tensorflow对象检测API中更改默认输出,输入张量名称?

问题描述

Tensorflow对象检测API认情况下具有输出张量名称 num_detections detection_Boxes detection_scores detection_classes 。如何更改这些名称?不幸的是,我正在测试的推理平台不接受带下划线的张量名称。 谢谢..

解决方法

输出张量名称在对象检测项目的{meta_architectures“ python文件中定义,可以在here中找到。 在进行模型导出以进行推断时,您可以尝试在那里调整/转换输出张量的命名,但是我自己还没有尝试过。