【深度学习】目标检测之评价指标

【深度学习】目标检测之评价指标mAP

评价指标

时隔一年,又重新开始了学习之路。哈哈哈废话不多说,开始我最近整理的一些内容叭。今天要整理的知识点是目标检测领域常用的几个评价指标,准确率Precision,召回率Recall,交并比IoU,平均精度AP,多个类别AP的平均值即mAP等等。

混淆矩阵

目标检测模型通常会输出很多个检测框,我们是通过统计并计算每个检测框是否能检测到目标的各种占比来衡量模型的检测效果,因此,我们会把检测框分成如下四种情况,判断的依据主要是通过计算交并比(IoU,这个下面会讲到)。

  • TP (True Positive) ,真的正样本 = 正样本 被分类为 正样本;
  • TN (True Negative) ,真的负样本 = 负样本 被分类为 负样本;
  • FP (False Positive) ,假的正样本 = 负样本 被分类为 正样本;
  • FN (False Negative) ,假的负样本 = 正样本 被分类为 负样本(通常为漏检);

按照检测框的检测情况,我们写成一个4×4的矩阵形式(称为混淆矩阵):
分类情况

预测为正类预测为负类真实正样本TPFN真实负样本FPTN

对于一张图像而言,预测的情况和真实样本的情况如下(记住这个,后面计算不会混乱)

【深度学习】目标检测之评价指标

; Precision 精确率(精度)

Precision:指的是在模型预测的结果中,其中正确的有多少个。看上面的图,预测结果有TP+FP。

P r e c i s i o n = T P T P + F P Precision = \frac{TP}{TP+FP}P r e c i s i o n =T P +F P T P ​

Recall 召回率

Recall:指的是在所有的真实目标中,其中正确的有多少个。看上面的图,真实目标有TP+FN。

R e c a l l = T P T P + F N Recall= \frac{TP}{TP+FN}R e c a l l =T P +F N T P ​

IoU 交并比

IoU:指的是检测框与样本标注的矩形框的交集与并集的比值。

【深度学习】目标检测之评价指标

; 平均精度 AP

AP(Average Precision)指的是Precision-Recall曲线下的面积。

先从单个类别(猫)开始计算,假设一共有3张图片,绿色框是GT(7个),红色框是预测框(7个)并带有置信度。现在假设IOU=50%,按照置信度排序得到下表

【深度学习】目标检测之评价指标

【深度学习】目标检测之评价指标

【深度学习】目标检测之评价指标
得到统计表后,根据confidence取不同的阈值,可以计算Precision和Recall
【深度学习】目标检测之评价指标

【深度学习】目标检测之评价指标
【深度学习】目标检测之评价指标

这样我们就可以根据表计算种类猫的AP值了

猫 A P = ∑ i = 1 R a n k ( R e c a l l i − R e c a l l i − 1 ) × P r e c i s i o n i 猫AP=\sum_{i=1}^{Rank}(Recall_i – Recall_{i-1})\times Precision_i 猫A P =i =1 ∑R a n k ​(R e c a l l i ​−R e c a l l i −1 ​)×P r e c i s i o n i ​

即:猫AP=(0.14-0)x1.0 + (0.28-0.14)x1.0 + (0.42-0.28)x1.0 + (0.57-0.42)x1.0 + (0.71-0.57)x0.71=0.6694

mAP

mAP (means Average Precision) 即各类别的AP值的均值

m A P = ∑ j c l a s s _ n u m b e r C l a s s _ A P j mAP = \sum_{j}^{class_number} Class_AP_j m A P =j ∑c l a s s _n u m b e r ​C l a s s _A P j ​

总结

完结。谢谢,觉得有帮助的可以点个赞~

Original: https://blog.csdn.net/Resume_f/article/details/122292101
Author: Florrie Zhu
Title: 【深度学习】目标检测之评价指标

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/683087/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球