论文标题
PIOU损失:在复杂环境中迈向准确的定向对象检测
PIoU Loss: Towards Accurate Oriented Object Detection in Complex Environments
论文作者
论文摘要
使用定向边界框(OBB)进行对象检测可以通过减少与背景区域的重叠来更好地靶向旋转对象。现有的OBB方法主要是通过引入距离损失优化的额外角度维度来建立在水平边界盒探测器上的。但是,由于距离损耗仅使OBB的角度误差最小化,并且它与IOU松散相关,因此它对高纵横比的对象不敏感。因此,制定了一种新颖的损失,像素-IOU(PIOU)损失,以利用角度和IOU以进行准确的OBB回归。 PIOU损失源自具有像素形式的IOU度量,该度量很简单,适合水平和定向边界框。为了证明其有效性,我们评估了基于锚点和无锚固框架的PIOU损失。实验结果表明,PIOU损失可以显着改善OBB检测器的性能,尤其是在具有较高纵横比和复杂背景的物体上。此外,以前的评估数据集不包括对象具有较高纵横比的方案,因此引入了一个新的数据集Retail50k,以鼓励社区适应更复杂的环境。
Object detection using an oriented bounding box (OBB) can better target rotated objects by reducing the overlap with background areas. Existing OBB approaches are mostly built on horizontal bounding box detectors by introducing an additional angle dimension optimized by a distance loss. However, as the distance loss only minimizes the angle error of the OBB and that it loosely correlates to the IoU, it is insensitive to objects with high aspect ratios. Therefore, a novel loss, Pixels-IoU (PIoU) Loss, is formulated to exploit both the angle and IoU for accurate OBB regression. The PIoU loss is derived from IoU metric with a pixel-wise form, which is simple and suitable for both horizontal and oriented bounding box. To demonstrate its effectiveness, we evaluate the PIoU loss on both anchor-based and anchor-free frameworks. The experimental results show that PIoU loss can dramatically improve the performance of OBB detectors, particularly on objects with high aspect ratios and complex backgrounds. Besides, previous evaluation datasets did not include scenarios where the objects have high aspect ratios, hence a new dataset, Retail50K, is introduced to encourage the community to adapt OBB detectors for more complex environments.