IOU计算

原创于 2020-12-17 14:02:56 发布 · 352 阅读

0 ·

CC 4.0 BY-SA版权

目标检测专栏收录该内容

2 篇文章

订阅专栏

博客提及输入坐标存在两种形式，分别是xyxy和xywh，这在信息技术领域的坐标处理等方面有应用。

输入坐标分xyxy,和xywh

def count_iou(r_a, r_b, mode='xyxy'):
    if mode == 'xyxy':  # xyxy: 即(x1,y1,x2,y2)
        pass
    elif mode == 'xywh':  # xywh: (x,y)为中心点坐标, wh为宽高
        x, y, w, h = r_a
        r_a = [x - w / 2, y - h / 2, x + w / 2, y + h / 2]
        x, y, w, h = r_b
        r_b = [x - w / 2, y - h / 2, x + w / 2, y + h / 2]

    x1 = max(r_a[0], r_b[0])
    y1 = max(r_a[1], r_b[1])
    x2 = min(r_a[2], r_b[2])
    y2 = min(r_a[3], r_b[3])

    inter_s = max(0, x2 - x1) * max(0, y2 - y1)
    outer_s = (r_a[2] - r_a[0]) * (r_a[3] - r_a[1]) + (r_b[2] - r_b[0]) * (r_b[3] - r_b[1]) - inter_s

    iou = inter_s / outer_s
    return iou

ra = [70, 70, 100, 100]
rb = [60, 60, 120, 120]

iou = count_iou(ra, rb)
print(iou)