A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n
文献类型:期刊论文
作者 | Wang, Jiale1,2; Bai, Zhe2![]() ![]() |
刊名 | Remote Sensing
![]() |
出版日期 | 2024-03 |
卷号 | 16期号:5 |
关键词 | deep learning lightweight network YOLOv5n Shufflenet v2 CA EIoU loss deformable convolution |
ISSN号 | 20724292 |
DOI | 10.3390/rs16050857 |
产权排序 | 1 |
英文摘要 | Due to the issues of remote sensing object detection algorithms based on deep learning, such as a high number of network parameters, large model size, and high computational requirements, it is challenging to deploy them on small mobile devices. This paper proposes an extremely lightweight remote sensing aircraft object detection network based on the improved YOLOv5n. This network combines Shufflenet v2 and YOLOv5n, significantly reducing the network size while ensuring high detection accuracy. It substitutes the original CIoU and convolution with EIoU and deformable convolution, optimizing for the small-scale characteristics of aircraft objects and further accelerating convergence and improving regression accuracy. Additionally, a coordinate attention (CA) mechanism is introduced at the end of the backbone to focus on orientation perception and positional information. We conducted a series of experiments, comparing our method with networks like GhostNet, PP-LCNet, MobileNetV3, and MobileNetV3s, and performed detailed ablation studies. The experimental results on the Mar20 public dataset indicate that, compared to the original YOLOv5n network, our lightweight network has only about one-fifth of its parameter count, with only a slight decrease of 2.7% in mAP@0.5. At the same time, compared with other lightweight networks of the same magnitude, our network achieves an effective balance between detection accuracy and resource consumption such as memory and computing power, providing a novel solution for the implementation and hardware deployment of lightweight remote sensing object detection networks. © 2024 by the authors. |
语种 | 英语 |
出版者 | Multidisciplinary Digital Publishing Institute (MDPI) |
源URL | [http://ir.opt.ac.cn/handle/181661/97279] ![]() |
专题 | 西安光学精密机械研究所_空间光学应用研究室 |
通讯作者 | Zhang, Ximing |
作者单位 | 1.University of Chinese Academy of Sciences, Beijing; 100049, China 2.Xi’an Institute of Optics and Precision Mechanics of CAS, Xi’an; 710119, China; |
推荐引用方式 GB/T 7714 | Wang, Jiale,Bai, Zhe,Zhang, Ximing,et al. A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n[J]. Remote Sensing,2024,16(5). |
APA | Wang, Jiale,Bai, Zhe,Zhang, Ximing,&Qiu, Yuehong.(2024).A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n.Remote Sensing,16(5). |
MLA | Wang, Jiale,et al."A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n".Remote Sensing 16.5(2024). |
入库方式: OAI收割
来源:西安光学精密机械研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。