10 November 2023 Lightweight multi-target detection algorithm for unmanned aerial vehicle aerial imagery
Yang Liu, Ding Ma, Yongfu Wang
Author Affiliations +
Abstract

Compared with the image captured in the natural scene, the image obtained by unmanned aerial vehicle (UAV) aerial photography has a more complex background and many dense small targets, which puts forward higher requirements for the detection accuracy of the target detection algorithm. However, because the UAV is a kind of small mobile device, how to ensure its real-time detection effect has been a problem. Aiming at these problems, the lightweight YOLOv7 algorithm, namely LRT-YOLOv7, is designed. First, the enhance feature fusion module and the transformer efficient layer aggregation networks module are proposed to improve the performance of feature extraction and fusion to enhance the efficiency of small target detection. Second, aiming at the problems of small target size and complex background in the UAV images, the detection head structure is redesigned in the YOLOv7-tiny algorithm to enhance the multi-scale feature fusion ability of the algorithm and thereby improve the algorithm’s detection accuracy for small targets. Finally, ablation, comparison, and visualization validation experiments were conducted using precision, recall, mean average precision, and frames per second (FPS) as evaluation indicators. The results show that the detection speed of the LRT-YOLOv7 algorithm on the self-made traffic target dataset is 133.8 FPS, and the precision indicator is 84.58%. Therefore, the LRT-YOLOv7 algorithm has high accuracy and real-time performance in traffic target detection tasks for UAV aerial imagery.

© 2023 Society of Photo-Optical Instrumentation Engineers (SPIE)
Yang Liu, Ding Ma, and Yongfu Wang "Lightweight multi-target detection algorithm for unmanned aerial vehicle aerial imagery," Journal of Applied Remote Sensing 17(4), 046505 (10 November 2023). https://doi.org/10.1117/1.JRS.17.046505
Received: 1 August 2023; Accepted: 25 October 2023; Published: 10 November 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Target detection

Detection and tracking algorithms

Unmanned aerial vehicles

Small targets

Head

Convolution

Feature extraction

Back to Top