Abstract:As Unmanned Aerial Vehicle (UAV) remote sensing technology progresses, the utilization of deep learning in UAV imagery object detection has become more prevalent. However, detecting small targets in complex backgrounds and distinguishing dense targets remains a major challenge. To address these issues and improve object detection efficiency, this study proposes an UAV imagery object detection method called YOLO-UAV by optimizing YOLOv5. YOLO-UAV first reconstructs the backbone and feature fusion networks by simplifying the network structure and reducing computational burden. The employment of a Dense_CSPDarknet53 backbone network, fashioned via the incorporation of dense connections, facilitates the extraction of latent image information through the recurrent utilization of features. In the Neck structure, an efficient feature fusion block with structural re-parameterization and ELAN strategies is integrated to effectively reduce interference from complex background noise while extracting more accurate and rich features. In addition, by proposing GS-Decoupled Head, this approach diminishes the parameter count of the decoupled head without compromising accuracy. It also separates classification tasks from regression tasks to lessen the influence of task disparities on prediction bias. To tackle the discrepancy between positive and negative samples in bounding box regression tasks, this study introduces a new loss function, Focal-ECIoU, capable of expediting network convergence and improve model positioning ability. Experimental findings from the public VisDrone2019 dataset indicate that YOLO-UAV outperforms other advanced object detection methods in comprehensive performance. Compared with the baseline model YOLOv5s, YOLO-UAV increased mAP0.5 from 35.1% to 46.7%, while mAP0.5:0.95 increased from 19.1% to 27.4%. For small-scale targets, AP $_{small}$ increased from 10.2% to 17.3%. The experiment proves that YOLO-UAV performs well in improving object detection accuracy and has strong generalization ability, satisfying the practical requirements of UAV imagery object detection tasks.

LA-YOLO: an effective detection model for multi-UAV under low altitude background

Detection and Recognition Method of Fast Low-Altitude Unmanned Aerial Vehicle Based on Dual Channel

Fast Detection and Recognition Method of UAV in Sky Background

A Novel Dataset and Detection Method for Unmanned Aerial Vehicles Using an Improved YOLOv9 Algorithm

YOLO-UAV: Object Detection Method of Unmanned Aerial Vehicle Imagery Based on Efficient Multi-Scale Feature Fusion

LightUAV-YOLO: a Lightweight Object Detection Model for Unmanned Aerial Vehicle Image

UAV-YOLOX: An Accurate Object Detection Method for UAV Images.

YOLO-HAMFF: A UAV detection model based on the hybrid attention and multi-level feature fusion for the YOLOv8 model.

YOLOD: A Target Detection Method for UAV Aerial Imagery

Mob-YOLO: A Lightweight UAV Object Detection Method

A Lightweight and Accurate UAV Detection Method Based on YOLOv4

Lightweight Low-Altitude UAV Object Detection Based on Improved YOLOv5s

UAV-YOLOv8: A Small-Object-Detection Model Based on Improved YOLOv8 for UAV Aerial Photography Scenarios

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

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

Small Target Detection for Unmanned Aerial Vehicle Images Based on YOLOv5l

Target Detection Method of UAV Aerial Imagery Based on Improved YOLOv5

M-YOLOv8s: an Improved Small Target Detection Algorithm for UAV Aerial Photography

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

A novel small object detection algorithm for UAVs based on YOLOv5

Method for Enhancing Unmanned Aerial Vehicle Target Detection in Challenging Environments Using Improved YOLOv8 Network