Laser & Optoelectronics Progress, Volume. 59, Issue 8, 0810002(2022)

Hyperspectral Image Classification Based on Fast Double Branch Dense Connection Network and Double Attention Mechanism

Guoliang Yang, Jiaren Gong*, Hao Xi, Shicong Li, and Junfeng Zou
Author Affiliations
  • School of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou , Jiangxi 341000, China
  • show less

    In recent years, the classification of hyperspectral images (HSI) based on deep learning has attracted extensive attention in various fields. The number of HSI spectral bands is large and information redundancy is high, which increases calculation complexity. However, a lack of training samples can easily lead to overfitting of model training and reduce classification accuracy. To improve the classification accuracy and reduce the training time, this study proposes a fast dense connection network based on a three-dimensional convolutional neural network (3D-CNN) with double branch and double attention mechanism for HSI classification. In this study, the original data is dimensionless using principal component analysis (PCA) to reduce redundant information. Then, a double efficient channel attention (ECA) mechanism with a double branch dense connection structure and a fast Fourier transform (FFT) is used. It not only ensures the model's classification accuracy but also increases the model's training speed. We conducted experiments on multiple public hyperspectral datasets, and the performance evaluation indices are overall classification accuracy (OA), average classification accuracy (AA), and Kappa coefficient. The experimental results show that the proposed method can improve the classification accuracy and significantly reduce the training time and testing time.

    Tools

    Get Citation

    Copy Citation Text

    Guoliang Yang, Jiaren Gong, Hao Xi, Shicong Li, Junfeng Zou. Hyperspectral Image Classification Based on Fast Double Branch Dense Connection Network and Double Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(8): 0810002

    Download Citation

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

    Category: Image Processing

    Received: Jul. 2, 2021

    Accepted: Aug. 25, 2021

    Published Online: Apr. 11, 2022

    The Author Email: Gong Jiaren (964331424@qq.com)

    DOI:10.3788/LOP202259.0810002

    Topics