一种高光谱特征提取方法、装置及存储介质

文档序号:30750369发布日期:2022-07-13 08:48阅读:190来源:国知局
一种高光谱特征提取方法、装置及存储介质

1.本发明涉及遥感技术领域,尤其是指一种高光谱特征提取方法、装置及计算机存储介质。


背景技术:

2.在遥感领域,高光谱成像技术在多种研究中被广泛使用。高光谱图像包含了极其丰富的空间信息和光谱信息,其中空间信息指在各个光谱波长下单一像元的具体空间位置信息,光谱信息指单一像元在各个光谱波长下光谱反射率组成的连续光谱曲线。通过对高光谱图像进行特征提取研究,可以得到蕴含重要判别信息的嵌入特征,减少图像的冗余信息,并可以进一步提升后续分类研究中的识别精度。早期,高光谱图像特征提取方法主要提取像元的光谱特征,并没有考虑不同像元位置信息之间的差异,因此这些方法难以取得较好的分类结果。近年来,随着计算机的算力急剧提升,并伴随着深度学习研究的不断深入,大量针对本研究领域的神经网络方法相继被提出,这些方法中引入了多传感器数据融合的理论思想,对空间信息和光谱信息构建映射来进行提取,并最后实现特征融合,这类方法避免了重要高光谱信息的大量缺失,从而进一步提升了相关算法的分类性能。
3.高光谱特征提取方法从数据多样性的角度一般可分为基于光谱信息的特征提取方法和基于空-谱信息的特征提取方法。
4.基于光谱信息的高光谱特征提取方法主要利用高光谱图像中单一的光谱曲线来对特征提取模型进行构建,同时不考虑像元在空间层面的空间位置,属于单模态学习。早期较为常用的方法一般包括主成分分析(principlecomponent analysis,pca),最小化噪声分离(minimum noise fraction, mnf),线性判别分析(linear discriminant analysis,lda)等等。同时为了应对非线性数据,一些基于流形学习的方法也被提出,例如局部线性嵌入(local linear embedding,lle),拉普拉斯特征映射(laplacianeigenmaps,le),多维尺度分析(multidimensional scaling,mds)等等。此类方法一般只考虑高光谱像元内部的重要判别信息,以此来保障像元的分类能力。近年来,随着深度学习研究的不断深入,一些深度网络模型也被广泛应用到高光谱图像特征提取相关的研究中,例如自动编码器 (auto-encoder,ae)、变分自动编码器(variational auto-encoder,vae)、长短期记忆网络(long short term memory,lstm)等等。然而,此类方法没有考虑高光谱像元的空间位置关系,因此也仅从光谱层面对高光谱图像中的判别信息进行详细的描述,缺乏重要空间信息,无法充分发挥高光谱图像的“图谱合一”的优势,即图像中空间信息和光谱信息之间存在协同性。目前来说,本研究领域的主流高光谱特征提取方法是基于空-谱信息的特征提取方法。
5.高光谱图像是一种典型的三维立方体数据,包含两个空间维和一个光谱维。这种数据将目标地物像元的局部空间信息与各波长下的独有光谱信息进行交互式结合,并共同在完整的高光谱图像中进行体现,因此高光谱图像具有“图谱合一”的重要特点,即高光谱图像的空间信息与光谱信息具有一致性和协同性。同时,由于高光谱图像较为复杂多变的
拍摄场景,以及拍摄环境的信息干扰影响,高光谱图像中还会存在“一物异谱”和“一谱异物”的现象,这些现象也会严重影响干扰高光谱像元的分析结果。高光谱图像的空间信息是指单一像元在空间层面上的局部邻域像元组成的信息,通过假定每个光谱像元都和它的局部邻域像元存在一定的贡献关系或关联关系,可以实现对局部空间的学习,进而掌握像元在真实地物中的多重判别信息。目前的方法一般采用卷积神经网络对像元在局部感受野的信息进行卷积操作,获取像元的空间特征,并将其与其他特征相结合,实现多模态学习。但这类方法存在如下缺陷:

通常使用卷积神经网络等方式进行特征提取,却没有对信息中重要的区域进行强化,无法对局部信息的重要部分进行突出,而是采用一致的方式进行处理,难以突出信息重点;

无法挖掘空-谱信息中特有的序贯信息,且没有考虑序贯信息在不同序贯方向的差异,在对空-谱信息中的序贯信息进行提取时没有考虑序贯方向对提取效果的影响

使用注意力机制时,权重矩阵的构建方式过于单一

空间信息和光谱信息的融合方式过于简单,较为单一,无法充分实现模态信息(空间信息和光谱信息)的融合。


技术实现要素:

6.为此,本发明所要解决的技术问题在于克服现有技术中高光谱特征提取精度低的问题。
7.为解决上述技术问题,本发明提供了一种高光谱特征提取方法包括:选取高光谱图像进行归一化处理得到原域高光谱图像x
×y×
b,x和y是各波长下高光谱图像的空间尺寸,b是高光谱图像的波段个数,设置邻域滑动窗口 s,嵌入维数为d,对所述原域高光谱图像的每一个高光谱像元选取其周围尺寸为s
×
s的邻域像元作为该像元的空间数据,得到邻域高光谱空间数据集;
8.将所述原域高光谱图像和所述邻域高光谱空间数据集输入预先训练好的高光谱深度网络模型中;
9.将所述邻域高光谱空间数据集输入静态空间维序贯特征提取模块中,所述邻域高光谱空间数据集经过尺寸变换层后输入m个静态单元,所述静态单元包括双向长短期记忆层、第一平均池化层和静态注意力层,利用所述静态注意力层计算所述第一平均池化层输出结果的欧式权重矩阵和光谱权重矩阵,并分别进行归一化操作,将所述欧式权重矩阵和所述光谱权重矩阵按位相乘得到静态权重矩阵,利用所述静态权重矩阵计算得到输出并经过第二平均池化层后生成静态序贯特征图像;
10.将所述邻域高光谱空间数据集输入动态空间维序贯特征提取模块中,所述邻域高光谱空间数据集经过所述尺寸变换层和拼接层后输入m个动态单元,所述动态单元包括所述双向长短期记忆层、所述第一平均池化层和动态注意力层,利用所述动态注意力层计算所述第一平均池化层输出结果的权重矩阵,并进行归一化操作,得到动态权重矩阵,利用所述动态权重矩阵计算得到输出并经过所述第二平均池化层后生成动态序贯特征图像;
11.利用光谱特征提取模块提取所述原域高光谱图像的特征得到原域光谱特征图像;
12.将所述静态序贯特征图像、所述动态序贯特征图像和所述原域光谱特征图像融合得到最终光谱特征图像;
13.利用解码模块对所述最终光谱特征图像解码生成目标光谱特征图像。
14.优选地,所述预训练高光谱深度网络模型包括:
15.依据公式γ=γr+γ
kl
构建损失函数:
[0016][0017][0018]
其中,∑(
·
)为将括号内的内容全部加在一起,indata为所述原域高光谱图像,outdata为所述目标光谱特征图像,μ为所述融合特征图像,δ为第三原域光谱特征图像;
[0019]
选用步长为10-3
的adam优化器训练所述高光谱深度网络模型直至所述损失函数收敛。
[0020]
优选地,所述将所述邻域高光谱空间数据集输入静态空间维序贯特征提取模块中包括:
[0021]
将所述邻域高光谱空间数据集的x
×
y个尺寸为1
×s×s×
b的空间数据输入所述尺寸变换层,输出尺寸为1
×
s2×
b;
[0022]
将x
×
y个尺寸为1
×
s2×
b的空间数据输入所述静态单元的双向长短期记忆层,输出尺寸为1
×
s2×
2d;
[0023]
将x
×
y个尺寸为1
×
s2×
2d的空间数据输入所述静态单元的第一平均池化层,输出尺寸为1
×
s2×
d;
[0024]
将x
×
y个尺寸为1
×
s2×
d的空间数据输入所述静态单元的静态注意力层,变换尺寸得到尺寸为1
×s×s×
d的矩阵
[0025]
对所述矩阵的x
×
y个空间数据依据公式对每个所述空间数据位置在[i,j](1≤i≤s,1≤j≤s)的样本p
i,j
与中心像元之间的欧氏距离进行按位存储得到x
×
y个尺寸为1
×s×
s的所述欧式权重矩阵me;
[0026]
对x
×
y个空间数据依据公式对每个所述空间数据位置在 [i,j]的样本p
i,j
与中心样本之间的光谱距离进行按位存储得到x
×
y个尺寸为1
×s×
s的所述光谱权重矩阵ms;
[0027]
对每个所述欧式权重矩阵me和每个所述光谱权重矩阵ms进行归一化操作
[0028]
对每个所述欧式权重矩阵与对应的所述光谱权重矩阵进行按位相乘操作得到x
×
y个静态权重矩阵m=me×ms

[0029]
将x
×
y个尺寸为1
×s×
s的所述静态权重矩阵m的尺寸维度扩张为 1
×s×s×
1;
[0030]
计算输出x
×
y个其尺寸为1
×s×s×
d,转换尺寸为 1
×
s2×
d输入所述第二平均池化层生成所述静态序贯特征图像output1。
[0031]
优选地,所述将所述邻域高光谱空间数据集输入动态空间维序贯特征提取模块中包括:
[0032]
将所述邻域高光谱空间数据集的x
×
y个尺寸为1
×s×s×
b的空间数据输入所述尺寸变换层,输出尺寸为
[0033]
将x
×
y个尺寸为的空间数据输入所述拼接层,输出尺寸为
[0034]
将x
×
y个尺寸为的空间数据输入所述动态单元的双向长短期记忆层,输出尺寸为1
×k×
2d;
[0035]
将x
×
y个尺寸为1
×k×
2d的空间数据输入所述静态单元的第一平均池化层,输出尺寸为为1
×k×
d的矩阵h;
[0036]
将所述矩阵h输入所述动态单元的动态注意力层,对矩阵h的x
×
y个空间数据依据公式计算每个所述空间数据位置在[i,j](1≤i≤s, 1≤j≤s)的样本p
i,j
与所有样本的光谱距离之和,进行按位存储得到x
×
y个尺寸为1
×s×
s的动态权重矩阵n,并对每个所述动态权重矩阵n进行归一化操作;
[0037]
将x
×
y个尺寸为1
×s×
s的所述动态权重矩阵n的尺寸维度扩张为 1
×s×s×
1;
[0038]
计算输出x
×
y个h+n
×
h,其尺寸为1
×k×
d,输入所述第二平均池化层生成所述静态序贯特征图像output2。
[0039]
优选地,所述将所述静态序贯特征图像、所述动态序贯特征图像和所述原域光谱特征图像融合得到最终光谱特征图像包括:
[0040]
对所述静态序贯特征图像、所述动态序贯特征图像和所述第二原域光谱特征图像进行快速傅里叶变换并进行按位相乘和拼接操作,得到所述融合特征图像;
[0041]
将所述第一原域光谱特征图像输入一个全连接层得到第三原域光谱特征图像,利用所述第三原域光谱特征图像、所述融合特征图像以及随机生成的噪声矩阵,生成最终光谱特征图像。
[0042]
优选地,所述利用所述第三原域光谱特征图δ、融合特征图μ以及随机生成的噪声矩阵γ,生成最终光谱特征图output=μ+(δ
×
γ)。
[0043]
优选地,所述对所述光谱特征图像解码生成目标光谱特征图像包括:
[0044]
将尺寸为1
×
2d的所述光谱特征图像输入第一全连接层,输出为 x
×y×
d;
[0045]
将x
×y×
d输入剩余第2到第n-1个全连接层,输出为x
×y×
d;
[0046]
将x
×y×
d输入第n个全连接层,输出为尺寸为1
×
b的目标光谱特征图像outdata。
[0047]
本发明还提供了一种高光谱特征提取的装置,包括:
[0048]
预处理模块,用于选取高光谱图像进行归一化处理得到原域高光谱图像 x
×y×
b,x和y是各波长下高光谱图像的空间尺寸,b是高光谱图像的波段个数,设置邻域滑动窗口s,嵌入维数为d,对所述原域高光谱图像的每一个高光谱像元选取其周围尺寸为s
×
s的邻
域像元作为该像元的空间数据,得到邻域高光谱空间数据集;
[0049]
输入模块,用于将所述原域高光谱图像和所述邻域高光谱空间数据集输入预先训练好的高光谱深度网络模型中;
[0050]
静态序贯特征提取模块,用于将所述邻域高光谱空间数据集经过尺寸变换层后输入m个静态单元,所述静态单元包括双向长短期记忆层、第一平均池化层和静态注意力层,利用所述静态注意力层计算所述第一平均池化层输出结果的欧式权重矩阵和光谱权重矩阵,并分别进行归一化操作,将所述欧式权重矩阵和所述光谱权重矩阵按位相乘得到静态权重矩阵,利用所述静态权重矩阵计算得到输出并经过第二平均池化层后生成静态序贯特征图像;
[0051]
动态序贯特征提取模块,用于将所述邻域高光谱空间数据集经过所述尺寸变换层和拼接层后输入m个动态单元,所述动态单元包括所述双向长短期记忆层、所述第一平均池化层和动态注意力层,利用所述动态注意力层计算所述第一平均池化层输出结果的权重矩阵,并进行归一化操作,得到动态权重矩阵,利用所述动态权重矩阵计算得到输出并经过所述第二平均池化层后生成动态序贯特征图像;
[0052]
光谱特征提取模块,用于提取所述原域高光谱图像的特征得到原域光谱特征图像;
[0053]
特征融合拼接模块,用于将所述静态序贯特征图像、所述动态序贯特征图像和所述原域光谱特征图像融合得到最终光谱特征图像;
[0054]
解码模块,用于利用解码模块对所述最终光谱特征图像解码生成目标光谱特征图像。
[0055]
本发明还提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现上述一种高光谱特征提取的方法的步骤。
[0056]
本发明的上述技术方案相比现有技术具有以下优点:
[0057]

本发明针对空间层面中两种序贯信息分别构建不同的权重矩阵,实现了方法与样本特性间高适配性匹配的特点,权重矩阵的构造更为符合高光谱图像的数据特性,对空间层面中序贯信息的重要部分进行了强化

本发明利用双向长短期记忆网络和注意力机制对空间层面中两种序贯信息进行特征提取,减少了序贯信息中序贯方向对最终性能的影响。

本发明利用空间层面中序贯信息对光谱信息的修正,实现了特征提取的优化,解决了像元信息表达形式较为单一的问题,实现了多模态的特征提取方法

本发明将两种序贯特征与光谱特征进行信息修正融合,充分实现了模态信息(空间信息和光谱信息)的融合。本发明提出了一种基于多模态注意力编码的高光谱特征提取方法,通过构建空间维序贯特征提取模块实现对像元空间数据中两种序贯信息的有效提取,通过双向长短期记忆层来减少序贯特征提取过程中序贯方向对性能的影响,通过针对不同序贯信息来构建不同注意力权重矩阵实现对序贯信息中重要部分的强化,实现信息的充分融合,最终有效提升提取到的特征的表达能力。
附图说明
[0058]
为了使本发明的内容更容易被清楚的理解,下面根据本发明的具体实施例并结合附图,对本发明作进一步详细的说明,其中:
[0059]
图1是高光谱特征提取的实现流程图;
[0060]
图2是静态注意力层的示意图;
[0061]
图3是动态注意力层的示意图;
[0062]
图4是高光谱特征提取实验流程图。图5为本发明实施例提供的一种高光谱特征提取装置的结构框
具体实施方式
[0063]
本发明的核心是提供一种高光谱特征提取的方法、装置、设备及计算机存储介质,以提高光谱特征提取的精度。
[0064]
为了使本技术领域的人员更好地理解本发明方案,下面结合附图和具体实施方式对本发明作进一步的详细说明。显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
[0065]
请参考图1,图1为本发明所提供的高光谱特征提取的实现流程图;具体操作步骤如下:
[0066]
s101:选取高光谱图像进行归一化处理得到原域高光谱图像x
×y×
b,x 和y是各波长下高光谱图像的空间尺寸,b是高光谱图像的波段个数,设置邻域滑动窗口s,对所述原域高光谱图像的每一个高光谱像元选取其周围尺寸为 s
×
s的邻域像元作为该像元的空间数据,得到邻域高光谱空间数据集;
[0067]
归一化处理公式如下:
[0068][0069]
其中,x
min
表示数据中最小值,x
max
为最大值。
[0070]
设置邻域滑动窗口s为大于0的奇数,嵌入维数为d,d大于0,各网络层的节点数均为d;
[0071]
对于边缘的高光谱像元,对缺失的部分进行补零操作。
[0072]
获取全部x
×
y个尺寸为1
×s×s×
b的空间数据作为输入样本一,即邻域高光谱空间数据集,获取全部x
×
y个尺寸为1
×
b的高光谱像元作为输入样本二,即原域高光谱图像,其中输入样本一和输入样本二的个数相同,且一一对应,同一序号的输入样本二是输入样本一的中心。
[0073]
将全部x
×
y个输入样本一和全部x
×
y个输入样本二按照相同的顺序进行打乱,并选取序号处于前30%的输入样本一和输入样本二作为训练样本一和训练样本二,若x
×y×
30%不是整数,则将其进行四舍五入操作;
[0074]
从全部x
×y×
30%个训练样本一和全部x
×y×
30%个训练样本二中选择相同位置的小批量训练样本一和小批量训练样本二输入到神经网络中,小批量样本个数为512,将512个小批量训练样本一作为空间维序贯特征提取模块一和空间维序贯特征提取模块二的输入,将512个小批量训练样本二作为光谱特征提取模块的输入,对深度网络进行训练。
[0075]
依据公式γ=γr+γ
kl
构建损失函数:
[0076][0077][0078]
其中,∑(
·
)为将括号内的内容全部加在一起,indata为所述原域高光谱图像,outdata为所述目标光谱特征图像,μ为所述融合特征图像,δ为第三原域光谱特征图像,γr保障了深度网络输入信息与输出信息的一致性,γ
kl
是保障了高斯分布和嵌入特征分布的一致性;
[0079]
选用步长为10-3
的adam优化器训练所述高光谱深度网络模型直至所述损失函数收敛(需训练至少1000次)。
[0080]
s102:将所述原域高光谱图像和所述邻域高光谱空间数据集输入预先训练好的高光谱深度网络模型中;
[0081]
构建高光谱深度网络模型,模型中包含静态空间维序贯特征提取模块、动态空间维序贯特征提取模块、光谱特征提取模块和解码模块。
[0082]
s103:将所述邻域高光谱空间数据集输入静态空间维序贯特征提取模块中,所述邻域高光谱空间数据集经过尺寸变换层后输入m个静态单元,所述静态单元包括双向长短期记忆层、第一平均池化层和静态注意力层,利用所述静态注意力层计算所述第一平均池化层输出结果的欧式权重矩阵和光谱权重矩阵,并分别进行归一化操作,将所述欧式权重矩阵和所述光谱权重矩阵按位相乘得到静态权重矩阵,利用所述静态权重矩阵计算得到输出并经过第二平均池化层后生成静态序贯特征图像;
[0083]
尺寸变换层用于将一定维度的多维矩阵重新排列构造一个新的保持同样元素数量但是不同维度尺寸的矩阵,仅改变矩阵的形状,平均池化层指通过计算平均值的方式减少矩阵某一个或多个维度,将这一个或多个维度的维数压缩为原来的其中α是池化系数。
[0084]
双向长短期记忆层是深度学习领域的已有模型,包括两个序贯方向的长短期记忆层,减少序贯方向对特征映射结果造成的影响,双向长短期记忆层使用的激活函数为tanh激活函数:
[0085]
静态空间维序贯特征提取模块的输入尺寸为1
×s×s×
b,该模块由一个尺寸变换层(reshape layer)、m个静态单元和第二平均池化层(averagepooling layer)组成,尺寸变换层的输入尺寸为1
×s×s×
b,输出尺寸为 1
×
s2×
b,第一个静态单元的输入尺寸为1
×
s2×
b,输出尺寸为1
×
s2×
d,剩余m-1个静态单元的输入尺寸为1
×
s2×
d,输出尺寸为1
×
s2×
d,平均池化层的输入尺寸为1
×
s2×
d,针对第二个维度的池化系数为s2,输出为 output1,其尺寸为1
×
d。
[0086]
一个静态单元由一个双向长短期记忆层(bidirectional long short termmemory layer)、第一平均池化层和一个静态注意力层组成。其中双向长短期记忆层的节点数为d,输入尺寸与其所在的静态单元一致,输出尺寸为 1
×
s2×
2d。平均池化层的输入尺寸为1
×
s2×
2d,针对第三个维度的池化系数为2,输出尺寸为1
×
s2×
d,静态注意力层的输
入尺寸为1
×
s2×
d,输出尺寸为1
×
s2×
d。
[0087]
s104:将所述邻域高光谱空间数据集输入动态空间维序贯特征提取模块中,所述邻域高光谱空间数据集经过所述尺寸变换层和拼接层后输入m个动态单元,所述动态单元包括所述双向长短期记忆层、所述第一平均池化层和动态注意力层,利用所述动态注意力层计算所述第一平均池化层输出结果的权重矩阵,并进行归一化操作,得到动态权重矩阵,利用所述动态权重矩阵计算得到输出并经过所述第二平均池化层后生成动态序贯特征图像;
[0088]
动态空间维序贯特征提取模块的输入尺寸为1
×s×s×
b,该模块由一个尺寸变换层、一个拼接层(concatenation layer)、m个动态单元和第二平均池化层组成,其中尺寸变换层的输入尺寸为1
×s×s×
b,输出尺寸为拼接层的输入尺寸为输出尺寸为第一个动态单元的输入尺寸为输出尺寸为1
×k×
d,剩余m-1个动态单元的输入尺寸为1
×k×
d,输出尺寸为1
×k×
d,平均池化层的输入尺寸为1
×k×
d,针对第二个维度的池化系数为k,输出为output2,其尺寸为 1
×
d。
[0089]
一个动态单元由一个双向长短期记忆层、第一平均池化层和一个动态注意力层组成。其中双向长短期记忆层的节点数为d,输入尺寸与其所在的动态单元一致,输出尺寸为1
×k×
2d,平均池化层的输入尺寸为1
×k×
2d,针对第三个维度的池化系数为2,输出尺寸为1
×k×
d。动态注意力层的输入尺寸为1
×k×
d,输出尺寸为1
×k×
d。
[0090]
s105:利用光谱特征提取模块提取所述原域高光谱图像的特征得到原域光谱特征图像;
[0091]
光谱特征提取模块由m个全连接层组成,全连接层使用的激活函数为 tanh激活函数,第一个全连接层的输入尺寸为1
×
b,输出尺寸为1
×
d,剩余 m-1个全连接层的输入尺寸为1
×
d,输出尺寸为1
×
d;
[0092]
将所述原域高光谱图像x
×y×
b输入第一全连接层,输出为x
×y×
d;
[0093]
将x
×y×
d输入剩余m-1个全连接层,第m-1个全连接层的输出为第一原域光谱特征图toutput,第m个全连接层的输出为第二原域光谱特征图 output3。
[0094]
s106:将所述静态序贯特征图像、所述动态序贯特征图像和所述原域光谱特征图像融合得到最终光谱特征图像;
[0095]
s107:利用解码模块对所述最终光谱特征图像解码生成目标光谱特征图像。
[0096]
本发明的两种空间信息是用lstm网络提取到的两种连续信息,也叫序贯信息,主要是空间信息中的两种连续性。
[0097]
本发明针对空间层面中两种序贯信息分别构建不同的权重矩阵,实现了方法与样本特性间高适配性匹配的特点,权重矩阵的构造更为符合高光谱图像的数据特性,对空间层面中序贯信息的重要部分进行了强化。本发明利用双向长短期记忆网络和注意力机制对空间层面中两种序贯信息进行特征提取,减少了序贯信息中序贯方向对最终性能的影响。本发明利用空间层面中序贯信息对光谱信息的修正,实现了特征提取的优化,解决了像元信息表达形式较为单一的问题,实现了多模态的特征提取方法。本发明将两种序贯特征与
光谱特征进行信息修正融合,充分实现了模态信息(空间信息和光谱信息) 的融合。本发明提出了一种基于多模态注意力编码的高光谱特征提取方法,通过构建空间维序贯特征提取模块实现对像元空间数据中两种序贯信息的有效提取,通过双向长短期记忆层来减少序贯特征提取过程中序贯方向对性能的影响,通过针对不同序贯信息来构建不同注意力权重矩阵实现对序贯信息中重要部分的强化,实现信息的充分融合,最终有效提升提取到的特征的表达能力。
[0098]
基于以上实施例,本实施例对步骤s103和s104进行进一步详细说明,具体如下:
[0099]
参考图2:
[0100]
s131:将所述邻域高光谱空间数据集的x
×
y个尺寸为1
×s×s×
b的空间数据输入所述尺寸变换层,输出尺寸为1
×
s2×
b;
[0101]
s132:将x
×
y个尺寸为1
×
s2×
b的空间数据输入所述静态单元的双向长短期记忆层,输出尺寸为1
×
s2×
2d;
[0102]
s133:将x
×
y个尺寸为1
×
s2×
2d的空间数据输入所述静态单元的第一平均池化层,输出尺寸为1
×
s2×
d;
[0103]
s134:将x
×
y个尺寸为1
×
s2×
d的空间数据输入所述静态单元的静态注意力层,变换尺寸得到尺寸为1
×s×s×
d的矩阵
[0104]
s135:对所述矩阵的x
×
y个空间数据依据公式对每个所述空间数据位置在[i,j](1≤i≤s,1≤j≤s)的样本p
i,j
与中心像元之间的欧氏距离进行按位存储得到x
×
y个尺寸为1
×s×
s的所述欧式权重矩阵me;
[0105]
s136:对x
×
y个空间数据依据公式对每个所述空间数据位置在[i,j]的样本p
i,j
与中心样本之间的光谱距离进行按位存储得到 x
×
y个尺寸为1
×s×
s的所述光谱权重矩阵ms;
[0106]
其中|
·
|2是计算向量各元素平方和再取平方根.
[0107]
s137:对每个所述欧式权重矩阵me和每个所述光谱权重矩阵ms进行归一化操作其中,x
min
表示矩阵中最小值,x
max
表示矩阵中最大值;
[0108]
s138:对每个所述欧式权重矩阵与对应的所述光谱权重矩阵进行按位相乘操作得到x
×
y个静态权重矩阵m=me×ms

[0109]
s139:将x
×
y个尺寸为1
×s×
s的所述静态权重矩阵m的尺寸维度扩张为1
×s×s×
1;
[0110]
s1310:计算输出x
×
y个其尺寸为1
×s×s×
d,转换尺寸为 1
×
s2×
d输入所述第二平均池化层生成所述静态序贯特征图像output1。
[0111]
参考图3:
[0112]
s141:将所述邻域高光谱空间数据集的x
×
y个尺寸为1
×s×s×
b的空间数据输入
所述尺寸变换层,输出尺寸为
[0113]
s142:将x
×
y个尺寸为的空间数据输入所述拼接层,输出尺寸为
[0114]
s143:将x
×
y个尺寸为的空间数据输入所述动态单元的双向长短期记忆层,输出尺寸为1
×k×
2d;
[0115]
s144:将x
×
y个尺寸为1
×k×
2d的空间数据输入所述静态单元的第一平均池化层,输出尺寸为为1
×k×
d的矩阵h;
[0116]
s145:将所述矩阵h输入所述动态单元的动态注意力层,对矩阵h的x
×
y 个空间数据依据公式计算每个所述空间数据位置在[i,j](1≤i≤ s,1≤j≤s)的样本p
i,j
与所有样本的光谱距离之和,进行按位存储得到x
×
y个尺寸为1
×s×
s的动态权重矩阵n,并对每个所述动态权重矩阵n进行归一化操作;
[0117]
s146:将x
×
y个尺寸为1
×s×
s的所述动态权重矩阵n的尺寸维度扩张为 1
×s×s×
1;
[0118]
s147:计算输出x
×
y个h+n
×
h,其尺寸为1
×k×
d,输入所述第二平均池化层生成所述静态序贯特征图像output2。
[0119]
根据权利要求1所述的高光谱特征提取方法,其特征在于,所述提取所述原域高光谱图像的特征得到原域光谱特征图像包括:
[0120]
将所述原域高光谱图像x
×y×
b输入第一全连接层,输出为x
×y×
d;
[0121]
将x
×y×
d输入剩余m-1个全连接层,第m-1个全连接层的输出为第一原域光谱特征图toutput,第m个全连接层的输出为第二原域光谱特征图 output3。
[0122]
本发明所提供的高光谱特征提取方法在针对空间层面中第一种序贯信息进行权重矩阵构建时,既考虑像元之间的光谱距离又考虑像元之间的欧氏距离,且采用二者折中的方式实现的权重矩阵的构建;在针对空间层面中第二种序贯信息进行权重矩阵构建时,重点考虑每一个小样本与整个序贯样本的平均光谱距离,重点突出它在数据中的局部相关性,有效提升重要信息的权重。
[0123]
基于以上实施例,本实施例对步骤s106进行详细说明,具体如下:
[0124]
s161:对所述静态序贯特征图像output1、所述动态序贯特征图像output2和所述第二原域光谱特征图像output3进行快速傅里叶变换并进行按位相乘和拼接操作,得到所述融合特征图像μ;
[0125]
对output1、output2和output3进行快速傅里叶变换,分别得到进行快速傅里叶变换,分别得到和对和进行按位相乘计算并得到对和进行按位相乘计算并得到对和进行快速傅里叶逆变换分别得到output
13
和output
23
,二者尺寸均为 1
×
d,利用一个拼接层
对二者进行拼接,得到尺寸为1
×
2d的输出μ。
[0126]
s162:将所述第一原域光谱特征图像输入一个全连接层得到第三原域光谱特征图像,利用所述第三原域光谱特征图像δ、所述融合特征图像μ以及随机生成的噪声矩阵γ,生成最终光谱特征图像output=μ+(δ
×
γ);
[0127]
γ为随机生成的噪声矩阵,其符合高斯分布,且尺寸为1
×
2d。
[0128]
快速傅里叶变换和快速傅里叶逆变换为信号处理领域的常用定义,指利用计算机计算离散傅里叶变换和离散傅里叶逆变换的高效、快速计算方法;按位相乘计算指将两个形状相同的矩阵或向量的相同位置的元素进行乘法计算,最终得到同样形状的矩阵或向量。
[0129]
本发明通过信息修正融合,将两种序贯特征与光谱特征进行融合,利用快速傅里叶变换的特性达到高效的目的,实现信息的高性能融合,最终有效提升提取到的特征的表达能力。
[0130]
基于以上实施例,本实施例提取高光谱图像中空-谱特征,并用于后续分类研究,采用印第安纳州森林数据集(indian pines dataset)为例,具体如下:
[0131]
参考图4:
[0132]
输入:输入的高光谱图像是大小为145
×
145
×
200的图像,共包含21025 个像元,每个像元包含200个波段,整个数据集共含有16个有效类别和背景噪声类别,将属于背景噪声类别的像元去掉后,共剩余10366个有效像元。
[0133]
参数设定:邻域滑动窗尺寸为7,静态空间维序贯特征提取模块和动态空间维序贯特征提取模块中单元个数均为3,光谱特征提取模块中网络层个数为 3,解码模块中网络层个数为3,嵌入维数为40。
[0134]
邻域信息选取:针对每一个像元,得到大小为7
×7×
200的邻域信息。
[0135]
训练此高光谱深度网络模型:
[0136]
利用10366个有效像元可得到10366个输入样本一和10366个输入样本二,这些样本随机排序并选择30%的样本用于训练深度网络模型,小批量样本个数为512,每次训练只用一份小批量样本。在训练结束后,将全部10366个输入样本一和全部10366个输入样本二输入深度模型进行测试,得到尺寸为 10366
×
40的嵌入特征,最后利用svm分类器进行分类。随机选择10%的样本来训练svm分类器,并使用剩余的90%样本进行测试,最终得到分类结果,选用总体分类精度和平均分类精度来对分类结果进行评价。总体分类结果指所有样本中分类正确的样本数目除以全部样本数目的比值。平均分类精度首先计算每一类中分类正确的样本数目除以该类样本的数目的比值,并求取各类比值的平均值。
[0137]
用本技术提出的基于多模态注意力编码的高光谱特征提取方法和基于普通变分自动编码的高光谱特征提取方法(包含编码器和解码器,其中编码器由3层全连接层组成,解码器由3层全连接层组成,网络层节点个数均与本技术实施的方法相同)所得到的分类结果如下表所示。
[0138][0139]
从表可以看出,本技术的方法可以较好地提升嵌入特征的分类性能,具有更少的错分样本。由此可见,本技术方法具有较强的特征提取能力。因此本技术方法可以有效提升嵌入特征的可分类能力和分类精度。
[0140]
请参考图5,图5为本发明实施例提供的一种高光谱特征提取装置的结构框图;具体装置可以包括:
[0141]
预处理模块100,用于选取高光谱图像进行归一化处理得到原域高光谱图像x
×y×
b,x和y是各波长下高光谱图像的空间尺寸,b是高光谱图像的波段个数,设置邻域滑动窗口s,嵌入维数为d,对所述原域高光谱图像的每一个高光谱像元选取其周围尺寸为s
×
s的邻域像元作为该像元的空间数据,得到邻域高光谱空间数据集;
[0142]
输入模块200,用于将所述原域高光谱图像和所述邻域高光谱空间数据集输入预先训练好的高光谱深度网络模型中;
[0143]
静态序贯特征提取模块300,用于将所述邻域高光谱空间数据集经过尺寸变换层后输入m个静态单元,所述静态单元包括双向长短期记忆层、第一平均池化层和静态注意力层,利用所述静态注意力层计算所述第一平均池化层输出结果的欧式权重矩阵和光谱权重矩阵,并分别进行归一化操作,将所述欧式权重矩阵和所述光谱权重矩阵按位相乘得到静态权重矩阵,利用所述静态权重矩阵计算得到输出并经过第二平均池化层后生成静态序贯特征图像;
[0144]
动态序贯特征提取模块400,用于将所述邻域高光谱空间数据集经过所述尺寸变换层和拼接层后输入m个动态单元,所述动态单元包括所述双向长短期记忆层、所述第一平均池化层和动态注意力层,利用所述动态注意力层计算所述第一平均池化层输出结果的权重矩阵,并进行归一化操作,得到动态权重矩阵,利用所述动态权重矩阵计算得到输出并经过所述第二平均池化层后生成动态序贯特征图像;
[0145]
光谱特征提取模块500,用于提取所述原域高光谱图像的特征得到原域光谱特征图像;
[0146]
特征融合拼接模块600,用于将所述静态序贯特征图像、所述动态序贯特征图像和所述原域光谱特征图像融合得到最终光谱特征图像;
[0147]
解码模块700,用于利用解码模块对所述最终光谱特征图像解码生成目标光谱特征图像。
[0148]
本实施例的高光谱特征提取装置用于实现前述的高光谱特征提取方法,因此高光谱特征提取装置中的具体实施方式可见前文高光谱特征提取方法的实施例部分,例如,预处理模块100,输入模块200,静态序贯特征提取模块 300,动态序贯特征提取模块400,光谱特征提取模块500,特征融合拼接模块600,解码模块700,分别用于实现上述高光谱特征提取方法中步骤s101, s102,s103,s104,s105,s106和s107,所以,其具体实施方式可以参照
相应的各个部分实施例的描述,在此不再赘述。
[0149]
本发明具体实施例还提供了一种高光谱特征提取的设备,包括:存储器,用于存储计算机程序;处理器,用于执行所述计算机程序时实现上述一种高光谱特征提取的方法的步骤。
[0150]
本发明具体实施例还提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现上述一种高光谱特征提取的方法的步骤。
[0151]
本领域内的技术人员应明白,本技术的实施例可提供为方法、系统、或计算机程序产品。因此,本技术可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且,本技术可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、cd-rom、光学存储器等)上实施的计算机程序产品的形式。
[0152]
本技术是参照根据本技术实施例的方法、设备(系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。
[0153]
这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中,使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品,该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。
[0154]
这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上,使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理,从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。
[0155]
显然,上述实施例仅仅是为清楚地说明所作的举例,并非对实施方式的限定。对于所属领域的普通技术人员来说,在上述说明的基础上还可以做出其它不同形式变化或变动。这里无需也无法对所有的实施方式予以穷举。而由此所引伸出的显而易见的变化或变动仍处于本发明创造的保护范围之中。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1