Back to Search Start Over

A Remote Sensing Target Detection Model Based on Lightweight Feature Enhancement and Feature Refinement Extraction

Authors :
Dongen Guo
Zhuoke Zhou
Fengshuo Guo
Chaoxin Jia
Xiaohong Huang
Jiangfan Feng
Zhen Shen
Source :
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, Vol 17, Pp 9569-9581 (2024)
Publication Year :
2024
Publisher :
IEEE, 2024.

Abstract

Remote sensing image (RSI) target detection methods based on traditional multiscale feature fusion (MSFF) have achieved great success. However, the traditional MSFF method significantly increases the computational cost during model training and inference, and the simple fusion operation may lead to the semantic confusion of the feature map, which cannot realize the refined extraction of features by the model. In order to reduce the computational effort associated with the MSFF operation and to enable the features in the feature map to present an accurate, fine-grained distribution, we propose a single-stage detection model (RS-YOLO). Our main additions to RS-YOLO are a computationally smaller and faster Quick and Small E-ELEN (QS-E-ELEN) module and a feature refinement extraction (FRE) module. In the QS-E-ELEN module, We utilize QSBlock, jump-join, and convolution operations to fuse features on different scales and reduce the computational effort of the model by exploiting the similarity of the RSI feature map channels. In order for the model to better utilize the enhanced features, we designed the FRE module to make the location of the enhanced features more accurate and fine. By conducting experiments on the popular NWPU-VHR- 10 and SSDD datasets, we derive results showing that RS-YOLO outperforms most mainstream models in terms of the tradeoff between accuracy and speed. Specifically, in terms of accuracy, it improves 1.6$\%$ and 1.7$\%$ compared to the current state-of-the-art models, respectively. At the same time, RS-YOLO reduces the number of parameters and computational effort.

Details

Language :
English
ISSN :
19391404 and 21511535
Volume :
17
Database :
Directory of Open Access Journals
Journal :
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Publication Type :
Academic Journal
Accession number :
edsdoj.05e9b0acacef4dffb2f01664196d1763
Document Type :
article
Full Text :
https://doi.org/10.1109/JSTARS.2024.3394887