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

相关内容

热门资讯

系统如何与安卓互通,技术融合与... 你有没有想过,你的手机系统竟然能和安卓系统这么默契地互通有无?这就像是一场跨越科技界的友谊赛,让我们...
安卓系统 扫码枪,安卓系统下扫... 你有没有想过,在繁忙的超市收银台,那些快速流畅的扫码操作,背后其实隐藏着一个小小的英雄——安卓系统扫...
平板插卡推荐安卓系统,安卓系统... 你有没有想过,你的平板电脑是不是也能像智能手机一样,随时随地扩充存储空间呢?没错,这就是今天我要跟你...
安卓系统固件安装失败,原因排查... 最近是不是你也遇到了安卓系统固件安装失败的问题?别急,让我来给你详细说说这个让人头疼的小麻烦,让你一...
ios系统和安卓区别,系统差异... 你有没有发现,现在手机市场上,iOS系统和安卓系统就像是一对双胞胎,长得差不多,但性格却截然不同。今...
安卓系统2.3优酷,优酷的崛起... 你有没有发现,安卓系统2.3时代的那股怀旧风?那时候,优酷可是视频界的巨头,多少人都是看着优酷长大的...
安卓导航系统密封,安卓导航系统... 你有没有发现,现在手机导航系统越来越智能了?尤其是安卓系统的导航,简直就像一个贴心的导航小助手,带你...
a版安卓11系统,a版深度解析... 你知道吗?最近手机界可是炸开了锅,各大品牌纷纷发布了搭载a版安卓11系统的手机。这可不是什么小打小闹...
安卓系统的模拟吉他,随时随地弹... 你有没有想过,在手机上也能弹奏吉他呢?没错,就是那种模拟吉他的安卓系统应用,让你随时随地都能享受音乐...
王者适配的安卓系统,深度解析适... 你有没有发现,最近玩《王者荣耀》的小伙伴们都在议论纷纷,说新出的安卓系统简直是为王者量身定做的!没错...
安卓系统自动定位关闭,隐私保护... 你有没有发现,手机里的安卓系统有时候会自动定位,这可真是让人又爱又恨啊!有时候,我们并不想让别人知道...
安卓系统电量耗尽测试,全面解析... 手机电量耗尽,这可是每个手机用户都头疼的问题。你有没有想过,你的安卓手机在电量耗尽前,到底经历了哪些...
如何升级车载安卓系统,车载安卓... 亲爱的车主朋友们,你是不是也和我一样,对车载安卓系统升级这件事充满了好奇和期待呢?想象当你驾驶着爱车...
安卓办公哪个系统好,深度解析哪... 你有没有想过,在安卓办公的世界里,哪个系统才是你的最佳拍档呢?在这个信息爆炸的时代,选择一个既强大又...
安卓系统差劲怎么解决,重拾流畅... 你有没有发现,安卓系统有时候真的让人头疼得要命?手机卡顿、应用崩溃、电池续航短,这些问题是不是让你抓...
喜欢安卓系统的原因,探索用户偏... 你有没有发现,身边的朋友、同事,甚至家人,越来越多的人开始使用安卓手机了呢?这可不是简单的潮流,而是...
安卓系统金立手机,品质生活新选... 你有没有发现,最近安卓系统下的金立手机突然火了起来?没错,就是那个曾经陪伴我们走过无数时光的金立手机...
无安卓系统的电视,新型无系统电... 亲爱的读者们,你是否厌倦了那些充斥着安卓系统的电视?想要尝试一些新鲜玩意儿?那就跟我一起探索一下无安...
麒麟系统能刷安卓系统吗,轻松刷... 你有没有想过,你的麒麟手机能不能装上安卓系统呢?这可是个让人好奇不已的问题。现在,就让我来带你一探究...
手机公司安卓系统吗,手机公司引... 你有没有想过,为什么你的手机里装的是安卓系统而不是苹果的iOS呢?这背后可是有着不少故事和门道的哦!...