python-opencv calculates overlapping rectangular area IOU

import cv2
import numpy as np
def calc_riou(r1,r2):
    r1=np.array(r1)
    r2=np.array(r2)
    rect1 = ((r1[0],r1[1]),(r1[2],r1[3]),r1[4])
    rect2 = ((r2[0],r2[1]),(r2[2],r2[3]),r2[4])
    int_pts = cv2.rotatedRectangleIntersection(rect1, rect2)[1]
    if int_pts is not None:
        order_pts = cv2.convexHull(int_pts, returnPoints=True)
        int_area = cv2.contourArea(order_pts)
    else:
        int_area=0
    return int_area
if __name__ == '__main__':
    bbox1=np.array([100,50,20,10,360],dtype=np.float)
    bbox2=np.array([100,50,20,10,0],dtype=np.float)
    print("riou ",calc_riou(bbox1,bbox2))

Calculate the coincident area between boundingboxes

The last dimension is the angle (degree)

Guess you like

Origin blog.csdn.net/li4692625/article/details/128005771