交并比(IOU)的计算方法
创始人
2024-05-31 04:52:54
0

        交并比(IOU, Intersetion Over Union),意思就是交集和并集的比值,用来评价两个几何图形的面积重合度,在目标检测算法模型中,通常被用来计算预测框与真实框的误差(损失函数)或者在非极大值抑制(NMS)算法中用于过滤多余重叠的目标,亦或者在计算mAP时用于判断目标是否被预测到。这是一个应用非常广泛的算法,这里详细介绍一下它的算法原理,以及代码的实现方式。

图1 交并比示意图

         如图1所示的是两个矩形方框的交并比示意图,中间重合部分就是矩形A和矩形B的交集,那么这两个矩形框的交并比计算公式如下:

IOU=\frac{A\cap B}{A\cup B}

其中,A与B的并集又可以通过以下公式计算:

A\cup B=A+B-A\cap B

定义矩形 A和矩形B的对角坐标分别为[\left ( xmin_{A},ymin_{A} \right ),\left ( xmax_{A},ymax_{A} \right )][\left ( xmin_{B},ymin_{B} \right ),\left ( xmax_{B},ymax_{B} \right )],同时又定义交集矩形的对角坐标为[\left ( xmin_{AB},ymin_{AB} \right ),\left ( xmax_{AB},ymax_{AB} \right )],那么交集矩形对角坐标的计算方法如下:

xmin_{AB}=max(xmin_{A},xmin_{B})

ymin_{AB}=max(ymin_{A},ymin_{B})

xmax_{AB}=min(xmax_{A},xmax_{B})

ymax_{AB}=min(ymax_{A},ymax_{B})

于是,交集和并集的计算如下:

A\cap B=\left ( xmax_{AB}-xmin_{AB} \right )*\left ( ymax_{AB} -ymin_{AB}\right )

W_{A}=xmax_{A}-xmin_{A}

H_{A}=ymax_{A}-ymin_{A}

W_{B}=xmax_{B}-xmin_{B}

H_{B}=xmax_{B}-xmin_{B}

A\cup B=W_{A}*H_{A}+W_{B}*H_{B}-A\cap B

以上公式,是根据两个矩形有交集的情况推导出来的,如果两个矩形没有交集,以上公式如何做出判断呢?如果是这两个矩形不存在交集,那么只要判断A\cap B\leq 0即可,即是说,如果两个矩形无交集,那么交并比为0。以下分别以python和C语言代码实现交并比的计算。

python代码实现如下:

def iou(b1,b2):xmin1,ymin1,xmax1,ymax1=b1xmin2,ymin2,xmax2,ymax2=b2w1=xmax1-xmin1h1=ymax1-ymin1w2=xmax2-xmin2h2=ymax2-ymin2xmin_inter=max(xmin1,xmin2)ymin_inter=max(ymin1,ymin2)xmax_inter=min(xmax1,xmax2)ymax_inter=min(ymax1,ymax2)inter=(xmax_inter-xmin_inter)*(ymax_inter-ymin_inter)union=w1*h1+w2*h2-interif inter<=0:return 0else:return inter/union

C语言代码实现如下:

struct Box
{double xmin;double ymin;double xmax;double ymax;
};double iou(Box b1, Box b2)
{double w1 = b1.xmax - b1.xmin;double h1 = b1.ymax - b1.ymin;double w2 = b2.xmax - b2.xmin;double h2 = b2.ymax - b2.ymin;double xmin_inter,ymin_inter,xmax_inter,ymax_inter;xmin_inter = (b1.xmin > b2.xmin) ? b1.xmin : b2.xmin;ymin_inter = (b1.ymin > b2.ymin) ? b1.ymin : b2.ymin;xmax_inter = (b1.xmax < b2.xmax) ? b1.xmax : b2.xmax;ymax_inter = (b1.ymax < b2.ymax) ? b1.ymax : b2.ymax;double inter=(xmax_inter-xmin_inter)*(ymax_inter-ymin_inter);double un=w1*h1+w2*h2-inter;if(inter<=0)return 0;else return inter/un;
}

相关内容

热门资讯

安装系统组件失败 系统组件安装失败?一文教你排查与解决在电脑使用过程中,我们经常会遇到系统组件安装失败的问题。这不仅影...
白酒过滤系统安装,确保酒质纯净... 白酒过滤系统安装指南:确保酒质纯净的关键步骤随着白酒行业的不断发展,对酒质的要求越来越高。过滤系统作...
昂达双系统出现系统安装,昂达双... 昂达双系统安装常见问题及解决方法一、昂达双系统安装前的准备工作在安装昂达双系统之前,我们需要做好以下...
清华同方安装那个系统好,哪个系... 清华同方电脑安装系统推荐:哪个系统更适合您?Windows 10:功能全面,兼容性强Windows ...
北京管道新风系统安装,打造健康... 北京管道新风系统安装指南:打造健康呼吸环境随着城市化进程的加快,空气质量问题日益受到关注。安装管道新...
宝山区消防系统设备安装,宝山区... 宝山区消防系统设备安装,筑牢安全防线随着城市化进程的加快,消防安全问题日益凸显。宝山区作为上海市的重...
安装香氛系统教程图解,小蜜芬香... 安装香氛系统教程图解随着汽车科技的不断发展,越来越多的车主开始关注车内环境的舒适度。香氛系统作为一种...
吧台收银系统安装教程,轻松实现... 吧台收银系统安装教程:轻松实现高效点餐与收银随着餐饮行业的快速发展,吧台收银系统的应用越来越广泛。本...
按照步骤安装系统,电脑安装系统... 电脑安装系统步骤详解随着电脑使用时间的增长,系统可能会出现卡顿、运行缓慢等问题。这时,重装系统是一个...
安装信息查询系统简报 安装信息查询系统简报随着信息化时代的到来,安装信息查询系统在各个领域中的应用越来越广泛。本文将为您详...
斑马系统怎么安装应用,斑马系统... 斑马系统安装应用全攻略随着智能汽车的发展,斑马系统作为国内领先的智能车机系统,为广大车主提供了丰富的...
安卓系统安装在电脑上,轻松实现... 电脑安装安卓系统:轻松实现双系统体验一、准备工作在开始安装安卓系统之前,我们需要做一些准备工作。1....
把手机系统安装在盘里,手机系统... 手机系统安装到盘里的详细步骤随着智能手机的普及,越来越多的用户开始关注手机系统的安装和升级。将手机系...
安装系统怎样清除分区,全面指南 安装系统前如何清除分区:全面指南在安装操作系统之前,清除分区是一个重要的步骤,它可以帮助确保系统安装...
安装系统自带xps,电脑xps... 如何安装Windows 10系统自带的XPS查看器随着Windows 10系统的普及,许多用户可能会...
保密管理系统安装包,全方位数据... 保密管理系统安装包:全方位数据安全守护神随着信息化时代的到来,数据安全成为企业和机构关注的焦点。保密...
把证书安装到系统,数字证书下载... 如何将证书安装到系统随着信息技术的不断发展,证书在网络安全中扮演着越来越重要的角色。证书的安装对于确...
搬瓦工安装win系统,搬瓦工V... 搬瓦工VPS安装Windows系统的详细教程搬瓦工(BandwagonHost)是一家知名的VPS服...
班级必须安装监控系统吗,班级必... 班级必须安装监控系统吗?——探讨校园监控系统的利与弊随着社会的发展和科技的进步,校园监控系统在校园安...
宝安防盗报警系统安装,全方位守... 宝安防盗报警系统安装:全方位守护您的财产安全随着社会经济的发展,人们对家庭和企业的安全防范意识日益增...