返回
Label Assignment Matters: A Gaussian Assignment Strategy for Tiny Object Detection
DOI:10.1109/TGRS.2024.3430071.png)
摘要
En 中文
Recently, impressive improvements have been achieved in general object detection. However, tiny object detection remains a very challenging problem since tiny objects only occupy a few pixels. Consequently, the label assignment strategies used in general object detectors are not suitable for tiny object detection, because these algorithms tend to assign few or even no positive samples for tiny objects. In this article, we propose a simple yet effective Gaussian assignment (GA) strategy to solve this problem. Specifically, we first model the bounding boxes as 2-D Gaussian distributions and then encode training samples with a threshold. This strategy can assign more high-quality positive samples for tiny objects and adjust the weight of positive samples to balance the contribution from different-size objects. Extensive experiments on four tiny object detection datasets show that the proposed strategy significantly and consistently improves the performance of single-stage tiny object detectors. In particular, with our strategy, we bridge the performance gap between single-stage and state-of-the-art multistage detectors on the AI-TOD dataset (24.2% versus 24.8% in mAP) while maintaining the inference speed. The code is available at https://github.com/zf020114/GaussianAssignment.
Keyword:
Detectors
Geoscience and remote sensing
Gaussian distribution
Annotations
Object detection
Heating systems
Feature extraction
Deep convolution neural networks
Gaussian distributions
tiny object detection
期刊
IF:
8.6
论文数:
2.2W
被引数:
10.7W
机构
引用论文
Conductivity Enhancement in Thin Silicon-on-Insulator Layer Embedding Artificial Dislocation Network
Inflexibility of mental planning: A characteristic disorder with prefrontal lobe lesions?心理计划的僵化: 前额叶病变的特征性障碍?
Multicomponent Orbital-Optimized Perturbation Theory with Density Fitting: Anharmonic Zero-Point Energies in Protonated Water Clusters具有密度拟合的多组分轨道优化摄动理论: 质子化水团簇中的非谐零点能量
A General Gaussian Heatmap Label Assignment for Arbitrary-Oriented Object Detection面向任意对象检测的通用高斯热图标签分配

