《计算机视觉之目标检测》IOU详解及代码

本文介绍了目标检测中的重要概念IOU(交并比),解释了其计算方式和作用。IOU用于衡量检测框与真实框的重叠程度,常用于评估模型性能,如mAP指标。在Fasterrcnn等模型中,IOU也用于选择提案框。代码示例展示了如何用Python计算两个框的IOU,并提供了相应的函数实现。
摘要由CSDN通过智能技术生成

今天在改模型的时候刚好用到了IOU,因此将IOU说说,记录一下代码,方便以后复用。

1、什么是IOU

IoU 的全称为交并比(Intersection over Union),它的计算也比较简单,就是两个目标proposal框重叠的区域比总体的区域。如下图所表示的,着就是所谓的IOU。

2、为什么需要Iou?

在做目标检测的过程中有一个指标来评价这个模型的好坏,也就mAP。这个指标我下次再说,用检测出来的框和ground-truth之间计算IOU,如果IOU大于0.5或者其它的一个阈值,那么就认为该目标框被检测出来了。计算mAP的其中一步就需要用到IOU的计算值。比如在Faster rcnn这个模型中也需要用Iou去挑选proposal(128前景+128背景)

3、IOU如何计算

从上图中可以看出来,两个区域的交集是黄色区域,因此我们只需要计算出来黄色区域的面积areaC,A,B区域的面积分别是areaA,areaB。因此Iou=C/A+B-C。公式十分明了,我们只需要计算来各个部分的面积就可以了。

首先将IOU计算的输入,分别是A,B区域的坐标,以A

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值