Abstract:Object detection in unmanned aerial vehicle (UAV) imagery is a meaningful foundation in various research domains. However, UAV imagery poses unique challenges, including large image sizes, small sizes detection objects, dense distribution, overlapping instances, and insufficient lighting impacting the effectiveness of object detection. In this article, we propose Drone-YOLO, a series of multi-scale UAV image object detection algorithms based on the YOLOv8 model, designed to overcome the specific challenges associated with UAV image object detection. To address the issues of large scene sizes and small detection objects, we introduce improvements to the neck component of the YOLOv8 model. Specifically, we employ a three-layer PAFPN structure and incorporate a detection head tailored for small-sized objects using large-scale feature maps, significantly enhancing the algorithm’s capability to detect small-sized targets. Furthermore, we integrate the sandwich-fusion module into each layer of the neck’s up–down branch. This fusion mechanism combines network features with low-level features, providing rich spatial information about the objects at different layer detection heads. We achieve this fusion using depthwise separable evolution, which balances parameter costs and a large receptive field. In the network backbone, we employ RepVGG modules as downsampling layers, enhancing the network’s ability to learn multi-scale features and outperforming traditional convolutional layers. The proposed Drone-YOLO methods have been evaluated in ablation experiments and compared with other state-of-the-art approaches on the VisDrone2019 dataset. The results demonstrate that our Drone-YOLO (large) outperforms other baseline methods in the accuracy of object detection. Compared to YOLOv8, our method achieves a significant improvement in mAP0.5 metrics, with a 13.4% increase on the VisDrone2019-test and a 17.40% increase on the VisDrone2019-val. Additionally, the parameter-efficient Drone-YOLO (tiny) with only 5.25 M parameters performs equivalently or better than the baseline method with 9.66M parameters on the dataset. These experiments validate the effectiveness of the Drone-YOLO methods in the task of object detection in drone imagery.

Improved YOLOv5 Network with CBAM for Object Detection Vision Drone

YOLOv5 with Dual Attention Network for Object Detection on Drone

Efficient Multi-Receptive Pooling YOLOv5 with Coordinate Attention Module for Object Detection on Drone

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

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

Enhanced Small Drone Detection Using Optimized YOLOv8 With Attention Mechanisms

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

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

Small Target Detection Algorithm for UAV Based on Improved YOLOv5

LAM-YOLO: Drones-based Small Object Detection on Lighting-Occlusion Attention Mechanism YOLO

Small object detection in UAV image based on improved YOLOv5

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

A Small Object Detection Method for Drone-Captured Images Based on Improved YOLOv7

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

A Modified YOLOv5 for Object Detection in UAV-captured Scenarios

Improved object detection algorithm for drone-captured dataset based on yolov5

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

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

Innovative Research on Small Object Detection and Recognition in Remote Sensing Images Using YOLOv5

DB-YOLOv5: A UAV Object Detection Model Based on Dual Backbone Network for Security Surveillance