Welcome to Journal of University of Chinese Academy of Sciences,Today is

›› 2020, Vol. 37 ›› Issue (2): 169-176.DOI: 10.7523/j.issn.2095-6134.2020.02.004

• Research Articles • Previous Articles     Next Articles

Hyperspectral image classification based on convolutional neural network and active learning algorithm

SONG Han, YANG Weitun, GENG Xiurui, ZHAO Yongchao   

  1. Institute of Electronics, Chinese Academy of Sciences, Beijing 100190, China;Key Laboratory of Technology in Geo-spatial Information Processing and Application System of Chinese Academy of Sciences, Beijing 100190, China;University of Chinese Academy of Sciences, Beijing 100049, China
  • Received:2018-12-20 Revised:2019-03-06 Online:2020-03-15

Abstract: When we apply the active learning method to the hyperspectral image classification, there always exist problems that the spectral-spatial features cannot be effectively utilized and the samples need to be manually labeled. In this work, we propose an active learning method combined with the convolutional neural networks to solve the above problems. Firstly, the neighborhood pixels of each pixel are extracted to form training samples. Then the spatial and spectral features of the samples are trained by the convolutional neural network and the data are preliminarily classified. Then, based on the spatial similarity and spectral similarity of the hyperspectral image, some of the unlabeled samples are labeled and added to the training set, which will further increase the accuracy of the classifier. Validated by Salinas image, Pavia University image, and Indian Pines image, the proposed method effectively improves the classification accuracy with less labeled samples.

Key words: hyperspectral image classification, active learning, convolutional neural network, spatial-spectral feature

CLC Number: