def iou(bbox1, bbox2):
# bbox1 = [x0,y0,x1,y1]
x0, y0, x1, y1 = bbox1
x2, y2, x3, y3 = bbox2
s1 = (x1-x0)*(y1-y0)
s2 = (x3-x2)*(y3-y2)
w = max(0, min(x1, x3) - max(x0,x2))
h = max(0, min(y1, y3) - max(y0, y2))
inter = w*h
iou = inter/(s1+s2-inter)
return iou
print(iou((8,5,15,9), (10,6,18,20)))
手撕iou python
最新推荐文章于 2024-07-29 22:34:10 发布