Lightweight UAV Detection Algorithm Based on YOLOv5

He-xiang HAO , Yue-ping PENG , Bai-xuan HAN , Wen-ji YIN

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (28) : 12251 -12258.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (28) : 12251-12258 DOI: 10.12404/j.issn.1671-1815.2400915
research-article
Lightweight UAV Detection Algorithm Based on YOLOv5
Author information +
History +
PDF

Abstract

In order to solve the security risks caused by UAV "black flight", aiming at the problems of large model parameters and long training time of the existing UAV target detection algorithm based on deep learning, a lightweight real-time object detection algorithm for UAV based on YOLOv5 (ED-YOLOv5s) was proposed. Firstly, the feature extraction part of YOLOv5 backbone network was reconstructed by combining the lightweight model efficient model (EMO). Secondly, normalized gaussian wasserstein distance (NWD) and scale factor were introduced to calculate the similarity between candidate boxes to partially replace IoU. Then, SimAM was introduced to optimize the weight distribution and improved the detection accuracy, so as to achieved the effect of optimizing the Backbone and Head of YOLOv5 network. Finally, the model size is 4.57 M. ED-YOLOv5s lightweight UAV detection algorithm with floating point computation of 26.1 (GFLOPs). The experimental data shows that the improved algorithm improves the detection accuracy and realizes the model lightweight. The proposed algorithm achieves 96.7% value on the DUT Anti-UAV dataset, and the detection speed reaches 37 frame/s on the RTX2060 GPU.

Keywords

unmanned aerial vehicle / lightweight / object detection / YOLOv5

Cite this article

Download citation ▾
He-xiang HAO, Yue-ping PENG, Bai-xuan HAN, Wen-ji YIN. Lightweight UAV Detection Algorithm Based on YOLOv5. Science Technology and Engineering, 2024, 24(28): 12251-12258 DOI:10.12404/j.issn.1671-1815.2400915

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (1733KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉