目标检测计算mAP，AP，Recall，Precision的计算方式和代码(YOLO和FastRCNN等)

最新推荐文章于 2025-06-25 16:52:12 发布

是否龙磊磊真的一无所有

最新推荐文章于 2025-06-25 16:52:12 发布

阅读量8.3k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：目标检测 opencv，图像处理，特征提取 CNN+深度学习文章标签：目标检测 YOLO mAP FastRCNN 深度学习

本文链接：https://blog.csdn.net/qq_32998593/article/details/88784243

opencv，图像处理，特征提取同时被 3 个专栏收录

23 篇文章

订阅专栏

CNN+深度学习

19 篇文章

订阅专栏

目标检测

15 篇文章

订阅专栏

本文详细介绍了目标检测中平均精度(mAP)的计算方法，包括构建精确率-召回率曲线及计算曲线下的面积。提供了具体的Python代码实现，适用于评估目标检测算法性能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

目标检测中计算mAP是较为复杂的，并不是很多讲解中说的那个计算precision和recall，然后总的ground truth 目标和检测出的真实目标做除法就可以了。而是需要构建precision和recall 曲线，然后计算曲线面积。

一下是代码，可以去相关网站查看计算过程，过程较为复杂。

mAP计算详解 GitHub - rafaelpadilla/Object-Detection-Metrics: Most popular metrics used to evaluate object detection algorithms.

FastRCNN的mAP计算过程。GitHub - rbgirshick/py-faster-rcnn: Faster R-CNN (Python implementation) -- see https://github.com/ShaoqingRen/faster_rcnn for the official MATLAB version

def ap_per_class(tp, conf, pred_cls, target_cls):
    """ Compute the average precision, given the recall and precision curves.
    Source: https://github.com/rafaelpadilla/Object-Detection-Metrics.
    # Arguments
        tp:    True positives (list).
        conf:  Objectness value from 0-1 (list).
        pred_cls: Predicted object classes (list).
        target_cls: True object classes (list).
    # Returns
        The average precision as computed in py-faster-rcnn.
    """

    # Sort by objectness
    i = np.argsort(-conf)
    tp, conf, pred_cls = tp[i], conf[i], pred_cls[i]

    # Find unique classes
    unique_classes = np.unique(np.concatenate((pred_cls, target_cls), 0))

    # Create Precision-Recall curve and compute AP for each class
    ap, p, r = [], [], []
    for c in unique_classes:
        i = pred_cls == c
        n_gt = sum(target_cls == c)  # Number of ground truth objects
        n_p = sum(i)  # Number of predicted objects

        if (n_p == 0) and (n_gt == 0):
            continue
        elif (n_p == 0) or (n_gt == 0):
            ap.append(0)
            r.append(0)
            p.append(0)
        else:
            # Accumulate FPs and TPs
            fpc = np.cumsum(1 - tp[i])
            tpc = np.cumsum(tp[i])

            # Recall
            recall_curve = tpc / (n_gt + 1e-16)
            r.append(tpc[-1] / (n_gt + 1e-16))

            # Precision
            precision_curve = tpc / (tpc + fpc)
            p.append(tpc[-1] / (tpc[-1] + fpc[-1]))

            # AP from recall-precision curve
            ap.append(compute_ap(recall_curve, precision_curve))

    return np.array(ap), unique_classes.astype('int32'), np.array(r), np.array(p)


def compute_ap(recall, precision):
    """ Compute the average precision, given the recall and precision curves.
    Source: https://github.com/rbgirshick/py-faster-rcnn.
    # Arguments
        recall:    The recall curve (list).
        precision: The precision curve (list).
    # Returns
        The average precision as computed in py-faster-rcnn.
    """
    # correct AP calculation
    # first append sentinel values at the end

    mrec = np.concatenate(([0.], recall, [1.]))
    mpre = np.concatenate(([0.], precision, [0.]))

    # compute the precision envelope
    for i in range(mpre.size - 1, 0, -1):
        mpre[i - 1] = np.maximum(mpre[i - 1], mpre[i])

    # to calculate area under PR curve, look for points
    # where X axis (recall) changes value
    i = np.where(mrec[1:] != mrec[:-1])[0]

    # and sum (\Delta recall) * prec
    ap = np.sum((mrec[i + 1] - mrec[i]) * mpre[i + 1])
    return ap