基于三维注意力与混合卷积的高光谱图像分类 |
| |
引用本文: | 赵晓枫,牛家辉,刘春桐,夏玉婷.基于三维注意力与混合卷积的高光谱图像分类[J].系统工程与电子技术,2023(9):2673-2680. |
| |
作者姓名: | 赵晓枫 牛家辉 刘春桐 夏玉婷 |
| |
作者单位: | 1. 火箭军工程大学导弹工程学院;2. 兵器发射理论与技术国家重点学科实验室 |
| |
基金项目: | 国家自然科学基金(41404022)资助课题; |
| |
摘 要: | 针对现有高光谱图像分类模型在特征提取的过程中有效特征关注缺乏的问题,提出了一种基于三维卷积注意力与混合卷积的高光谱图像分类方法。该方法使用三维卷积和二维卷积串联完成对高光谱图像空谱特征的提取,并在三维卷积阶段引入注意力机制,使得模型在提取底层空谱特征的同时实现对有效特征的关注和激活。相对于传统三维卷积模型,提出的分类模型减小了运算复杂度,提升了模型噪声抑制能力,提高了分类效果。针对该方法的消融实验证明了提出的三维卷积注意力机制的有效性,在印第安松树林和帕维亚大学两个公开数据集上与其他5种分类模型的对比实验中取得了最优的分类精度。
|
关 键 词: | 高光谱图像分类 三维注意力机制 混合卷积 空谱特征提取 |
|
|