Laser & Optoelectronics Progress, Volume. 59, Issue 2, 0210014(2022)

Hyperspectral Image Classification Based on Modified DenseNet and Spatial Spectrum Attention Mechanism

Xin Wang* and Yanguo Fan
Author Affiliations
  • College of Oceanography and Spatial Information, China University of Petroleum (East China), Qingdao , Shandong 266500, China
  • show less

    Aiming at the problems of hyperspectral image with high dimension, a few training samples, over fitting and too many training parameters, we propose an modified dense connection network (DenseNet) combined with spatial spectrum attention mechanism network (MDSSAN). First, the hyperspectral images are analyzed by principal component analysis, and the spatial neighborhoods of the central pixels are input into the modified network model. Then, three-dimensional DenseNet is improved, and the three-dimensional convolution block in the model is decomposed into the sampling convolution of the spatial dimension and the spectral dimension. Finally, the spatial attention mechanism is introduced in the spatial dimension, and the channel attention mechanism is introduced in the spectral dimension to reduce the training parameters of the model and extract more discriminative space-spectrum joint features. Experimental results show that the overall classification accuracy of the MDSSAN model on the Indian Pines, Pavia University, and KSC data sets are 99.43%, 99.74%, and 98.98%, respectively. Compared with other comparison models, the model has faster convergence speed and better classification performance.

    Tools

    Get Citation

    Copy Citation Text

    Xin Wang, Yanguo Fan. Hyperspectral Image Classification Based on Modified DenseNet and Spatial Spectrum Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(2): 0210014

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Jan. 21, 2021

    Accepted: Mar. 15, 2021

    Published Online: Dec. 23, 2021

    The Author Email: Wang Xin (3166588225@qq.com)

    DOI:10.3788/LOP202259.0210014

    Topics