Drones, Vol. 10, Pages 20: A Lightweight Multi-Module Collaborative Optimization Framework for Detecting Small Unmanned Aerial Vehicles in Anti-Unmanned Aerial Vehicle Systems

Drones doi: 10.3390/drones10010020

Authors:
Zhiling Chen
Kuangang Fan
Jingzhen Ye
Zhitao Xu
Yupeng Wei

In response to the safety threats posed by unauthorized unmanned aerial vehicles (UAVs), the importance of anti-UAV systems is becoming increasingly apparent. In tasks involving UAV detection, small UAVs are particularly difficult to detect due to their low resolution. Therefore, this study proposed YOLO-CoOp, a lightweight multi-module collaborative optimization framework for detecting small UAVs. First, a high-resolution feature pyramid network (HRFPN) was proposed to retain more spatial information of small UAVs. Second, a C3k2-WT module integrated with wavelet transform convolution was proposed to enhance feature extraction capability and expand the model’s receptive field. Then, a spatial-channel synergistic attention (SCSA) mechanism was introduced to integrate spatial and channel information and enhance feature fusion. Finally, the DyATF method replaced the upsampling with Dysample and the confidence loss with adaptive threshold focal loss (ATFL), aiming to restore UAV details and balance positive–negative sample weights. The ablation experiments show that YOLO-CoOp achieves 94.3% precision, 93.1% recall, 96.2% mAP50, and 57.6% mAP50−95 on the UAV-SOD dataset, with improvements of 3.6%, 10%, 5.9%, and 5% over the baseline model, respectively. The comparison experiments demonstrate that YOLO-CoOp has fewer parameters while maintaining superior detection performance. Cross-dataset validation experiments also demonstrate that YOLO-CoOp exhibits significant performance improvements in small object detection tasks.

Source link

Zhiling Chen www.mdpi.com