Abstract:Object detection in drone-captured scenarios is a recent popular task. Due to the high flight altitude of unmanned aerial vehicle (UAV), the large variation of target scales, and the existence of dense occlusion of targets, in addition to the high requirements for real-time detection. To solve the above problems, we propose a real-time UAV small target detection algorithm based on improved ASFF-YOLOv5s. Based on the original YOLOv5s algorithm, the new shallow feature map is passed into the feature fusion network through multi-scale feature fusion to improve the extraction capability for small target features, and the Adaptively Spatial Feature Fusion (ASFF) is improved to improve the multi-scale information fusion capability. To obtain anchor frames for the VisDrone2021 dataset, we improve the K-means algorithm to obtain four different scales of anchor frames on each prediction layer. The Convolutional Block Attention Module (CBAM) is added in front of the backbone network and each prediction network layer to improve the capture capability of important features and suppress redundant features. Finally, to address the shortcomings of the original GIoU loss function, the SIoU loss function is used to accelerate the convergence of the model and improve accuracy. Extensive experiments conducted on the dataset VisDrone2021 show that the proposed model can detect a wide range of small targets in various challenging environments. At a detection rate of 70.4 FPS, the proposed model obtained a precision value of 32.55%, F1-score of 39.62%, and a mAP value of 38.03%, which improved 2.77, 3.98, and 5.1%, respectively, compared with the original algorithm, for the detection performance of small targets and to meet the task of real-time detection of UAV aerial images. The current work provides an effective method for real-time detection of small targets in UAV aerial photography in complex scenes, and can be extended to detect pedestrians, cars, etc. in urban security surveillance.

SF-YOLOv5: Improved YOLOv5 with swin transformer and fusion-concat method for multi-UAV detection

FE-YOLOv5: Improved YOLOv5 Network for Multi-scale Drone-Captured Scene Detection

UN-YOLOv5s: A UAV-Based Aerial Photography Detection Algorithm

Target Detection Method of UAV Aerial Imagery Based on Improved YOLOv5

A novel small object detection algorithm for UAVs based on YOLOv5

YOLOv5_mamba: unmanned aerial vehicle object detection based on bidirectional dense feedback network and adaptive gate feature fusion

An Improved YOLOv5 Method for Small Object Detection in UAV Capture Scenes

SMFF-YOLO: A Scale-Adaptive YOLO Algorithm with Multi-Level Feature Fusion for Object Detection in UAV Scenes

Research on Object Detection and Recognition Method for UAV Aerial Images Based on Improved YOLOv5

Multi-Object Detection Algorithm for UAV Aerial Images based on Improved YOLOV8

An improved UAV target detection algorithm based on ASFF-YOLOv5s.

YOLOv4_Drone: UAV image target detection based on an improved YOLOv4 algorithm

Small Target-YOLOv5: Enhancing the Algorithm for Small Object Detection in Drone Aerial Imagery Based on YOLOv5

Small Target Detection Algorithm for UAV Based on Improved YOLOv5

YOLO-Extract: Improved YOLOv5 for Aircraft Object Detection in Remote Sensing Images

An improved YOLOv5 method for large objects detection with multi-scale feature cross-layer fusion network

MCF-YOLOv5: A Small Target Detection Algorithm Based on Multi-Scale Feature Fusion Improved YOLOv5

An Aerial Image Detection Algorithm Based on Improved YOLOv5

An Efficient UAV Image Object Detection Algorithm Based on Global Attention and Multi-Scale Feature Fusion

DetailCaptureYOLO: Accurately Detecting Small Targets in UAV Aerial Images

CNTR-YOLO: Improved YOLOv5 Based on ConvNext and Transformer for Aircraft Detection in Remote Sensing Images