论文标题

RFLA:高斯接受场的标签分配,用于微小对象检测

RFLA: Gaussian Receptive Field based Label Assignment for Tiny Object Detection

论文作者

Xu, Chang, Wang, Jinwang, Yang, Wen, Yu, Huai, Yu, Lei, Xia, Gui-Song

论文摘要

检测小物体是阻碍对象检测开发的主要障碍之一。通用对象检测器的性能在微小的对象检测任务上往往会大大恶化。在本文中,我们指出的是,基于锚的检测器中的先验盒或无锚检测器中的点是小物体的优化。我们的主要观察结果是,目前的基于锚或无锚的标签分配范例将引起许多离群小的小地面真相样本,从而导致检测器对小物体的关注较少。为此,我们提出了一个基于高斯接受场的标签分配(RFLA)策略,以进行微小的对象检测。具体而言,RFLA首先利用了特征接受场遵循高斯分布的先前信息。然后,提出了一个新的接受场距离(RFD),而不是通过IOU或中心采样策略分配样品,以直接测量高斯接受场和地面真相之间的相似性。考虑到基于阈值的和中心的采样策略偏向大型对象,我们进一步设计了基于RFD的层次标签分配(HLA)模块,以实现微小对象的平衡学习。在四个数据集上进行的广泛实验证明了所提出的方法的有效性。尤其是,我们的方法在AI-TOD数据集上以4.0 AP点优于最先进的竞争对手。代码可从https://github.com/chasel-tsui/mmdet-rfla获得

Detecting tiny objects is one of the main obstacles hindering the development of object detection. The performance of generic object detectors tends to drastically deteriorate on tiny object detection tasks. In this paper, we point out that either box prior in the anchor-based detector or point prior in the anchor-free detector is sub-optimal for tiny objects. Our key observation is that the current anchor-based or anchor-free label assignment paradigms will incur many outlier tiny-sized ground truth samples, leading to detectors imposing less focus on the tiny objects. To this end, we propose a Gaussian Receptive Field based Label Assignment (RFLA) strategy for tiny object detection. Specifically, RFLA first utilizes the prior information that the feature receptive field follows Gaussian distribution. Then, instead of assigning samples with IoU or center sampling strategy, a new Receptive Field Distance (RFD) is proposed to directly measure the similarity between the Gaussian receptive field and ground truth. Considering that the IoU-threshold based and center sampling strategy are skewed to large objects, we further design a Hierarchical Label Assignment (HLA) module based on RFD to achieve balanced learning for tiny objects. Extensive experiments on four datasets demonstrate the effectiveness of the proposed methods. Especially, our approach outperforms the state-of-the-art competitors with 4.0 AP points on the AI-TOD dataset. Codes are available at https://github.com/Chasel-Tsui/mmdet-rfla

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源