/Python/Function to measure intersection over union

Function to measure intersection over union

# function to compute the IoU function
def bb_intersection_over_union(boxA, boxB):
	# determine the (x, y)-coordinates of the intersection rectangle
	xA = max(boxA[0], boxB[0])
	yA = max(boxA[1], boxB[1])
	xB = min(boxA[2], boxB[2])
	yB = min(boxA[3], boxB[3])
	# compute the area of intersection rectangle
	interArea = max(0, xB - xA + 1) * max(0, yB - yA + 1)
	# compute the area of both the prediction and ground-truth
	# rectangles
	boxAArea = (boxA[2] - boxA[0] + 1) * (boxA[3] - boxA[1] + 1)
	boxBArea = (boxB[2] - boxB[0] + 1) * (boxB[3] - boxB[1] + 1)
	# compute the intersection over union by taking the intersection
	# area and dividing it by the sum of prediction + ground-truth
	# areas - the interesection area
	iou = interArea / float(boxAArea + boxBArea - interArea)
	# return the intersection over union value
	return iou

Source: pyimagesearch.com

Similar codes

14 votes152 views1 answer
Create data dir in python
Python
32 votes127 views6 answers
Add to list python
Python
35 votes126 views6 answers
Django version check
Python
1 vote334 views1 answer
How to use an indefinite number of args in python
Python
-1 vote114 views2 answers
Python array use numpy arange
Python
0 vote118 views1 answer
How to create list of objects in python
Python
0 vote156 views1 answer
Python a, b =
Python
0 vote79 views1 answer
Node alpine install python
Python
0 vote131 views2 answers
Comment faire un long commentaire en python
Python
3 votes122 views1 answer
Presentation in jupyter notebook
Python