激光与光电子学进展, 2019, 56 (11): 111007, 网络出版: 2019-06-13
基于超图和卷积神经网络的高光谱图像分类 下载: 1349次
Hyperspectral Image Classification Based on Hypergraph and Convolutional Neural Network
图像处理 高光谱图像 分类 超图 卷积神经网络 谱空联合信息 image processing hyperspectral image classification hypergraph convolutional neural network spectral space joint information
摘要
针对高光谱图像数据维数多,光谱信息和空间信息难以提取的问题,提出了一种基于超图和卷积神经网络的分类算法,依据高光谱图像中像素之间的光谱关系和空间关系构建超图;通过超图构建具有谱空联合特征的样本,将其送入卷积神经网络进行特征提取,实现分类。在3种常用的高光谱数据集上进行实验,于Indian Pines数据集上取得了96.63%的总体分类精度。相比于其他算法,所提算法的分类精度高、速度快,而且避免了传统方法在特征提取和融合时出现的不稳定性,验证了其提取的谱空联合信息对高光谱图像具有更强的特征表达能力。
Abstract
To solve the problem that hyperspectral image data has many dimensions and it is difficult to extract spectral information and spatial information, a classification algorithm is proposed based on a hypergraph and a convolutional neural network. In this algorithm, the hypergraph is first constructed based on the spectral and spatial relationships among pixels in a hyperspectral image, and then a sample with spectral space joint features is constructed through this hypergraph, which is finally sent to the convolutional neural network for feature extraction and thus the classification is finally achieved. The experiment is performed on three most commonly used hyperspectral datasets and an overall classification accuracy of 96.63% on the Indian Pines dataset is achieved. Compared with other algorithms, the proposed algorithm has a high classification accuracy and a high speed, which avoids the instability in feature extraction and fusion by traditional methods. It is verified that the spectral space joint information extracted by the proposed algorithm has a strong feature expression of hyperspectral images.
刘玉珍, 蒋政权, 马飞, 张春华. 基于超图和卷积神经网络的高光谱图像分类[J]. 激光与光电子学进展, 2019, 56(11): 111007. Yuzhen Liu, Zhengquan Jiang, Fei Ma, Chunhua Zhang. Hyperspectral Image Classification Based on Hypergraph and Convolutional Neural Network[J]. Laser & Optoelectronics Progress, 2019, 56(11): 111007.