目标检测计算两个标注框的交并比(python代码)

Posted 悠悠青青

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了目标检测计算两个标注框的交并比(python代码)相关的知识,希望对你有一定的参考价值。

def iou(box1, box2):
    '''
    两个框(二维)的 iou 计算
    box:[xmin,ymin,xmax,ymax], 依次为左上、右下坐标
    @return: IOU值
    '''
    h = max(0, min(box1[2], box2[2]) - max(box1[0], box2[0]))	
    w = max(0, min(box1[3], box2[3]) - max(box1[1], box2[1]))
    area_box1 = ((box1[2] - box1[0]) * (box1[3] - box1[1]))
    area_box2 = ((box2[2] - box2[0]) * (box2[3] - box2[1]))
    inter = w * h
    union = area_box1 + area_box2 - inter
    iou = inter / union
    return iou

以上是关于目标检测计算两个标注框的交并比(python代码)的主要内容,如果未能解决你的问题,请参考以下文章

目标检测中的precision,recall,AP,mAP计算详解

目标检测评价指标总结

如何计算交并比IoU | 含图片说明和伪代码实现

目标检测中各种IoU说明

YOLO目标检测之IOU计算及其衍变体

intersection-over-union(IoU: 交并比)