机器视觉中的坐标体系
注意区分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
参考
添加链接描述
添加链接描述