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.

Evaluating the influence of backbone network architectures for object detection in aerial images

DroneNet: Rescue Drone-View Object Detection

A Training-time Friendly Network for Real-time Drone Detection.

A study on detecting drones using deep convolutional neural networks

Aerial Data Exploration: An in-Depth Study From Horizontal to Oriented Viewpoint

Object detection on aerial imagery using CenterNet

Enhancing object detection in aerial images

Object Recognition in Aerial Images Using Convolutional Neural Networks

On-Board Small-Scale Object Detection for Unmanned Aerial Vehicles (UAVs)

DeepVision: Enhanced Drone Detection and Recognition in Visible Imagery through Deep Learning Networks

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

Towards a High-Performance Object Detector: Insights from Drone Detection Using ViT and CNN-based Deep Learning Models

A Novel Approach to Detect Drones Using Deep Convolutional Neural Network Architecture

Multiple-Oriented and Small Object Detection with Convolutional Neural Networks for Aerial Image

Convolutional network architectures for super-resolution/sub-pixel mapping of drone-derived images

Monitoring and surveillance of urban road traffic using low altitude drone images: a deep learning approach

IoT Enabled Deep Learning Based Framework for Multiple Object Detection in Remote Sensing Images

Efficient Object Detection Model for Real-Time UAV Applications

On the usability of deep networks for object-based image analysis

Learning to Reduce Information Bottleneck for Object Detection in Aerial Images