Affiliation:
1. National Key Laboratory of Transient Physics, Nanjing University of Science and Technology, Nanjing 210094, China
Abstract
In recent years, the realm of deep learning has witnessed significant advancements, particularly in object detection algorithms. However, the unique challenges posed by remote sensing images, such as complex backgrounds, diverse target sizes, dense target distribution, and overlapping or obscuring targets, demand specialized solutions. Addressing these challenges, we introduce a novel lightweight object detection algorithm based on Yolov5s to enhance detection performance while ensuring rapid processing and broad applicability. Our primary contributions include: firstly, we implemented a new Lightweight Asymmetric Detection Head (LADH-Head), replacing the original detection head in the Yolov5s model. Secondly, we introduce a new C3CA module, incorporating the Coordinate Attention mechanism, strengthening the network’s capability to extract precise location information. Thirdly, we proposed a new backbone network, replacing the C3 module in the Yolov5s backbone with a FasterConv module, enhancing the network’s feature extraction capabilities. Additionally, we introduced a Content-aware Feature Reassembly (content-aware reassembly of features) (CARAFE) module to reassemble semantic similar feature points effectively, enhancing the network’s detection capabilities and reducing the model parameters. Finally, we introduced a novel XIoU loss function, aiming to improve the model’s convergence speed and robustness during training. Experimental results on widely used remote sensing image datasets such as DIOR, DOTA, and SIMD demonstrate the effectiveness of our proposed model. Compared to the original Yolov5s algorithm, we achieved a mean average precision (mAP) increase of 3.3%, 6.7%, and 3.2%, respectively. These findings underscore the superior performance of our proposed model in remote sensing image object detection, offering an efficient, lightweight solution for remote sensing applications.
Subject
General Earth and Planetary Sciences
Reference110 articles.
1. Liu, P., Wang, Q., Zhang, H., Mi, J., and Liu, Y. (2023). A Lightweight Object Detection Algorithm for Remote Sensing Images Based on Attention Mechanism and YOLOv5s. Remote Sens., 15.
2. Satellite Remote Sensing: Sensors, Applications and Techniques;Roy;Proc. Natl. Acad. Sci. India Sect. A-Phys. Sci.,2017
3. Classifying urban land use by integrating remote sensing and social media data;Liu;Int. J. Geogr. Inf. Sci.,2017
4. Multi-source remote sensing data fusion: Status and trends;Zhang;Int. J. Image Data Fusion,2010
5. Efficient Inductive Vision Transformer for Oriented Object Detection in Remote Sensing Imagery;Zhang;IEEE Trans. Geosci. Remote Sens.,2023