今天在改模型的时候刚好用到了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