A Efficient and Accurate UAV Detection Method Based on YOLOv5s

被引:2
作者
Feng, Yunsong [1 ]
Wang, Tong [1 ,2 ]
Jiang, Qiangfu [2 ]
Zhang, Chi [1 ]
Sun, Shaohang [1 ]
Qian, Wangjiahe [1 ]
机构
[1] Natl Univ Def Technol, State Key Lab Pulsed Power Laser Technol, Hefei 230037, Peoples R China
[2] Anhui Univ, Sch Phys & Optoelect Engn, Hefei 230601, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2024年 / 14卷 / 15期
关键词
UAV detection; lightweight network; ShuffleNetV2; bidirectional feature pyramid network; coordinate attention; ghost convolution; EIoU; NETWORK;
D O I
10.3390/app14156398
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Due to the limited computational resources of portable devices, target detection models for drone detection face challenges in real-time deployment. To enhance the detection efficiency of low, slow, and small unmanned aerial vehicles (UAVs), this study introduces an efficient drone detection model based on YOLOv5s (EDU-YOLO), incorporating lightweight feature extraction and balanced feature fusion modules. The model employs the ShuffleNetV2 network and coordinate attention mechanisms to construct a lightweight backbone network, significantly reducing the number of model parameters. It also utilizes a bidirectional feature pyramid network and ghost convolutions to build a balanced neck network, enriching the model's representational capacity. Additionally, a new loss function, EloU, replaces CIoU to improve the model's positioning accuracy and accelerate network convergence. Experimental results indicate that, compared to the YOLOv5s algorithm, our model only experiences a minimal decrease in mAP by 1.1%, while reducing GFLOPs from 16.0 to 2.2 and increasing FPS from 153 to 188. This provides a substantial foundation for networked optoelectronic detection of UAVs and similar slow-moving aerial targets, expanding the defensive perimeter and enabling earlier warnings.
引用
收藏
页数:17
相关论文
共 57 条
[1]   Detection of Unauthorized Unmanned Aerial Vehicles Using YOLOv5 and Transfer Learning [J].
Al-Qubaydhi, Nader ;
Alenezi, Abdulrahman ;
Alanazi, Turki ;
Senyor, Abdulrahman ;
Alanezi, Naif ;
Alotaibi, Bandar ;
Alotaibi, Munif ;
Razaque, Abdul ;
Abdelhamid, Abdelaziz A. ;
Alotaibi, Aziz .
ELECTRONICS, 2022, 11 (17)
[2]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[3]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[4]   Extended Feature Pyramid Network for Small Object Detection [J].
Deng, Chunfang ;
Wang, Mengmeng ;
Liu, Liang ;
Liu, Yong ;
Jiang, Yunliang .
IEEE TRANSACTIONS ON MULTIMEDIA, 2022, 24 :1968-1979
[5]   Infrared and visible image fusion with the use of multi-scale edge-preserving decomposition and guided image filter [J].
Gan, Wei ;
Wu, Xiaohong ;
Wu, Wei ;
Yang, Xiaomin ;
Ren, Chao ;
He, Xiaohai ;
Liu, Kai .
INFRARED PHYSICS & TECHNOLOGY, 2015, 72 :37-51
[6]   NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detection [J].
Ghiasi, Golnaz ;
Lin, Tsung-Yi ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :7029-7038
[7]   Effective Fusion Factor in FPN for Tiny Object Detection [J].
Gong, Yuqi ;
Yu, Xuehui ;
Ding, Yao ;
Peng, Xiaoke ;
Zhao, Jian ;
Han, Zhenjun .
2021 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2021), 2021, :1159-1167
[8]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[9]   AMC: AutoML for Model Compression and Acceleration on Mobile Devices [J].
He, Yihui ;
Lin, Ji ;
Liu, Zhijian ;
Wang, Hanrui ;
Li, Li-Jia ;
Han, Song .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :815-832
[10]   Coordinate Attention for Efficient Mobile Network Design [J].
Hou, Qibin ;
Zhou, Daquan ;
Feng, Jiashi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13708-13717