精确度和召回率
但是在目标检测中,单独看precision和recall指标并不能完全反映预测效果,所以需要计算AP或mAP指标。
AP计算
AP:Average Precision,即平均精确度。
AP的值就是Precision-Recall曲线下的面积。
AP是针对单个类别进行计算的。
计算过程:
1.准备测试集;
2.统计GT的class和box,预测框的score、class和box;
3.对于每个类别,取出所有目标,按score从大到小排序(标记图片id);
4.开始循环,按score依次取出预测框,和GT同id图片中的同class目标进行IOU计算(设置IOU阈值0.5/0.95等);
5.如果大于IOU_thresh,统计对应的TP或FP,直到该类别循环结束;
6.计算该类别各score下的presicion和recall,画出PR曲线;
7.计算PR曲线下的面积即为该类的AP。
某一类的PR曲线如下所示:
mAP计算
mAP为对所有类的AP值求平均。