论文标题:《Shape-Adaptive Selection and Measurement for Oriented Object Detection》
论文发表:AAAI 2022 论文链接:https://ojs.aaai.org/index.php/AAAI/article/view/19975 论文代码:https://github.com/houliping/SASM
@inproceedings{hou2022shape,
title={Shape-adaptive selection and measurement for oriented object detection},
author={Hou, Liping and Lu, Ke and Xue, Jian and Li, Yuqiu},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={36},
number={1},
pages={923--932},
year={2022}
}
Name | Value |
---|---|
标签 | #标签分配 #旋转目标检测 |
数据集 | #DOTA #HRSC2016 #UCAS-AOD #ICDAR2015 |
目的 | 解决旋转目标检测中样本选择没有考虑目标形状信息、没有区分不同质量正样本的问题 |
方法 | 提出shape-adaptive selection以及shape-adaptive measurement |
作者提到旋转目标检测仍然面临挑战,其中最主要的挑战来自目标的形状(如长宽比)。在通用目标检测任务中,样本选择(sample selection,也叫标签分配,label assignment)对于性能提升具有重要作用。然而现有的样本选择策略存在以下不足:
针对上述问题,作者提出了shape-adaptive selection(SA-S)和shape-adaptive measurement(SA-M)策略。
作者做了一个实验,分析不同正负样本IoU划分阈值对不同长宽比类别性能的影响,可以发现当长宽比变大时,IoU阈值越小越好,这可以说明传统的固定IoU阈值的划分方式并不是最优的。
根据ATSS,目标真值和预测值的IoU均值和方法可以用于计算动态的IoU阈值,因此对于第i个真值框,IoU阈值可以计算为:
其中:
而
代表真值框的长宽比。 根据上面的分析,权重应该随着纵横比的增加而减小,因此长宽比较大的目标被分配一个较低的IoU阈值,故而:
其中ω是一个权重参数,默认为4。当数据集包含大量大长宽比目标时,较大的ω通常可以获得更好的性能。这样,当IoU大于等于\tau_i^{IoU}时,认为其是正样本。
作者认为与位于物体内部的点相比,位于物体边界附近的点包含更多关于杂波背景,甚至附近物体的信息。因此,位于物体内部的点,特别是位于物体中心周围的点,比位于物体边界附近的点更能代表物体的特征。因此如果用所有正样本都有同样的权重会导致一些高质量正样本被远离物体中心的低质量样本点抑制,且每个样本点的质量与物体的形状密切相关,而不仅仅与每个点到物体中心的距离有关。
,其分别代表中心点坐标,宽度,高度和角度。归一化形状距离计算如下:
作者认为较大偏差的孤立点会极大地影响凸包(convex hull,其由预测点集计算得到)的质量,并对精确定位产生不利影响,因此提出了边界中心损失(Boundary-Center Loss),其从点集中选取最左点、最右点、最上点和最下点,用点集中所有点的x坐标和y坐标的平均值来计算平均中心点:
因此总损失为:
是权重系数,根据经验设为1.0,0.375和1.0. 其中分类损失计算为:
代表focal loss。初始检测损失为:
代表GIoU Loss。精炼阶段检测损失为: