首页 > 其他分享> > COCO 目标检测 mAP AR等指标的计算

COCO 目标检测 mAP AR等指标的计算

2022-09-03 22:30:08 作者：互联网

AR

Average Precision(AP)

used for Object Detection

mAP = mean AP(per class)

https://kharshit.github.io/blog/2019/09/20/evaluation-metrics-for-object-detection-and-segmentation

如何快速计算IoU

这里不太确定是不是要计算所有Ground Truth以及Bounding Boxes的IoU

怎样的检测结果算是正确的

没有绝对正确,\(IoU = 1.0\)的情况

通过设定阈值，将足够接近的例子认定为TP

这样的认定方法对不同准确率的方法是不公平的
mAP是对AP的推广，通过取不同的阈值计算衡量总体性能

怎样计算AP

Average Precision (AP)的精确定义是 Area Under Precision-Recall Curve

\[AP = \int_{0}^{1}\text{PR}(r)\text{d}r \]

Precision-Recall Curve

注意，在实际工程当中，没有真实的连续曲线，需要指定离散的recall

比如COCO采用的就是[0:.01:1],101个 recall 作为阈值，并按照下列公式近似

\[\text{AP} = \frac{1}{101}(P_{0}+P_{0.01}+P_{0.02} + \dots + P_{1}) \]

\(P_{\text{recall}}\)代表特定recall（即特定IoU threshold）下的准确率

如何计算TP,FN

在目标检测中，检测器得到的结果要么是True Positive，要么是False Positive

P1是对应ground truth A的TP，
P2是FN，

假设提供的有标签数据中，共有\(K\)个 ground truth标注框，在\(N\)帧中共检出\(M\)个框

对于任意一个检测器的输出，都需要进行考察

首先，检测器的框一定包含cls confidence，一般是一个\(C\)（类别个数）维的向量。对每一个检测框，都为之计算一个IOU分数

Confidence score is the probability that an anchor box contains an object. It is usually predicted by a classifier.

采用如下伪代码计算

for each bbox that has a confidence score > threshold:

  among the ground-truths

  choose one that belongs to the same class and has the highest IoU with the bbox

  if no ground-truth can be chosen or IoU < threshold (e.g., 0.5):
    the detection bbox is a false positive
  else:
    the detection bbox is a true positive