PIoU Loss: Towards Accurate Oriented Object Detection in Complex Environments
一种面向复杂环境下的精确旋转目标检测的IOU损失计算方法。
Zhiming Chen, Kean Chen, Weiyao Lin, John See,Hui Yu, Yan Ke, and Cong Yang
Clobotics, China Department of Electronic Engineering, Shanghai Jiao Tong University, ChinaFaculty of Computing and Informatics, Multimedia University,Malaysia
发表于:ECCV 2020
- 论文地址:https://arxiv.org/abs/2007.09584
- 论文下载:https://arxiv.org/pdf/2007.09584
- 论文代码:https://github.com/clobotics/piou
一. 研究动机
目标检测是计算机视觉中的一项基本任务,最近的检测器大都是使用卷积神经网络的。尽管这些检测器都有最先进的性能,但是在旋转目标检测中,物体往往有以下特征 (1) 物体往往都很小(2)物体很密集 (3)物体的方向是任意的 。传统的目标检测中的边界框(BB)在检测这类物体时通常会有大量的背景,这会一定程度的误导分类器。当边界框内物体很密集或者有重叠时又会使检测器难以分离这些物体。为了解决这种缺陷,研究者们提出了定向边界框(OBB)。BB通常的表现形式(Cx,Cy,w,h),与此相比,OBB由(Cx,Cy,w,h,θ)组成。因此,OBB可以更好的包围目标以便于旋转和密集的物体可以被更好的检测和分类。
使用定向边界框(OBB)的目标检测可以通过减少目标与背景的重叠更好的的预测旋转的物体。现有的OBB方法大多建立在水平边界框检测器的基础上,引入了一个由距离损失优化的额外角度维度。但是,距离损失仅仅最小化了OBB的角度损失,它与IOU相关性不够好(目标检测的评判标准)