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.
Original language | English (US) |
---|---|
Pages (from-to) | 134800-134808 |
Number of pages | 9 |
Journal | IEEE Access |
Volume | 9 |
DOIs | |
State | Published - 2021 |
Keywords
- Attention mechanism
- Image classification
- Remote sensing
- Residual network
- Spatial information
ASJC Scopus subject areas
- General Computer Science
- General Materials Science
- General Engineering
- Electrical and Electronic Engineering