Main Image
Volume 299 - The 7th International Conference on Computer Engineering and Networks (CENet2017) - Session I - Machine Learning
The Classification of Hyperspectral Images Based on Band-Grouping and Convolutional Neural Network
G. Fu,* H. Gu
*corresponding author
Full text: pdf
Pre-published on: 2017 July 17
Published on: 2017 September 06
Abstract
For the classification of hyperspectral images, a classification algorithm based on band grouping
and three-dimensional convolutional neural network(3D-CNN-BG) is proposed. The algorithm
uses the correlation matrix of hyperspectral images to determine the similarity of bands, and the
high similarity bands
are
grouped together. Then, every bands group is extracted spatial-spectral
feature using 3D convolutional neural network. Finally, the high-level feature of every 3D-CNN
is stacked together trained by the classifier. With band regroup, the feature in the similar group
can be extracted fully by the 3D-CNN, which is good for classification. The experiments
showed that compared with other hyperspectral images classification algorithms based on CNN,
the proposed algorithm converges more faster, has less training parameters, and acquires higher
classification accuracy. After training 100 times, the overall classification accuracy on the indian
pines data set can reach 97.42%, increased by approximately 2% ~ 5%, which shows the strong
practicability of proposed algorithm in hyperspectral application
DOI: https://doi.org/10.22323/1.299.0005
Open Access
Creative Commons LicenseCopyright owned by the author(s) under the term of the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.