A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n

Bibliographic Details
Title: A Lightweight Remote Sensing Aircraft Object Detection Network Based on Improved YOLOv5n
Authors: Jiale Wang, Zhe Bai, Ximing Zhang, Yuehong Qiu
Source: Remote Sensing, Vol 16, Iss 5, p 857 (2024)
Publisher Information: MDPI AG, 2024.
Publication Year: 2024
Collection: LCC:Science
Subject Terms: deep learning, lightweight network, YOLOv5n, Shufflenet v2, CA, EIoU loss, Science
More Details: 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.
Document Type: article
File Description: electronic resource
Language: English
ISSN: 2072-4292
Relation: https://www.mdpi.com/2072-4292/16/5/857; https://doaj.org/toc/2072-4292
DOI: 10.3390/rs16050857
Access URL: https://doaj.org/article/087689d44cbd491ea9050fff72c9ffc4
Accession Number: edsdoj.087689d44cbd491ea9050fff72c9ffc4
Database: Directory of Open Access Journals
Full text is not displayed to guests.
More Details
ISSN:20724292
DOI:10.3390/rs16050857
Published in:Remote Sensing
Language:English