目标检测中计算图像的交并比(IOU)

交并比为目标检测中一个重要的概念,来进行预测框(ground truth)与标注框的重合率来对检测的正确率进行比较。图示如下:
在这里插入图片描述
交并比计算公式:
I O U = C ⋂ G C ⋃ G IOU=\frac{C \bigcap G }{C \bigcup G} IOU=CGCG
代码示例:

# -*-coding:utf-8 -*- 
# 计算图像的(IOU)交并比

def Iou(box1,box2):
	xmin1, ymin1, xmax1, ymax1 = box1
	xmin2, ymin2, xmax2, ymax2 = box2
	# 求交集部分左上角的点
	xmin = max(xmin1,xmin2)
	ymin = max(ymin1,ymin2)
	# 求交集部分右下角的点
	xmax = min(xmax1,xmax2)
	ym=ax = min(ymax1,ymax2)
	# 计算输入的两个矩形的面积
	s1 = (xmax1-xmin1) * (ymax1 - ymin1)
	s2 = (xmax2-xmin2) * (ymax2 - ymin2)

	#计算总面积
	s = s1 + s2 
	# 计算交集
	inter_area = (xmax - xmin) * (ymax - ymin)

	iou = inter_area / (s - inter_area)
	return iou

if __name__ == '__main__':
	b1 = [1,1,3,3]
	b2 = [2,2,4,4]
	print(Iou(b1,b2))           # 0.14285714285714285

计算交并比的方法多种多样,在目标检测中也不止交并比一个值来计算检测的准确性,其他方法有待发现。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值