节点文献

融合注意力机制与混合CNN模型的高光谱图像分类

Hyperspectral Image Classification Based on Attention Mechanism and Hybrid CNN Model

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 王立国于佳石瑶杨京辉

【Author】 WANG Li-guo;YU Jia;SHI Yao;YANG Jing-hui;College of Information and Communication Engineering, Harbin Engineering University;School of Information and Communication Engineering, Dalian Minzu University;College of Information Engineering, China University of Geosciences (Beijing);

【机构】 哈尔滨工程大学信息与通信工程学院大连民族大学信息与通信工程学院中国地质大学(北京)信息工程学院

【摘要】 在基于深度学习的高光谱图像分类算法研究中,一些网络使用二维卷积,一些网络使用三维卷积,这两种算法都各有其优势和不足。针对基于深度学习的高光谱图像分类算法不足,提出一种融合注意力机制与混合CNN模型的算法,将高光谱图像中的空间和光谱的互补信息分别以三维卷积层和二维卷积层组合到一起。本文方法相比于经典SVM算法和目前几种性能优异的算法,在分类性能OA、AA以及Kappa系数上均有提升,证明了本算法通过结合二维卷积和三维卷积,并引入注意力机制的方法,对高光谱图像分类精度提升的有效性。

【Abstract】 In the research of hyperspectral image classification algorithms based on deep learning, some networks use 2 D-CNN, and some networks use 3 D-CNN. Both algorithms have their own advantages and shortcomings. Aiming at the shortcomings, this paper proposes an algorithm that combines the spatial information and the spectrum information in the hyperspectral image with 3 D-CNN and 2 D-CNN, and introduces an attention mechanism. Compared with the classic SVM algorithm and several excellent algorithms at present, the classification performance of the proposed method has improved in terms of OA, AA and Kappa coefficients, which proves that this algorithm combining 2 D-CNN, 3 D-CNN and attention mechanism is effective in improving the accuracy of hyperspectral image classification.

【基金】 国家自然科学基金项目(62071084,62001434)
  • 【文献出处】 大连民族大学学报 ,Journal of Dalian Minzu University , 编辑部邮箱 ,2022年01期
  • 【分类号】TP751;TP18
  • 【被引频次】1
  • 【下载频次】299
节点文献中: 

本文链接的文献网络图示:

本文的引文网络