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.

SDWBF Algorithm: A Novel Pedestrian Detection Algorithm in the Aerial Scene

A Small UAV Detection Method Based on Optical Flow and Visual Feature Fusion

Pedestrian Detection Optimization Algorithm Based on Low-Altitude UAV

High-Sensity Pedestrian Detection from Aerial View

Lightweight Pedestrian Detection Network for UAV Remote Sensing Images Based on Strideless Pooling

The Automatic Detection of Pedestrians under the High-Density Conditions by Deep Learning Techniques

Pedestrian Detection in Unmanned Aerial Vehicle Scene

Near-surface Pedestrian Detection Method Based on Deep Learning for UAVs in Low Illumination Environments

A New Algorithm for Small Target Detection From the Perspective of Unmanned Aerial Vehicles

Aero-YOLO: An Efficient Vehicle and Pedestrian Detection Algorithm Based on Unmanned Aerial Imagery

UPDN: Pedestrian Detection Network for Unmanned Aerial Vehicle Perspective.

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

Improved YOLOv4 for Pedestrian Detection and Counting in UAV Images

A Pedestrian Detection Algorithm Based on Score Fusion for Multi-LiDAR Systems

Small Target Detection Algorithm for UAV Aerial Photography Based on Improved YOLOv5s

Pedestrian Detection And Tracking From Low-Resolution Unmanned Aerial Vehicle Thermal Imagery

Small-Object Detection for UAV-Based Images Using a Distance Metric Method

A Small-Scale Pedestrian Detection Method in Railway Scene

An Improved Lightweight Real-Time Detection Algorithm Based on the Edge Computing Platform for UAV Images

UAV Image Small Object Detection Based on Composite Backbone Network

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8