Laser & Optoelectronics Progress, Volume. 57, Issue 20, 201101(2020)

GGCN: GPU-Based Hyperspectral Image Classification Algorithm

Minghua Zhang1, Yaqing Zou1, Wei Song1, Dongmei Huang1,2、*, and Zhixiang Liu1
Author Affiliations
  • 1College of Information Science, Shanghai Ocean University, Shanghai 201306, China
  • 2College of Electronics and Information Engineering, Shanghai University of Electric Power, Shanghai 200090, China
  • show less

    Hyperspectral image classification is one of the research hotspots in the field of remote sensing. It is an important means of earth observation and has important applications in areas such as fine identification of ground objects. The use of convolutional neural networks (CNN) can effectively extract advanced features from the original image with high classification accuracy. However, CNN has a huge amount of calculations and requires high-performance hardware. In order to improve the computational efficiency of the model, the CNN model can be trained on the GPU. Existing parallel algorithms such as GCN (GPU based Cube-CNN) cannot make full use of the parallel capabilities of the GPU, and the algorithm acceleration effect is not ideal. In order to further improve the efficiency of the algorithm, the GGCN (GPU based Cube-CNN improved by GEMM) parallel acceleration algorithm based on the general matrix multiply (GEMM) algorithm is proposed. G-PNPE(GEMM based Parallel Neighbor Pixels Extraction) reorganizes and arranges the input data and convolution kernel to achieve parallel calculation of convolution, which effectively improves the utilization of GPU and increases the training efficiency of the algorithm. By analyzing the experimental results on the three datasets, the classification accuracy of the improved algorithm is consistent with the original algorithm, and the training time of the CNN network is shortened by about 30%, which proves the effectiveness and superiority of the algorithm.

    Tools

    Get Citation

    Copy Citation Text

    Minghua Zhang, Yaqing Zou, Wei Song, Dongmei Huang, Zhixiang Liu. GGCN: GPU-Based Hyperspectral Image Classification Algorithm[J]. Laser & Optoelectronics Progress, 2020, 57(20): 201101

    Download Citation

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

    Category: Imaging Systems

    Received: Dec. 16, 2019

    Accepted: Feb. 25, 2020

    Published Online: Oct. 13, 2020

    The Author Email: Huang Dongmei (dmhuang@shou.edu.cn)

    DOI:10.3788/LOP57.201101

    Topics