Cross-YOLO: an object detection algorithm for UAV based on improved YOLOv8 model
摘要
In this study, Cross-YOLO, an enhanced version of the YOLOv8 model, is specifically designed to address the challenge of detecting small objects in UAV target detection scenarios. The model refines the original YOLOv8 through several innovative improvements: Firstly, in order to improve the detection accuracy of small targets, we propose Cross-FPN to bolster the original FPN. Secondly, we have redesigned a lightweight detection head, DELDH, to solve the problem of network bloat caused by the introduction of small object detection heads. Thirdly, a new attention mechanism CMCA is designed, that unifies the Coordinate attention mechanism with the Multi-scale convolutional attention mechanism to further enhance the feature extraction of small targets. Finally, the WIoU loss function is introduced to improve the accuracy of bounding box regression and improve detection performance. Experimental data in the Visdrone dataset indicate that, under the condition of the selected model size n, Cross-YOLO achieves a substantial reduction of 35.4% in parameter count compared to YOLOv8n, with only a marginal increase of 7.8% in computational load, and a significant improvement of 5.3% in