/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

0 vote346 views2 answers
Iris data pandas scatterplot
Python
0 vote284 views1 answer
Python which __divs__ are there
Python
52 votes368 views6 answers
Sorting a dictionary in python
Python
0 vote317 views1 answer
How to use assert in python
Python
5 votes310 views1 answer
Random forest regressor python
Python
0 vote322 views1 answer
Django loop through form errors
Python
1 vote365 views2 answers
Load data python
Python
2 votes332 views2 answers
Pandas unique values to list
Python
9 votes303 views6 answers
Functions in python
Python
0 vote337 views1 answer
IntersectAll dynamo revit
Python