一种高光谱图像分类方法技术

本发明专利技术公开一种高光谱图像分类方法

【技术实现步骤摘要】
一种高光谱图像分类方法、系统、电子设备及介质


[0001]本专利技术涉及高光谱图像处理领域，特别是涉及一种高光谱图像分类方法
、
系统
、
电子设备及介质
。

技术介绍

[0002]目前存在的高光谱图像像素级分类方法主要有两类：一类是基于人工特征的；另一类是基于深度学习特征的
。
文献“Ramachandran P,ParmarN,VaswaniA,et al.Stand
‑
alone self
‑
attention invision models[J].Advances in neural information processing systems,2019,32.”公开了一种应用于视觉领域的基于自注意力机制模型架构，利用这个独立的注意力层建立了一个完全的注意力视觉模型，该模型在图像分类和物体检测方面都优于卷积基线，同时参数和计算效率都很高
。
该文献提出了独立的局部自我注意力模块的设计，提高了分类性能
。
然而，上述本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种高光谱图像分类方法，其特征在于，所述分类方法包括：对待分类高光谱图像应用因子分析进行降维，并在预设区域内以各像素点为中点从降维后的待分类高光谱图像中提取多个三维像素块；将多个所述三维像素块输入分类模型，得到待分类高光谱图像的分类结果；所述分类模型是通过训练集对深度学习网络进行训练得到的；所述深度学习网络包括依次连接的多个特征融合
‑
置换注意力机制模型
、
采样层和全连接层；所述特征融合
‑
置换注意力机制模型包括依次连接的特征融合模型和置换注意力机制；所述特征融合模型包括依次连接的二维卷积
、
多尺度残差上下文
Transformer
网络和拼接模块；所述训练集包括高光谱图像样本的多个三维像素块和所述高光谱图像样本对应的分类结果
。2.
根据权利要求1所述的高光谱图像分类方法，其特征在于，所述在预设区域内以各像素点为中点从降维后的待分类高光谱图像中提取多个三维像素块，具体包括：根据所述预设区域内各所述像素点的坐标，按照预设空间滑动窗口尺寸，从降维后的待分类高光谱图像中提取多个三维像素块
。3.
根据权利要求2所述的高光谱图像分类方法，其特征在于，当所述预设区域内一个所述像素点的坐标为
(a
，
β
)
且预设空间滑动窗口尺寸为
S
×
S
时，以坐标为
(a
，
β
)
的像素点为中心的三维像素块的宽度为
[a
‑
(S
‑
1)/2
，
a+(S
‑
1)/2]
，高度为
[
β
‑
(S
‑
1)/2
，
β
+(S
‑
1)/2]
，以坐标为
(a
，
β
)
的像素点为中心的三维像素块的通道维度与所述降维后的待分类高光谱图像的通道维度一致
。4.
根据权利要求1所述的高光谱图像分类方法，其特征在于，所述深度学习网络包括连接的两个所述特征融合
‑
置换注意力机制模型；将多个所述三维像素块输入第一个所述特征融合
‑
置换注意力机制模型中的特征融合模型，输出第一融合后特征图；应用置换注意力机制，对所述第一融合后特征图的空间维度的特征信息和通道维度的特征信息进行加权，得到初始校准后的特征图；将所述初始校准后的特征图输入第二个所述特征融合
‑
置换注意力机制模型中的特征融合模型，输出第二融合后特征图；应用置换注意力机制，...

【专利技术属性】
技术研发人员：丁晨，岳嘉豪，杨宇堃，陈有法，赵俊豪，李乾芃，徐耀阳，
申请(专利权)人：西安邮电大学，
类型：发明
国别省市：