有没有一种方便的方法来获取 TensorFlow 对象检测模型的特定类别的平均精度分数？

【问题标题】：Is there a convenient way to get class-specific Average Precision scores for a TensorFlow Object Detection model?有没有一种方便的方法来获取 TensorFlow 对象检测模型的特定类别的平均精度分数？
【发布时间】：2021-08-31 18:47:44
【问题描述】：

TensorFlow 对象检测 API 提供了一种非常方便的方法来获取训练模型的性能指标（在他们的教程 here 中进行了描述）。不幸的是，没有提供每个类的平均精度分数。

有人知道无需编写自己的脚本即可获得 TensorFlow 对象检测模型的特定类别 AP 分数的便捷方法吗？

same question 在 2019 年被问到但没有得到答复。

谢谢！

【问题讨论】：

不，这没有意义，你可以按班级获得AP，mAP适用于所有班级。
已编辑，感谢反馈

标签： python tensorflow object-detection-api

【解决方案1】：

经过一番搜索，我找到了几个解决方案。

1.使用不同的评估配置

只需将模型的*.config 文件中的metrics_set 值更改为"pascal_voc_detection_metrics"。

TensorFlow 对象检测 API 支持各种评估指标，文档here 中有详细说明。 PASCAL VOC 2010 检测指标给出每个类别的 AP 分数。

2。编辑 pycocotools 包中的 cocoeval.py 文件

此方法涉及将 8 行代码粘贴到 cocoeval.py 文件中。 this 帖子中对此进行了很好的解释和记录。

【讨论】：