Affiliation:
1. School of Instrument Science and Engineering, Southeast University, Nanjing 210096, China
Abstract
The specific characteristics of remote sensing images, such as large directional variations, large target sizes, and dense target distributions, make target detection a challenging task. To improve the detection performance of models while ensuring real-time detection, this paper proposes a lightweight object detection algorithm based on an attention mechanism and YOLOv5s. Firstly, a depthwise-decoupled head (DD-head) module and spatial pyramid pooling cross-stage partial GSConv (SPPCSPG) module were constructed to replace the coupled head and the spatial pyramid pooling-fast (SPPF) module of YOLOv5s. A shuffle attention (SA) mechanism was introduced in the head structure to enhance spatial attention and reconstruct channel attention. A content-aware reassembly of features (CARAFE) module was introduced in the up-sampling operation to reassemble feature points with similar semantic information. In the neck structure, a GSConv module was introduced to maintain detection accuracy while reducing the number of parameters. Experimental results on remote sensing datasets, RSOD and DIOR, showed an improvement of 1.4% and 1.2% in mean average precision accuracy compared with the original YOLOv5s algorithm. Moreover, the algorithm was also tested on conventional object detection datasets, PASCAL VOC and MS COCO, which showed an improvement of 1.4% and 3.1% in mean average precision accuracy. Therefore, the experiments showed that the constructed algorithm not only outperformed the original network on remote sensing images but also performed better than the original network on conventional object detection images.
Funder
National Natural Science Foundation of China
Post graduate Research and Practice Innovation Program of Jiangsu Province
Subject
General Earth and Planetary Sciences
Reference154 articles.
1. Haq, M.A., Ahmed, A., Khan, I., Gyani, J., Mohamed, A., Attia, E., Mangan, P., and Pandi, D. (2022). Analysis of environmental factors using AI and ML methods. Sci. Rep., 12.
2. Deep Learning Based Modeling of Groundwater Storage Change;Haq;CMC Comput. Mat. Contin.,2022
3. CDLSTM: A Novel Model for Climate Change Forecasting;Haq;CMC Comput. Mat. Contin.,2022
4. SMOTEDNN: A Novel Model for Air Pollution Forecasting and AQI Classification;Haq;CMC Comput. Mat. Contin.,2022
5. Ning, Z., Sun, S., Wang, X., Guo, L., Wang, G., Gao, X., and Kwok, R.Y.K. (2021). Intelligent resource allocation in mobile blockchain for privacy and security transactions: A deep reinforcement learning based approach. Sci. China Inf. Sci., 64.
Cited by
8 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献