python - 如何在 tensorflow 对象检测 API 中更改默认输出、输入张量名称？

Question

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

Answer 1

输出张量名称在对象检测项目的“meta_architectures”python 文件中定义，可在此处找到。当您进行模型导出以进行推理时，您可以尝试调整/转换输出张量的命名，但我自己没有尝试过。