In recent years, deep learning methods utilizing convolutional neural networks have been extensively employed in hyperspectral image classification (HSI) applications. Nevertheless, while a substantial number of stack...
详细信息
In recent years, deep learning methods utilizing convolutional neural networks have been extensively employed in hyperspectral image classification (HSI) applications. Nevertheless, while a substantial number of stacked 3dconvolutions can indeed achieve high classification accuracy, they also introduce a significant number of parameters to the model, resulting in inefficiency. Furthermore, such intricate models often exhibit limited classification accuracy when confronted with restricted sample data, i.e., small sample problems. Therefore, we propose a spectral-spatial double-branch network (SSdBN) with an attention mechanism for HSI classification. The SSdBN is designed with two independent branches to extract spectral and spatial features, respectively, incorporating multi-scale 2d convolution modules, long short-term memory (LSTM), and an attention mechanism. The flexible use of 2dconvolution, instead of 3dconvolution, significantly reduces the model's parameter count, while the effective spectral-spatial double-branch feature extraction method allows SSdBN to perform exceptionally well in handling small sample problems. When tested on 5%, 0.5%, and 5% of the Indian Pines, Pavia University, and Kennedy Space Center datasets, SSdBN achieved classification accuracies of 97.56%, 96.85%, and 98.68%, respectively. Additionally, we conducted a comparison of training and testing times, with results demonstrating the remarkable efficiency of SSdBN.
暂无评论