Unmanned Aerial Vehicle Small Target Detection Algorithm in Complex Scene
To enhance the model's recognition capability,this paper introduces a small target detection layer and a fea-ture enhancement module based on void convolution and attention mechanism to extract more features from multi-scale feature maps.Additionally,this paper incorporates a foreground awareness module to associate contextual information relat-ed to the foreground and reduce background interference.Experimental results on the VisDrone2021 dataset demonstrate significant improvements over YOLOv5s,with an increase of 5.3%in mAP50 and 3.4%in mAP50:95.Moreover,our model trained at 960×960 resolution achieves an mAP50 of 47.9%,which is 4.9%higher than YOLOv4,making it suitable for UAV aerial image detection.
UAVsmall target detectionattention mechanismcontext information