Improved Attention Mechanism and Residual Network for Remote Sensing Image Scene Classification

Abstract

Aiming at the problem that the accuracy of traditional remote sensing image classification model is not ideal, a classification method based on improved attention mechanism and residual network is proposed. In order to prevent overfitting, we choose the network structure of ResNet18 as the framework. Meanwhile, ResNet18 has a small number of parameters and a fast calculation speed. Then, a parallel attention mechanism, ProCBAM, is proposed and added to the BasicBlock of the residual network. By optimizing the representation of the feature map from the spatial dimension and channel dimension of the feature map, more detailed image information is learned and image recognition errors are reduced. The experimental results on the open source dataset show that the accuracy is 93.86%, and a more accurate image classification model is successfully trained.

Keywords

• Remote sensing
• image classification
• attention mechanism
• residual network
• spatial information