交并比(IOU)的计算方法
创始人
2024-05-31 00:31:58
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;
}

相关内容

热门资讯

开源证券给予指南针买入评级,金...   炒股就看金麒麟分析师研报,权威,专业,及时,全面,助您挖掘潜力主题机会! 开源证券02月25日...
欧洲地下天然气储备库存降至41... 格隆汇2月25日|根据欧洲天然气基础设施协会数据,目前欧洲地下天然气储气设施的天然气库存降至41%以...
阳春三月 上海帆船公开赛展现活... 阳春三月,2025上海帆船公开赛,将为申城黄浦江和滴水湖带来群帆竞发的活力。今天赛事组委会宣布,比赛...
黑芝麻:2月25日召开董事会会... 黑芝麻(SZ 000716,收盘价:7.13元)2月25日晚间发布公告称,公司第十一届2025年第三...
经纬恒润:2024年亏损5.0... 中证智能财讯经纬恒润(688326)2月25日晚间披露2024年业绩快报,公司实现营业收入55.41...
日本大阪世博会门票销售低迷 石... 中新网2月25日电 据法新社消息,当地时间25日,由于计划于4月13日开幕的大阪世博会门票销售疲软,...
江苏金租遭第五大股东法巴租赁首...   来源:财联社  2月24 日晚,江苏金租公告外方股东法巴租赁将减持部分股份引发市场关注。据了解,...
黑龙江牡丹江:专项检查出版物市... 中国消费者报哈尔滨讯(记者刘传江)为进一步规范出版物市场秩序、严厉打击非法出版物,近日,黑龙江省牡丹...
我终于知道一元广东红中癞子麻将... 微【ab120590】 【mj120590】【hf420624】(广东一元一分红中癞子爆炸码麻将群)...
为您揭秘一元一分红中麻将的加我... 群主微【ab120590】 【mj120590】【hf420624】免带押进群,群内跑包包赔支持验证...
中国移动2月25日现2笔大宗交...   炒股就看金麒麟分析师研报,权威,专业,及时,全面,助您挖掘潜力主题机会!   2月25日,中国...
珠海冠宇2024年度业绩快报:... 2月25日,珠海冠宇(688772)发布公告,2024年度业绩快报显示,公司实现营业总收入115.4...
研报掘金丨中金:维持同花顺“跑... 中金公司研报指出,同花顺(300033.SZ)2024年净利润同比+30%至18.2 亿元,其中4Q...
农业银行2月25日现1笔大宗交...   炒股就看金麒麟分析师研报,权威,专业,及时,全面,助您挖掘潜力主题机会!   2月25日,农业...
科恒股份(300340.SZ)... 科恒股份(300340.SZ)公告,公司与北京石墨烯研究院有限公司(简称“北京石墨烯研究院”)于近日...
24小时在线24小时上下分红中... 微【ab120590】 【mj120590】【hf420624】等风也等你。喜欢打麻将的兄弟姐妹们、...
为您揭秘一元一分红中麻将群@2... 认准管理加v:微【ab120590】【mj120590】【hf420624】七年稳定老群!随时拿起手...
苯乙烯:估值下移,矛盾仍在积累 .ct_hqimg {margin: 10px 0;} .hqimg_wrapper {text-a...
今日推荐加入附近红中麻将群贴吧... 微【ab120590】 【mj120590】【hf420624】(广东一元一分红中癞子爆炸码麻将群)...
春日限定 “头茬鲜”,竟是温补... 转自:佛慈制药春回大地,万物复苏,此时此刻,饮食养生也有着独特的讲究。有一种春日限定的 “头茬鲜”,...