calculate iou of two rectangles

Coderunner answered on July 6, 2022 Popularity 3/10 Helpfulness 1/10

answer calculate iou of two rectangles

calculate iou of two rectangles

Comment

from shapely.geometry import Polygon
""" Assuming two rectangle with the following corner points:
box_1 = [[x_1, y_1], [x_2, y_1], [x_2,y_2], [x_1,y_2]]
box_2 = [[x_1, y_1], [x_2, y_1], [x_2,y_2], [x_1,y_2]]
(x_1,y_1)+---- +(x_2,y_1)
         |box_1|   
(x_1,y_2)+-----+(x_2,y_2)"""

def calculate_iou(box_1, box_2):
    poly_1 = Polygon(box_1)
    poly_2 = Polygon(box_2)
    iou = poly_1.intersection(poly_2).area / poly_1.union(poly_2).area
    return iou


box_1 = [[511, 41], [577, 41], [577, 76], [511, 76]]
box_2 = [[544, 59], [610, 59], [610, 94], [544, 94]]

print(calculate_iou(box_1, box_2))

Popularity 3/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: python rectangles

Link to this answer
Share Copy Link

Contributed on Jul 06 2022

Coderunner

0 Answers Avg Quality 2/10

calculate iou of two rectangles

Contents

More Related Answers

calculate iou of two rectangles

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.