calculate IoU of two boxes

最新推荐文章于 2024-06-12 13:50:32 发布

儒雅的小Z

最新推荐文章于 2024-06-12 13:50:32 发布

阅读量284

点赞数

文章标签： IOU python

本文链接：https://blog.csdn.net/guilutian0541/article/details/81329520

版权

def cal_bbox_iou(boxA, boxB): 

    xA = max(boxA[0], boxB[0]) #xmin
    yA = max(boxA[2], boxB[2]) #ymin
    xB = min(boxA[1], boxB[1]) #xmax
    yB = min(boxA[3], boxB[3]) #ymax

    if xA < xB and yA < yB: 
        interArea = (xB - xA + 1) * (yB - yA + 1) 
        boxAArea = (boxA[1] - boxA[0] + 1) * (boxA[3] - boxA[2] + 1) 
        boxBArea = (boxB[1] - boxB[0] + 1) * (boxB[3] - boxB[2] + 1) 
        iou = interArea / float(boxAArea + boxBArea - interArea+0.00001) 
    else: 
        iou=0.0

    return iou