System.ArgumentOutOfRangeException: 索引和长度必须引用该字符串内的位置。 参数名: length 在 System.String.Substring(Int32 startIndex, Int32 length) 在 zhuanliShow.Bind() 基于超像素自注意力机制的高光谱遥感图像分类方法技术_技高网

基于超像素自注意力机制的高光谱遥感图像分类方法技术

技术编号:41396826 阅读:8 留言:0更新日期:2024-05-20 19:20
本发明专利技术公开了基于超像素自注意力机制的高光谱遥感图像分类方法,包括:对输入高光谱遥感图像进行PCA降维处理,选取出降维图像的第一主成分;基于第一主成分进行SLIC超像素分割标记,得到多个尺度的超像素块;输入到超像素注意力网络分类模型,该模型包括超像素间注意力模块、超像素内注意力模块、双分支特征融合与分类网络;分别获得图像的两种像素特征表示,并通过双分支特征融合与分类网络进行特征表示融合与预测分类,输出像素标签类别;设计超像素注意力网络分类模型的损失函数并进行训练;将测试集输入到训练好的超像素注意力网络分类模型,获得符合要求的测试分类精度,完成高光谱遥感图像的分类。

【技术实现步骤摘要】

本专利技术涉及高光谱遥感图像处理,具体涉及基于超像素自注意力机制的高光谱遥感图像分类方法


技术介绍

1、高光谱遥感图像在许多领域都有广泛的应用,如军事目标检测与识别、城市规划与土地利用、生态环境与自然资源以及地质勘测等,有着十分重要的作用。高光谱图像由于具有丰富的光谱信息,可以捕捉到地物的空间分布和光谱反射特征,更能够提供更全面、更细致的地物信息,通过分析高光谱遥感图像,可以获取地物的光谱特征、形态特征、结构特征等多方面的信息,有助于更深入地理解和解释地物的属性和特征。但同时高光谱遥感图像分类也存在以下难点:1、高光谱遥感图像通常由数十甚至上百个连续谱段组成,每个谱段对应一个波段。这导致数据维度非常高,某些情况下甚至维度远大于样本数量,带来大量冗余信息。2、高光谱遥感图像中的像素往往是由多个地物的光谱组合而成的,即存在光谱混合现象,这种混合使得每个像素的光谱不纯净,使得分类任务更加复杂。3、在高光谱遥感图像中,不同地物类别的分布可能不均衡,即某些类别的样本数量较少。这会导致分类器倾向于较多样本的类别,而对少样本的类别性能较差。4、选择规则的窗口作为输入可能导致细节信息被忽略或模糊,无法充分捕捉像素之间的空间关系。

2、近年来,深度学习在高光谱遥感图像分类领域取得了显著的进展,为高光谱数据的特征提取和分类提供了新的方法和思路。现有的深度学习分类方法有卷积神经网络、循环神经网络、自编码器以及生成对抗网络等。这些深度学习方法在高光谱图像分类中取得了较好的结果,但也面临一些挑战,如标注困难、过拟合和样本不平衡等。

<br/>

技术实现思路

1、本专利技术目的:在于提供基于超像素自注意力机制的高光谱遥感图像分类方法,针对输入特征不规则和不同类别的样本不平衡,提供一种能够充分表征图像特征和克服样本数量不平衡的分类模型,以提高深度学习模型的分类准确率。

2、为实现以上功能,本专利技术设计基于超像素自注意力机制的高光谱遥感图像分类方法,针对高光谱遥感图像,执行如下步骤s1-步骤s5,完成高光谱遥感图像的分类:

3、步骤s1:采集包含各种类地物的高光谱遥感图像,对高光谱遥感图像进行降维和提取主成分处理,将降维后所提取的第一主成分,作为超像素分割的基础;

4、步骤s2:针对所提取的第一主成分,采用slic方法对第一主成分进行超像素分割,得到多个不同尺度的超像素块,以所得到的超像素块构建训练集和测试集;

5、步骤s3:构建超像素注意力网络分类模型,模型包括超像素间注意力模块、超像素内注意力模块、双分支特征融合与分类网络;将步骤s2分割获得的超像素块分别输入超像素间注意力模块、超像素内注意力模块,其中超像素间注意力模块基于注意力机制,计算不同超像素块之间的特征联系,经过超像素和像素信息的交互,获得像素特征表示x1;超像素内注意力模块根据输入的超像素块,计算块内像素间的关联度,得到像素特征表示x2;将像素特征表示x1、x2输入到双分支特征融合与分类网络进行特征融合,并进行像素所属地物种类的标签预测,输出像素所属地物种类的预测标签;

6、步骤s4:将训练集输入到超像素注意力网络分类模型中,根据损失函数,利用adam优化器优化模型中需要迭代和更新的参数,直至超像素注意力网络分类模型达到预设迭代次数,训练完成;

7、步骤s5:针对训练好的超像素注意力网络分类模型,利用测试集进行测试,直至测试集分类精度达到要求,应用超像素注意力网络分类模型,完成高光谱遥感图像中各像素所属地物的分类。

8、有益效果:相对于现有技术,本专利技术的优点包括:

9、1、本专利技术通过超像素分割法slic对原始图像进行超像素分割,提取像素局部空间特征,将超像素块分别输入超像素间注意力模块和超像素内注意力模块,对超像素块和像素之间的特征关系与超像素块内像素之间的特征关系进行建模,将得到的两种像素特征表示输入双分支特征融合与分类网络进行融合与分类,将融合特征输入mlp中进行更深层次的学习和表征,利用softmax函数计算融合特征属于某个类别的概率,选取概率最高的类别作为模型的预测结果。重复上述步骤,直到达到最大迭代次数,生成分类结果。

10、2、本专利技术通过设计超像素注意力网络分类模型来度量样本间的相关性,为深度学习领域的分类提供模型帮助。试验结果表明,在公开的高光谱分类数据集上,随机挑选占总样本数量百分之五的样本比例进行训练,以svm为基准模型的总体精度为94.47%,通过本模型训练后的样本分类总体精度为99.45%,相比之下总体分类精度提高了4.98%。

本文档来自技高网
...

【技术保护点】

1.基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,针对高光谱遥感图像,执行如下步骤S1-步骤S5,完成高光谱遥感图像的分类:

2.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S1中采用主成分分析法PCA将高光谱遥感图像进行降维处理,获得多个主成分,提取第一个主成分,作为超像素分割的基础。

3.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S2中采用SLIC方法对第一主成分进行超像素分割,得到多个不同的超像素块,将每个超像素块中的像素按照预设比例划分出训练集和测试集。

4.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S3中所构建的超像素注意力网络分类模型如下式:

5.根据权利要求4所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S3的具体步骤如下:

6.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S4中的损失函数计算像素特征表示X1和像素特征表示X2融合损失、超像素注意力网络分类模型的预测输出与实际标签之间的损失;对超像素注意力网络分类模型的优化器梯度归零,通过反向传播更新和优化训练参数,超像素注意力网络分类模型更新和迭代的训练参数有学习率l,迭代次数epoch、训练批次大小batch。

7.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤S5将测试集输入到训练好的超像素注意力网络分类模型中进行预测分类,直至预测分类精度达到精度要求,高光谱遥感图像分类完成。

...

【技术特征摘要】

1.基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,针对高光谱遥感图像,执行如下步骤s1-步骤s5,完成高光谱遥感图像的分类:

2.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤s1中采用主成分分析法pca将高光谱遥感图像进行降维处理,获得多个主成分,提取第一个主成分,作为超像素分割的基础。

3.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤s2中采用slic方法对第一主成分进行超像素分割,得到多个不同的超像素块,将每个超像素块中的像素按照预设比例划分出训练集和测试集。

4.根据权利要求1所述的基于超像素自注意力机制的高光谱遥感图像分类方法,其特征在于,步骤s3中所构建的超像素注意力网络分类模型如下式:

【专利技术属性】
技术研发人员:涂兵陈云云曹兆楼叶井飞王琳匡文剑
申请(专利权)人:南京信息工程大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1