交并比(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;
}

相关内容

热门资讯

彻底删除安卓系统更新,重拾纯净... 你有没有想过,你的安卓手机每次更新后,那些新功能、新界面,其实有时候并不是那么让人心动呢?有时候,我...
华为系统怎么打败安卓,揭秘如何... 你知道吗?在智能手机的世界里,最近可是掀起了一场不小的风波呢!那就是华为系统如何打败安卓系统,成为新...
安卓用户切换苹果系统,探索苹果... 你知道吗?最近在朋友圈里,有个话题可是热得不得了,那就是安卓用户纷纷切换到苹果系统。这究竟是怎么回事...
掌学派安卓系统密码,解锁便捷学... 亲爱的读者朋友们,你是否也有过这样的经历:手机里的掌学派安卓系统密码突然忘记了,心里那个急啊,就像热...
阿里云系统刷机安卓系统,轻松实... 你有没有想过,你的手机系统是不是也能像电脑一样,随时更新换代呢?今天,就让我带你一探究竟,看看阿里云...
安卓系统9.0会卡吗,流畅度解... 听说安卓系统9.0要来了,是不是又得担心手机卡成“龟速”啦?别急,今天咱们就来聊聊这个话题,看看安卓...
安卓系统平板图片分类,轻松浏览 你有没有发现,现在市面上平板电脑的种类真是五花八门,让人眼花缭乱?尤其是安卓系统的平板,更是层出不穷...
安卓系统查找位置软件,基于安卓... 手机不见了?别慌!安卓系统里藏着不少查找位置的小秘密,今天就来跟你聊聊这些神奇的小工具,让你的手机瞬...
安卓也能用ios系统,揭秘安卓... 哎呀呀,你知道吗?现在科技界可是掀起了一股热潮,那就是——安卓手机也能用iOS系统啦!是不是听起来有...
安卓系统如何清理文件,提升手机... 亲爱的手机,你是不是也觉得最近有点“胖”了呢?别担心,今天就来教你几招,让你的安卓系统文件轻松瘦身,...
安卓系统哪个平板好些,揭秘哪款... 你有没有想过,在这个科技飞速发展的时代,拥有一款性能出众的安卓平板,简直就像拥有了移动的办公室和娱乐...
安卓双系统进入动画,流畅体验与... 你知道吗?现在很多小伙伴都喜欢在自己的手机上玩转双系统,就像变魔术一样,一会儿是安卓,一会儿是iOS...
安卓系统无root权限,揭秘无... 亲爱的手机控们,你们有没有遇到过这样的情况:手机里装了各种软件,可有些功能却因为“无root权限”而...
安卓系统有区别吗,深度探讨不同... 你有没有想过,为什么你的手机里装的是安卓系统,而朋友的手机里却是苹果系统呢?这两种系统,它们之间到底...
安卓系统怎么录制唱歌,打造个人... 亲爱的音乐爱好者们,你是否曾在某个瞬间,想要记录下自己美妙的歌声,却又苦于找不到合适的方法?别担心,...
安卓系统需要安装什么,安卓系统... 你有没有想过,你的安卓手机里那些神奇的软件是怎么来的呢?没错,就是安卓系统在背后默默支持着它们。那么...
深圳苹果安卓系统下载,畅享智能... 你知道吗?最近有个游戏在朋友圈里火得一塌糊涂,那就是深圳城里3A字牌安卓版——魔Q游戏!这款游戏不仅...
苹果改安卓系统了,揭秘iOS系... 哇塞,你知道吗?苹果竟然改安卓系统了!是的,你没听错,就是那个一直以封闭著称的苹果,竟然开始向安卓系...
安卓手表更新不了系统,解锁系统... 最近是不是发现你的安卓手表更新系统时遇到了点小麻烦?别急,让我来给你支支招,让你轻松解决这个让人头疼...
海信电视 linux 安卓系统... 亲爱的读者们,你是否在寻找一款既时尚又实用的电视呢?今天,我要给你介绍一款在市场上备受好评的产品——...