从Tensorflow对象检测API中获得的分数计算mAP

Question

我实际上在一个服务器上工作，我运行Tensorflow OD API来训练我的自定义数据集的模型。因此，我将我的图像分为训练，验证和测试集，并在训练和验证集上运行train.py。接下来，我使用导出的模型检查点和冻结图在我的测试图像上运行推理。现在，我的问题是当我运行示例中提供的推理时，我得到一个输出字典，其中包含每个图像的检测分数，检测数，检测类，检测掩码等，因此从这些输出中我如何计算mAP为我的测试集？

在此提前，任何指导都将非常有用。

Answer 1

您可以使用COCO的API通过TF OD API计算COCO的指标。见this。 TF通过您的检测和GT提供COCO API，COCO API将计算COCO的指标并将其返回TF（因此您可以在TensorBoard中显示其进度）。 [email protected]可能是最相关的度量标准（它是用于PASCAL VOC，Open Images等的标准度量标准），而[email protected]：0.95是一个更难以定位的方法。