Enhanced YOLOv11-Based River Aerial Image Detection Research
Lei Zhang, Ao Zheng, Xiaoyan Sun, Zhipeng Sun
Abstract
The UAV encounters challenges in detecting similar small targets during target detection tasks. Consequently, current target detection algorithms struggle to accurately identify river debris, overgrazing, and suspected sand mining activities. To address the issues of low precision and high complexity associated with small target detection in existing models, this paper introduces an enhanced version of YOLOv11, referred to as PAB-YOLOv11. Firstly, the C3K2-PPA module is employed to replace the C3K2 module within the backbone network. Additionally, a multi-branch fusion approach is utilized to enhance the model’s feature extraction capabilities for small targets across various scales. The AFGC attention mechanism is integrated between the neck network and the detection head to improve the recognition of similar objects. This is achieved by emphasizing local fine features and dynamically adjusting the distribution of attention. The experimental results demonstrate that, on the dataset obtained from the Sanggan River basin, the [email protected] of PAB-YOLOv11 reaches 64.9%, reflecting an improvement of 2.1% over the original YOLOv11 model. Compared to the three mainstream models, YOLOv5s, YOLOv8s, and YOLOv11n, PAB-YOLOv11 achieves improvements of 3.1%, 3.2%, and 2.6% in [email protected], respectively. When compared to more advanced models such as RT-DETR and DINO, PAB-YOLOv11 also shows enhancements in [email protected] of 5.1% and 2.8%, respectively. These findings indicate that the PAB-YOLOv11 model proposed in this study is an effective method for river channel inspection.