DTSSNet: Dynamic Training Sample Selection Network for UAV Object Detection

Object detectors often struggle with accuracy and generalization when applied to aerial imagery, primarily due to the following challenges. 1) great scale variation of objects in aerial images: both extremely small and large objects are visible in the same image; 2) an extreme imbalance of the training sample between positive and negative anchors: there are several positive ground truth anchors and an abundance of negative anchors. In this paper, we propose a Dynamic Training Sample Selection Network (DTSSNet) to solve above-mentioned problems in two dimensions. An Attention Enhanced Feature Module (AEFM) is proposed to enhance the basic features by focusing on both channel and semantic information related to targets. This module provides more valuable information for accurately classifying objects of different scales. To tackle the imbalance in training samples, this paper implements a Dynamic Training Sample Selection (DTSS) module that divides the training samples based on ground truth information. This module dynamically selects samples, ensuring a more balanced representation of positive and negative anchors, leading to improved learning. Importantly, the combination of AEFM and DTSS does not introduce any additional computational costs. Experimental evaluations on the VisDrone2019-DET dataset demonstrate that DTSSNet outperforms base detectors and generic approaches. Furthermore, the effectiveness of DTSSNet is validated on the UAVDT benchmark dataset, where it achieves state-of-the-art performance.