交并比(IOU)的计算方法

news/2024/11/7 14:41:42/

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


http://www.ppmy.cn/news/30883.html

相关文章

运维视角:rabbitmq教程(三)镜像集群

上期回顾 RabbitMQ集群中节点包括内存节点、磁盘节点。内存节点就是将所有数据放在内存&#xff0c;磁盘节点将数据放在磁盘上。如果在投递消息时&#xff0c;打开了消息的持久化&#xff0c;那么即使是内存节点&#xff0c;数据还是安全的放在磁盘。那么内存节点的性能只能体现…

字节3次都没裁掉的7年老测试。掌握设计业务与技术方案,打开上升通道!

前言职场中的那些魔幻操作&#xff0c;研发最烦的是哪个&#xff1f;“面对业务需求的时候&#xff0c;可能都听过这样一句话&#xff1a;这个很简单&#xff0c;直接开发&#xff0c;三天内上线&#xff1b;”朋友说&#xff1a;“产品听了流泪&#xff0c;测试见了崩溃&#…

博弈论小课堂:非零和博弈(实现双赢)【纳什均衡点】

文章目录 引言I 非零和博弈1.1 囚徒问题1.2 博弈中双方的收益矩阵II 在现实中找均衡点2.1 博弈通常不是一次性的,而是反复进行的2.2 博弈论讲的都是阳谋的策略2.3 人类还处于文明的初级阶段,人的道德水准不容高估2.4 乌合之众效应2.5 很多时候看似是双赢,其实是在更大范围内…

2023 最新可用Google镜像地址 长期更新

Google镜像说明 由于种种原因&#xff0c;国家还未开放Google搜索的使用。虽然可以通过某些技术手段实现访问&#xff0c;但是还是有一些同学需要借助Google搜索镜像才可以达到访问的目的&#xff1b;笔者特意搜集了一些2022年最新的Google搜索镜像供有需求的童鞋使用&#xf…

微信聊天的一个创新方向

开门见山&#xff0c;简单&#xff08;简陋&#xff09;展示下新的聊天界面&#xff1a; 注意到除了原本的发送键&#xff0c;多了几个别的按钮。为了对比方便&#xff0c;先放上当前的聊天方式&#xff1a; 给兄弟发消息时&#xff0c;点击发送键&#xff0c;显示的是如下发…

如何通过DAS连接GaussDB

文章目录1 实验介绍2 实验目的3 配置DAS服务4 SQL使用入门1 实验介绍 本实验主要描述如何通过华为云数据管理服务 (Data Admin Service&#xff0c;简称DAS) 来连接华为云GaussDB数据库实例&#xff0c;DAS是一款专业的简化数据库管理工具&#xff0c;提供优质的可视化操作界面…

10.系统级I/O

1.基础所有的I/O设备被模型化为文件&#xff0c;所有的输入和输出被当作相应文件的读和写来执行应用程序在文件结尾检测到EOF(end of file)条件文本文件是只含有ASCII或Unicode字符的普通文件二进制文件是所有的其他文件对于内核&#xff0c;文本文件和二进制文件没有区别目录是…

移动OA系统,联动企业协作让办公高效无间断

随着信息技术的不断发展&#xff0c;企业管理也逐渐从传统的纸质管理转向数字化管理。移动OA系统作为数字化管理的重要工具&#xff0c;被越来越多的企业所采用。移动OA系统不仅可以提高企业管理的效率和准确性&#xff0c;而且还能实现多人协作和无间断工作。 移动OA系统是一…