MSD-YOLO11n: an improved small target detection model for high precision UAV aerial imagery
Xin Zhang, Baoyang Du, Yaoxing Jia, Kairui Luo, Li Jiang
Abstract
Abstract The capacity for precise target detection in UAV aerial images represents a pivotal prerequisite for the advancement of low-altitude economic activities. The technology for large target detection in images has attained a state of relative maturity, while the identification of small targets remains encumbered by challenges such as indistinct edge information, the absence of comprehensive deep feature information, and the constrained capacity for the expression of detection head information. To address these challenges, the MSD-YOLO11n target detection model has been proposed. Firstly, the P2 layer downsampling convolution is adopted as SPDConv, while the Multiscale Edge Information Selection (MEIS) module is proposed to replace the residual module in C3K2 in order to enhance the edge information feature extraction of small targets. Secondly, the Smalltarget Feature Enhancement Pyramid (SFEP) module has been proposed as a means of achieving the fusion of feature information between the P2 and P3 layers. This is achieved by combining Dysample up sampling with SPDConv down sampling and reconstructing low-quality images to obtain high-quality images using the CSP-OmniKernel module. Subsequently, the DyHead module was utilised to integrate scale, space and task-aware attention. Finally, the proposed MSD-YOLO11n model was evaluated based on the VisDrone2019 dataset through a combination of ablation and comparison experiments. In comparison with YOLO11n, both $${mAP}_{0.5}^{val}$$ <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"> <mml:msubsup> <mml:mrow> <mml:mi>mAP</mml:mi> </mml:mrow> <mml:mrow> <mml:mn>0.5</mml:mn> </mml:mrow> <mml:mrow> <mml:mi>val</mml:mi> </mml:mrow> </mml:msubsup> </mml:math> and $${mAP}_{0.5}^{test}$$ <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"> <mml:msubsup> <mml:mrow> <mml:mi>mAP</mml:mi> </mml:mrow> <mml:mrow> <mml:mn>0.5</mml:mn> </mml:mrow> <mml:mrow> <mml:mi>test</mml:mi> </mml:mrow> </mml:msubsup> </mml:math> demonstrate enhancements of 20.9% and 19.7%, respectively. Furthermore, the parameters $$A{P}_{vt}$$ <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"> <mml:mrow> <mml:mi>A</mml:mi> <mml:msub> <mml:mi>P</mml:mi> <mml:mrow> <mml:mi>vt</mml:mi> </mml:mrow> </mml:msub> </mml:mrow> </mml:math> , $$A{P}_{t}$$ <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"> <mml:mrow> <mml:mi>A</mml:mi> <mml:msub> <mml:mi>P</mml:mi> <mml:mi>t</mml:mi> </mml:msub> </mml:mrow> </mml:math> and $$A{P}_{s}$$ <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"> <mml:mrow> <mml:mi>A</mml:mi> <mml:msub> <mml:mi>P</mml:mi> <mml:mi>s</mml:mi> </mml:msub> </mml:mrow> </mml:math> exhibit improvements of 44.4%, 26.9%, and 37.3%, respectively, for small targets. Furthermore, experimental evidence demonstrates that the MSD-YOLO11n model attains superior target detection accuracy compared to the TA-YOLO-n model, despite utilising a significantly reduced parameter count of 3.6 M. The efficacy of the proposed MSD-YOLO11n model in target detection tasks involving UAV aerial images has been substantiated.