Yamei Zhang, Keding Yan
To address the recognition challenges caused by the high proportion, low resolution, and significant multi-scale variations of small objects in UAV small-object detection tasks, a UAV small-object detection and recognition algorithm based on CEF-YOLOv8n is proposed. The proposed algorithm uses YOLOv8n as the baseline network and introduces a Partial Convolution-based Cross Partial Feature (CPF) module into the backbone network to enhance the local feature extraction capability for low-resolution small objects. In the neck network, the concept of feature focusing and diffusion is adopted to construct a Focusing Generalized Feature Pyramid Network (FGFPN). A Feature Semantic Fusion Module (FSFM) based on a cross-attention mechanism is designed to complementarily fuse shallow detail features with deep semantic features, thereby enhancing information interaction among objects at different scales. In addition, a Lightweight Weight-Sharing Detection Head (LWSD) is proposed to improve the computational efficiency and real-time performance of the model while maintaining detection accuracy. Publicly available datasets are used for network training and detection evaluation, and comparative experiments are conducted with other algorithms. The results show that the proposed detection and recognition algorithm achieves 37.6% and 22.6% in terms of mAP50 and mAP50-95, respectively, representing improvements of 3.4 and 2.6 percentage points over the original YOLOv8n. Meanwhile, the number of parameters and FLOPs are reduced from 3.2 M and 8.7 G to 2.5 M and 6.9 G, respectively.