旋转框IoU计算

最新推荐文章于 2025-02-22 00:34:27 发布

戊辰happy

最新推荐文章于 2025-02-22 00:34:27 发布

阅读量403

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_42569775/article/details/137052189

版权

本文介绍了一个使用Python和OpenCV库函数计算两个旋转矩形之间的交并比（IOU）的方法，通过定义函数`iou_rotate_calculate`，并给出了实际的代码示例和应用实例。

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

# 中心点坐标、矩形的宽和高、旋转角（单位是角度，不是弧度）
def iou_rotate_calculate(box1, box2):
    area1 = box1[2] * box1[3]
    area2 = box2[2] * box2[3]
    r1 = ((box1[0], box1[1]), (box1[2], box1[3]), box1[4])
    r2 = ((box2[0], box2[1]), (box2[2], box2[3]), box2[4])
    int_pts = cv2.rotatedRectangleIntersection(r1, r2)[1]
    if int_pts is not None:
        order_pts = cv2.convexHull(int_pts, returnPoints=True)
        intersec = cv2.contourArea(order_pts)
        union = area1 + area2 - intersec
        iou = intersec * 1.0 / union
    else:
        iou = 0.0
    return iou

box1 = [30, 40, 40, 80, 0]
box2 = [30, 40, 40, 80, 90]
iou = iou_rotate_calculate(box1,box2)
print(iou)