Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2020, Vol. 41 ›› Issue (6): 897-904.DOI: 10.11996/JG.j.2095-302X.2020060897

Previous Articles     Next Articles

DenseNet-attention for hyperspectral remote sensing image classification  

  

  1. (School of Computer Science and Engineering, North Minzu University, Yinchuan Ningxia 750021, China) 
  • Online:2020-12-31 Published:2021-01-08
  • Supported by:
    Foundation items:National Natural Science Foundation of China (61461002, 61762003); Key Research and Development Project of Ningxia Hui Autonomous Region (2019BDE03011); Ningxia University First-Class Discipline Construction Project (Electronic Science and Technology) (NXYLXK2017A07) 

Abstract: Abstract: A new neural network, called DenseNet-Attention (DANet), was proposed in this paper for hyperspectral images classification to solve the problems of small sample quantity, insufficient features extraction, and indiscriminating contribution of the extracted features. First, it employed the three-dimensional convolution kernel to simultaneously extract both spectral and spatial features. Meanwhile, due to its dense blocks, DenseNet can not only fully extract more robust features, but reduce a large number of parameters. Second, the self-attention mechanism was added to the dense block as a module. Before the extracted feature was passed into the next layer of network, the weight was assigned to the feature according to its contribution through this model, thus strengthening the representation of the feature with ground object information. DANet was an end-to-end deep learning framework, which took the neighborhood block of the original hyperspectral image as an input without any preprocessing. Comparative experiments on Indian Pines and Pavia University datasets show that the classification accuracy of the network model proposed in this paper can reach 99.43% and 99.99% respectively, effectively enhancing the classification accuracy of hyperspectral images.

Key words: Keywords: three-dimensional convolution, hyperspectral remote sensing image classification, DenseNet, self-attenrion, skip connect 

CLC Number: