Object Detection for Remote Sensing images based on multi-scale features and attention mechanism

• Published in: IEEE access p. 1
• Main Authors: Zhao, Hu, Chu, Kaibin, Zhang, Ji, Feng, Chengtao
• Format: Journal Article
• Language: English
• Published: IEEE 16-05-2023
• Subjects: Attention mechanisms; Complexity theory; Convolution; Deep learning; Feature extraction; Feature fusion; Marine vehicles; Object detection; Remote sensing; Remote Sensing images; Semantics
• ISSN: 2169-3536
• DOI: 10.1109/ACCESS.2023.3277227

Object detection for remote sensing images has problems such as complex backgrounds, multi-scale and difficulty in the detection of small targets. Because of the above problems, an improved object detection algorithm for remote-sensing images is proposed. Firstly, fusing the characteristics of shallow and deep networks, the original feature pyramid structure is reconstructed, the improved adaptive feature fusion structure is introduced before the network prediction, and the location and category information of different feature maps are fused. Secondly, the attention mechanism is introduced into the network to reduce the interference caused by complex backgrounds; Finally, the up-sampling module is improved to expand the receptive field and realize up-sampling based on semantic information. Through experiments, we show that our network achieves 93.9% and 98.5% detection accuracy on Levir and RSOD dataset, respectively, with the detection speed, reaching 98.26fps. Compared with the original network, the accuracy is increasing by 4% and 1.2%, respectively. Experimental results show that the detection accuracy of our algorithm is improved while maintaining the light weights and efficiencies of the original network.