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 条
[21]   Small Object Detection in Unmanned Aerial Vehicle Images Using Feature Fusion and Scaling-Based Single Shot Detector With Spatial Context Analysis [J].
Liang, Xi ;
Zhang, Jing ;
Zhuo, Li ;
Li, Yuzhao ;
Tian, Qi .
IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2020, 30 (06) :1758-1770
[22]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[23]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37
[24]   ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design [J].
Ma, Ningning ;
Zhang, Xiangyu ;
Zheng, Hai-Tao ;
Sun, Jian .
COMPUTER VISION - ECCV 2018, PT XIV, 2018, 11218 :122-138
[25]   ESPNetv2: A Light-weight, Power Efficient, and General Purpose Convolutional Neural Network [J].
Mehta, Sachin ;
Rastegari, Mohammad ;
Shapiro, Linda ;
Hajishirzi, Hannaneh .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9182-9192
[26]   ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation [J].
Mehta, Sachin ;
Rastegari, Mohammad ;
Caspi, Anat ;
Shapiro, Linda ;
Hajishirzi, Hannaneh .
COMPUTER VISION - ECCV 2018, PT X, 2018, 11214 :561-580
[27]   A Deep Learning Approach for Drone Detection and Classification using Radar and Camera Sensor Fusion [J].
Mehta, Varun ;
Dadboud, Fardad ;
Bolic, Miodrag ;
Mantegh, Iraj .
2023 IEEE SENSORS APPLICATIONS SYMPOSIUM, SAS, 2023,
[28]  
Misbah Maham, 2023, Wireless and Satellite Systems: 13th EAI International Conference, WiSATS 2022, Virtual Event, Proceedings. Lecture Notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering (509), P3, DOI 10.1007/978-3-031-34851-8_1
[29]   Investigating Cost-effective RF-based Detection of Drones [J].
Phuc Nguyen ;
Ravindranathan, Mahesh ;
Anh Nguyen ;
Han, Richard ;
Vu, Tam .
DRONET'16: PROCEEDINGS OF THE 2ND WORKSHOP ON MICRO AERIAL VEHICLE NETWORKS, SYSTEMS, AND APPLICATIONS FOR CIVILIAN USE, 2016, :17-22
[30]   Small Object Detection Method Based on Adaptive Spatial Parallel Convolution and Fast Multi-Scale Fusion [J].
Qi, Guanqiu ;
Zhang, Yuanchuan ;
Wang, Kunpeng ;
Mazur, Neal ;
Liu, Yang ;
Malaviya, Devanshi .
REMOTE SENSING, 2022, 14 (02)