红外技术, 2020, 42 (3): 264, 网络出版: 2020-04-13   

基于 3D卷积联合注意力机制的高光谱图像分类

Hyperspectral Image Classification Based on 3D Convolution Joint Attention Mechanism
作者单位
1 安徽大学电气工程与自动化学院计算智能与信号处理教育部重点实验室, 安徽合肥 230601
2 偏振光成像探测技术安徽省重点实验室, 安徽合肥 230031
摘要
由于高光谱图像存在较高的数据维数, 会给分类过程带来一些困难。为了提高分类的准确率, 提出了一种使用 3D卷积联合注意力机制的高光谱图像分类方法。首先, 将中心像素与周围相邻的其它像素进行配对, 可以通过配对构成多组新的像素对, 充分利用了像素之间的邻域相关性。接着, 将像素对放入 3D卷积联合注意力机制网络框架中进行分类, 它能够对高光谱图像中的特征进行选择性的学习。最后, 通过投票策略获得像素标签。实验是在两个真实的高光谱图像数据集上进行。结果表明, 所提出的方法充分挖掘了高光谱图像的光谱空间特征, 能有效地提高分类精度。
Abstract
The high data dimension of hyperspectral images causes difficulties in the classification process. To improve the accuracy of classification, a hyperspectral image classification method using a 3D convolution joint attention mechanism is proposed. First, by pairing the center pixel with other pixels adjacent to it, it can form multiple sets of new pixel-pairs, and the neighborhood correlation between the pixels can be fully utilized. Then, the pixel-pairs are classified into the 3D convolution joint attention mechanism network framework, which can selectively learn the features in the hyperspectral image. Finally, the pixel label is obtained through the voting strategy. An experiment was carried out on two real hyperspectral image datasets. The results show that the proposed method fully exploits the spectral–spatial features of hyperspectral images, and this can effectively improve the classification accuracy.

王浩, 张晶晶, 李园园, 王峰, 寻丽娜. 基于 3D卷积联合注意力机制的高光谱图像分类[J]. 红外技术, 2020, 42(3): 264. WANG Hao, ZHANG Jingjing, LI Yuanyuan, WANG Feng, XUN Lina. Hyperspectral Image Classification Based on 3D Convolution Joint Attention Mechanism[J]. Infrared Technology, 2020, 42(3): 264.

本文已被 1 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!