2023-简单点-IOU计算

机器视觉中的坐标体系

注意区分x,y坐标系和row,col排布

IOU交集

代码

def IOU(RecA, RecB):
	'''recA是坐标形式是[X[左上点],y[左上点],x[右下点],y[右下点]]
	'''
	#找到交集框的左上和右下点，可以计算交集面积
    xA = max(RecA[0], RecB[0])
    yA = max(RecA[1], RecB[1])
    xB = min(RecA[2], RecB[2])
    yB = min(RecA[3], RecB[3])
    # 计算交集部分面积; 注意可能没有交集
    interArea = max(0, xB - xA + 1) * max(0, yB - yA + 1)
    # 计算预测值和真实值的面积
    RecA_Area = (RecA[2] - RecA[0] + 1) * (RecA[3] - RecA[1] + 1)
    RecB_Area = (RecB[2] - RecB[0] + 1) * (RecB[3] - RecB[1] + 1)
    # 计算IOU
    iou = interArea / float(RecA_Area + RecB_Area - interArea)
    
    return iou



1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

参考

添加链接描述
添加链接描述