Small target detection algorithm based on the fusion attention mechanism and multi-layer convolution.

Abstract

In the realm of unmanned aerial vehicles, we proposed an enhanced small target detection algorithm, MGAC-YOLO, to address the challenges of missed detections and low accuracy associated with small target identification. Initially, we designed the MConv (Multi-layer Convolution) module to replace the conventional Conv module within the backbone network, thereby augmenting the dimensionality of information capture and enhancing the detection performance for small targets. Subsequently, we harnessed the advantages of both attention mechanisms-GAM (Global Attention Mechanism) and CloAttention (Contextualized Local and Global Attention)-to create a GACAttention module that extracts small target features from both global and local perspectives, thereby enriching the network's focus on small target feature information and further enhancing its feature processing capabilities. Finally, we incorporated an additional small target detection layer to capture feature information at a shallower level, thereby reducing the likelihood of missed detections and bolstering the detection capabilities for small targets. Experimental results on the VisDrone2019 dataset demonstrate that the Precision, mAP50, and mAP50-95 of the MGAC-YOLO algorithm have improved by 5.3%, 6.3%, and 4.4%, respectively, in comparison to the baseline model YOLOv8s. Furthermore, when compared to other leading algorithms, the MGAC-YOLO algorithm has exhibited notable superiority.