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

相关内容

热门资讯

安卓系统刷miui系统怎么样,... 你有没有想过,给你的安卓手机换换口味呢?没错,就是刷机!今天,咱们就来聊聊安卓系统刷MIUI系统怎么...
洛基亚为啥不用安卓系统,探寻其... 你有没有想过,为什么洛基亚(Lokia)手机不用安卓系统呢?是不是觉得这个品牌有点神秘,有点特别?今...
安卓系统与小米miu系统哪个好... 你有没有想过,手机里的操作系统就像是我们的大脑,它决定了我们手机的使用体验。今天,咱们就来聊聊这个话...
安卓怎么苹果系统锁屏,揭秘系统... 你是不是也和我一样,对安卓手机和苹果手机的系统锁屏功能充满了好奇呢?想知道它们各自有什么特别的地方,...
苹果安卓系统软件互通,共创未来... 你知道吗?在这个科技飞速发展的时代,手机已经成为了我们生活中不可或缺的一部分。而说到手机,不得不提的...
天籁系统的安卓怎么更新 亲爱的安卓用户们,你是否也和我一样,对天籁系统的更新充满了期待呢?每次系统更新,都仿佛是给我们的手机...
手机安卓木马系统高版本,技术演... 手机安卓木马系统:高版本的潜在风险与防范策略在数字化时代,智能手机已成为我们生活中不可或缺的一部分。...
小米用的是安卓系统,引领智能生... 你知道吗?现在手机市场上,各种品牌琳琅满目,让人挑花了眼。但是,说到性价比之王,那可非小米莫属了。你...
小游戏安卓换装系统下载,打造个... 你有没有发现,手机里的小游戏越来越好玩了?这不,最近我发现了一个超级有趣的小游戏——安卓换装系统下载...
鸿蒙系统融合安卓怎么用,跨平台... 你知道吗?最近手机圈里可是热闹非凡呢!华为的鸿蒙系统终于和安卓系统来了个亲密接触,这可是个让人眼前一...
安卓10系统为什么老化,揭秘老... 你有没有发现,你的安卓手机自从升级到安卓10系统后,好像变得有点“老态龙钟”了呢?别急,让我来给你细...
安卓sdk和windows系统... 你有没有想过,为什么你的安卓手机上的应用总是那么酷炫?这其中,安卓SDK可是功不可没的大功臣呢!今天...
王者荣耀系统安卓转苹果,轻松跨... 你有没有想过,从安卓手机转到苹果手机,王者荣耀的游戏体验会有哪些变化呢?这可是不少玩家关心的问题哦!...
安卓中高端系统手机推荐,性能与... 你有没有发现,最近手机圈里可是热闹非凡呢!各种新机发布,让人眼花缭乱。不过,咱们今天不聊那些入门级的...
安卓系统有个人收藏相册 你有没有发现,手机里的照片越来越多,有时候找起来就像大海捞针一样?别急,今天就来给你揭秘安卓系统里的...
如果美国封锁安卓系统,全球科技... 如果美国封锁安卓系统在数字化时代,智能手机已经成为我们生活中不可或缺的一部分。它不仅仅是一个通信工具...
安卓机器人系统人形,智能与美学... 你知道吗?在科技飞速发展的今天,有一种神奇的玩意儿正在悄悄改变我们的生活。它不是科幻电影里的机器人,...
魅族note安卓5.0系统,性... 你有没有听说过魅族Note这款手机呢?它可是最近在手机圈里小火了一把哦!今天,我就要来和你聊聊这款手...
安卓x9系统 你有没有发现,最近你的安卓手机好像变得不一样了?没错,就是那个神秘的安卓x9系统!今天,就让我带你一...
安卓7.0系统还会卡,安卓7.... 你有没有发现,尽管安卓系统更新换代了好几轮,但有些老手机还是时不时地卡个不停呢?这不,最近就有小伙伴...