Journal of Lanzhou University of Technology ›› 2024, Vol. 50 ›› Issue (2): 87-95.

• Automation Technique and Computer Technology • Previous Articles     Next Articles

Hyperspectral image classification based on improved SE-Net anddepth-separable residuals

WANG Yan, WANG Zhen-yu   

  1. College of Computer and Communication, Lanzhou Univ. of Tech., Lanzhou 730050, China
  • Received:2022-03-04 Online:2024-04-28 Published:2024-04-29

Abstract: In response to the challenges posed by convolutional neural network (CNNs) commonly used for hyperspectral image classification, namely, their high parameter count, extended training times, and sensitivity to sample quantity, a classification network MDSR&SE-Net based on improved squeeze and excitation network and depth-separable residuals was proposed for limited training samples. First, the principal component analysis is employed in this model to reduce the dimension of the original HSI. Then, the multi-feature residual structure is connected by 3D convolutional neural network, and the spatial & spectral details of hyperspectral images are extracted by embedding the improved squeeze and excitation block. Finally, the extracted feature information is input into Softmax classifier to activate classification. To further lightweight the network, the number of parameters is reduced by using the depth separable convolution in the residual structure and introducing global average pooling. Experimental results show that overall accuracy of the three common hyperspectral data sets with the limited training samples are above 99%.

Key words: hyperspectral image, depth separable convolution, residual network, SE Net

CLC Number: