UAV small target detection algorithm based on improved YOLOv5s
An unmanned aerial vehicle (UAV) small target detection algorithm based on YOLOv5,termed FDB-YOLO,was proposed to address the significant issue of misidentification and omissions in traditional target detection algorithms when applied to UAV aerial photography of small targets.Initially,a small target detection layer was added on the basis of YOLOv5,and the feature fusion network was optimized to fully leverage the fine-grained information of small targets in shallow layers,thereby enhancing the network's perceptual capabilities.Subsequently,a novel loss function,FPIoU,was introduced,which capitalized on the geometric properties of anchor boxes and utilized a four-point positional bias constraint function to optimize the anchor box positioning and accelerate the convergence speed of the loss function.Furthermore,a dynamic target detection head (DyHead) incorporating attention mechanism was employed to enhance the algorithm's detection capabilities through increased awareness of scale,space,and task.Finally,a bi-level routing attention mechanism (BRA) was integrated into the feature extraction phase,selectively computing relevant areas to filter out irrelevant regions,thereby improving the model's detection accuracy.Experimental validation conducted on the VisDrone2019 dataset demonstrated that the proposed algorithm outperformed the YOLOv5s baseline in terms of Precision by an increase of 3.7 percentage points,Recall by an increase of 5.1 percentage points,mAP50 by an increase of 5.8 percentage points,and mAP50:95 by an increase of 3.4 percentage points,showcasing superior performance compared to current mainstream algorithms.
万方数据
