首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

PIoU Loss: 实现复杂场景下的精确定向目标检测

使用定向包围框(oriented bounding box)进行目标检测可以通过减少与背景区域的重叠来更好地定位有旋转倾斜的目标。现有的OBB方法大多是在水平包围框检测器(horizontal bounding box)上通过引入额外的角度尺度(通过距离损失进行优化)构建的。 但是,由于距离损失只将OBB的角度误差优化至最小,而且与IoU的相关性较松散,因此它对具有高长宽比的目标不敏感。 因此,本文提出了一种新的损失,即Pixels-IoU(PIoU)损失,以利用角度和IoU实现更精确的OBB回归。 PIoU损失是从IoU指标以像素的形式导出的,形式简单但适用于水平和定向包围框。为了证明其有效性,本文评估了ancho-based和anchor-free框架下使用PIoU损失的效果。实验结果表明,PIoU损失可以显著提高OBB检测器的性能,特别是在具有高长宽比和复杂背景的目标检测上。此外,现有的评估数据集不含有大量高长宽比目标,因此引入了新的数据集Retail50K,以鼓励大家应用OBB检测器来处理更加复杂的环境。

01
领券