Hyperspectral image classification based on multi-scale hybrid convolutional network

被引:4
|
作者
Yang, Yun [1 ]
Zhou, Yao [1 ]
Chen, Jia-ning [1 ]
机构
[1] Shaanxi Univ Sci & Technol, Sch Elect Informat & Artificial Intelligence, Xian 710021, Peoples R China
基金
中国国家自然科学基金;
关键词
hyperspectral image; hybrid convolutional network; multi-scale features; attention mechanism;
D O I
10.37188/CJLCD.2022-0225
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
To solve the problems of uneven distribution of hyperspectral image data, insufficient spatial -spectral feature extraction, and network degradation caused by the increase of network layers, a hyperspectral image classification algorithm based on multi-scale hybrid convolutional network is proposed. Firstly, principal component analysis is applied to reduce the dimension of hyperspectral data. Then, the neighborhood extraction is applied to take all pixels in the neighborhood as a sample to supplement the corresponding spatial information. Next, an improved multi-scale hybrid convolutional network is applied to extract features from the preprocessed sample data, and the mixed domain attention mechanism is added to enhance the useful information in the spatial and spectral dimensions. Finally, the Softmax classifier is used to classify each pixel sample. The proposed model is tested on hyperspectral datasets of Indian Pines and Pavia University. Experiments show that the overall classification accuracy, average classification accuracy and Kappa coefficient can reach 0. 987 9, 0. 983 3, 0. 986 2 and 0. 999 0, 0. 996 9, 0. 998 6, respectively. Compared with other classification methods, this algorithm can extract the feature information of hyperspectral images more fully, and achieves better classification results.
引用
收藏
页码:368 / 377
页数:10
相关论文
共 21 条
  • [1] [代晶晶 Dai Jingjing], 2020, [地质学报, Acta Geologica Sinica], V94, P2520
  • [2] HU L, 2015, CHINA SCIENCEPAPER, V10, P197
  • [3] An effective feature selection method for hyperspectral image classification based on genetic algorithm and support vector machine
    Li, Shijin
    Wu, Hao
    Wan, Dingsheng
    Zhu, Jiali
    [J]. KNOWLEDGE-BASED SYSTEMS, 2011, 24 (01) : 40 - 48
  • [4] Spectral-Spatial Classification of Hyperspectral Imagery with 3D Convolutional Neural Network
    Li, Ying
    Zhang, Haokui
    Shen, Qiang
    [J]. REMOTE SENSING, 2017, 9 (01)
  • [5] Liao Jinlei, 2021, Sci. Technol. Eng., V21, P11656
  • [6] Hyperspectral image classification using Support Vector Machine
    Moughal, T. A.
    [J]. 6TH VACUUM AND SURFACE SCIENCES CONFERENCE OF ASIA AND AUSTRALIA (VASSCAA-6), 2013, 439
  • [7] Rajarajeswari P., 2020, Int. J. Psychosocial Rehabil., V24, P5068
  • [8] HybridSN: Exploring 3-D-2-D CNN Feature Hierarchy for Hyperspectral Image Classification
    Roy, Swalpa Kumar
    Krishna, Gopal
    Dubey, Shiv Ram
    Chaudhuri, Bidyut B.
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (02) : 277 - 281
  • [9] HYPERSPECTRAL IMAGE CLASSIFICATION BASED ON KNN SPARSE REPRESENTATION
    Song, Weiwei
    Li, Shutao
    Kang, Xudong
    Huang, Kunshan
    [J]. 2016 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2016, : 2411 - 2414
  • [10] Overview of hyperspectral image classification
    Yan J.-W.
    Chen H.-D.
    Liu L.
    [J]. Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2019, 27 (03): : 680 - 693