两个旋转矩形的IOU计算方法
# 中心点 矩形的w h, 旋转的theta(角度,不是弧度)
def iou_rotate_calculate(boxes1, boxes2):
area1 = boxes1[:, 2] * boxes1[:, 3]
area2 = boxes2[:, 2] * boxes2[:, 3]
ious = []
for i, box1 in enumerate(boxes1):
temp_ious = []
r1 = ((box1[0], box1[1]), (box1[2], box1[3]), box1[4])
for j, box2 in enumerate(boxes2):
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)
int_area = cv2.contourArea(order_pts)
inter = int_area * 1.0 / (area1[i] + area2[j] - int_area)
temp_ious.append(inter)
else:
temp_ious.append(0.0)
ious.append(temp_ious)
return np.array(ious, dtype=np.float32)
box1=np.array([[2.5, 3.5, 4.242, 2.828, 0.785]])
# box2=np.array([[2.5, 3.5, 2.828, 4.242, -0.785]]) # 0.5
# box2=np.array([[3, 2, 3.16, 3.16, 0.32]]) # 0.27
# box2=np.array([[3.5, 2.5, 4.242, 2.828, 0.785]]) # 0.33
box2=np.array([[2.5, 3.5, 3, 3, 0]]) # 0.678
a=iou_rotate_calculate(box1,box2)
print(a)