Hyperspectral Image Classification Based on 3D Convolution Joint Attention Mechanism

Hao WANG,Jingjing ZHANG,Yuanyuan LI,Feng WANG,Lina XUN
DOI: https://doi.org/10.3724/SP.J.7101182188
2020-01-01
Abstract:The high data dimension of hyperspectral images causes difficulties in the classification process.To improve the accuracy of classification,a hyperspectral image classification method using a 3D convolution joint attention mechanism is proposed.First,by pairing the center pixel with other pixels adjacent to it,it can form multiple sets of new pixel-pairs,and the neighborhood correlation between the pixels can be fully utilized.Then,the pixel-pairs are classified into the 3D convolution joint attention mechanism network framework,which can selectively learn the features in the hyperspectral image.Finally,the pixel label is obtained through the voting strategy.An experiment was carried out on two real hyperspectral image datasets.The results show that the proposed method fully exploits the spectral–spatial features of hyperspectral images,and this can effectively improve the classification accuracy.
What problem does this paper attempt to address?