Abstract:With the accelerated development of the technological power of society, aerial images of drones gradually penetrated various industries. Due to the variable speed of drones, the captured images are shadowed, blurred, and obscured. Second, drones fly at varying altitudes, leading to changing target scales and making it difficult to detect and identify small targets. In order to solve the above problems, an improved ASG-YOLOv5 model is proposed in this paper. Firstly, this research proposes a dynamic contextual attention module, which uses feature scores to dynamically assign feature weights and output feature information through channel dimensions to improve the model's attention to small target feature information and increase the network's ability to extract contextual information; secondly, this research designs a spatial gating filtering multi-directional weighted fusion module, which uses spatial filtering and weighted bidirectional fusion in the multi-scale fusion stage to improve the characterization of weak targets, reduce the interference of redundant information, and better adapt to the detection of weak targets in images under unmanned aerial vehicle remote sensing aerial photography; meanwhile, using Normalized Wasserstein Distance and CIoU regression loss function, the similarity metric value of the regression frame is obtained by modeling the Gaussian distribution of the regression frame, which increases the smoothing of the positional difference of the small targets and solves the problem that the positional deviation of the small targets is very sensitive, so that the model's detection accuracy of the small targets is effectively improved. This paper trains and tests the model on the VisDrone2021 and AI-TOD datasets. This study used the NWPU-RESISC dataset for visual detection validation. The experimental results show that ASG-YOLOv5 has a better detection effect in unmanned aerial vehicle remote sensing aerial images, and the frames per second (FPS) reaches 86, which meets the requirement of real-time small target detection, and it can be better adapted to the detection of the weak and small targets in the aerial image dataset, and ASG-YOLOv5 outperforms many existing target detection methods, and its detection accuracy reaches 21.1% mAP value. The mAP values are improved by 2.9% and 1.4%, respectively, compared with the YOLOv5 model. The project is available at https://github.com/woaini-shw/asg-yolov5.git.

AVS-YOLO: Object Detection in Aerial Visual Scene

AIOD-YOLO: an algorithm for object detection in low-altitude aerial images

Analysis and Adaptation of YOLOv4 for Object Detection in Aerial Images

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

Drone-YOLO: An Efficient Neural Network Method for Target Detection in Drone Images

VAMYOLOX: an Accurate and Efficient Object Detection Algorithm Based on Visual Attention Mechanism for UAV Optical Sensors

Efficient-Lightweight YOLO: Improving Small Object Detection in YOLO for Aerial Images

From Blurry to Brilliant Detection: YOLOv5-Based Aerial Object Detection with Super Resolution

SOD‐YOLO: Small Object Detection Network for UAV Aerial Images

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

MS-YOLO: integration-based multi-subnets neural network for object detection in aerial images

DM-YOLOX Aerial Object Detection Method with Intensive Attention Mechanism

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

YOLOv7-UAV: An Unmanned Aerial Vehicle Image Object Detection Algorithm Based on Improved YOLOv7

BGF-YOLOv10: Small Object Detection Algorithm from Unmanned Aerial Vehicle Perspective Based on Improved YOLOv10

SOD-YOLO: Small-Object-Detection Algorithm Based on Improved YOLOv8 for UAV Images

YOLOv3 with Spatial Pyramid Pooling for Object Detection with Unmanned Aerial Vehicles

Efficient Small Object Detection You Only Look Once: A Small Object Detection Algorithm for Aerial Images

ASG-YOLOv5: Improved YOLOv5 unmanned aerial vehicle remote sensing aerial images scenario for small object detection based on attention and spatial gating

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