python 计算iou

参考：

https://blog.csdn.net/sinat_34474705/article/details/80045294

 

def caliou(im1, im2):
    x11, y11, x12, y12 = im1[0], im1[1], im1[2], im1[3]
    x21, y21, x22, y22 = im2[0], im2[1], im2[2], im2[3]
#    print(im1)
#    print(im2)
    if x11>=x12 or y11<=y12 or x21>=x22 or y21<=y22:return 0
    newx1 = max(x11, x21)
    newy1 = min(y11, y21)
    newx2 = min(x12, x22)
    newy2 = max(y12, y22)

    if abs(newx1-newx2)<=0 or abs(newy1-newy2)<=0:return 0
    area = abs(newx1-newx2)*abs(newy1-newy2)
    
    s1 = abs(y12-y11)*abs(x12-x11)
    s2 = abs(y22-y21)*abs(x22-x21)
    
    iou = area/(s1+s2-area)
    
    return iou

im1 = [0, 0, 3, 3]
im2 = [2, 2, 4, 4]

print(caliou(im1, im2))