一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

基于动态超图卷积网络的高光谱图像分类方法及系统

2022-11-12 11:27:26 来源:中国专利 TAG:


1.本技术涉及深度学习及图像处理技术领域,特别涉及一种基于动态超图卷积网络的高光谱图像分类方法及系统。


背景技术:

2.随着高光谱实时分类的技术逐步成熟,应用领域也逐渐从航天遥测、农业、食品安全、医学诊断等领域拓展到工业领域的产品质量检测和分类方面。
3.高光谱遥感是成像技术和细分光谱技术有机结合的成像光谱遥感。高光谱图像是由高光谱遥感器系统通过获取、记录带有地物信息的太阳辐射信号得到的含有上百通道且波段连续的图像。高光谱图像含有丰富的光谱和空间信息,由于不同成分对光谱吸收程度不同,在某个特定波长下图像对某个缺陷会有较显著的反映,这使得高光谱图像不仅可以反映样本的大小、形状、缺陷等外部品质特征,还可以充分反映样品内部的物理结构、化学成分的差异。
4.通过搭载在不同空间平台上的高光谱传感器,即成像光谱仪,捕获到的高光谱图像具有图谱合一且光谱信息丰富的特点,与普通图像和多光谱图像不同,高光谱图像具有上百个光谱特征。高光谱图像的高维特性、波段间高度相关性、光谱混合等特点使得高光谱图像分类面临巨大挑战。
5.现有的高光谱图像分类的方法大多数是基于卷积神经网络的,而卷积神经网络通常需要足够多的标记样本进行训练,这对于高光谱图像分类是非常困难的。同时,大部分的卷积神经网络架构都是将图片进行分批输入网络,这会导致网络不能捕捉到远距离上的像素之间的关系。


技术实现要素:

6.鉴于以上所述现有技术的缺点,本技术的目的在于提供一种基于动态超图卷积网络的高光谱图像分类方法及系统,进而至少在一定程度上克服由于相关技术的限制和缺陷而导致的一个或者多个问题。
7.第一方面,本技术提供一种基于动态超图卷积网络的高光谱图像分类方法包括:
8.获取高光谱图像,并对所述高光谱图像进行图像预处理;
9.对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;
10.根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;
11.对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;
12.根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;
13.对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;
14.根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对
应原始像素点的光谱特征;
15.基于所述原始像素点的光谱特征对所述原始像素点进行分类。
16.可选地,对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点的步骤,包括:
17.采用简单线性迭代聚类对图像预处理后的所述高光谱图像中的原始像素点进行聚类分组,以生成若干超像素点。
18.可选地,根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图的步骤,具体包括:
19.根据图像预处理后的所述高光谱图像的原始像素点的光谱特征,计算获取每个所述超像素点的第一光谱特征,其中,每个所述超像素点的第一光谱特征为对应所述超像素点所包含的所有原始像素点的光谱特征的平均值;
20.基于所述超像素点的第一光谱特征的相似度构建第一超边,以获取超图关联矩阵,完成所述第一超图的构建。
21.可选地,在基于所述超像素点的第一光谱特征的相似度构建第一超边的步骤中,所述第一超边ei的定义公式如下:
[0022][0023]
其中,vi为一个超像素点节点,为与超像素点节点vi相似度满足预设阈值的若干其他超像素点节点组成的邻域。
[0024]
可选地,在对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征的步骤中,所述超图卷积的公式为:
[0025][0026]
其中,x表示输入的超像素点的光谱特征,h是超图关联矩阵,dv和de分别是节点度矩阵和超边度矩阵,θ是可训练的参数矩阵,y超图卷积输出的超像素点的光谱特征。
[0027]
可选地,在根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图的步骤中,根据所述超像素点的第二光谱特征,采用knn方法和k-means聚类方法构成第二超边。
[0028]
可选地,对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征的步骤,具体包括:
[0029]
对所述第二超图进行顶点卷积,以将每个所述超像素点的第二光谱特征聚合到所述第二超边,以获取超边特征;
[0030]
对所述超边特征进行超边卷积,以将邻近超边特征聚集到所述超像素点,以获取所述超像素点的第三光谱特征。
[0031]
可选地,所述对所述第二超图进行顶点卷积,以将每个所述超像素点的第二光谱特征聚合到所述第二超边,以获取超边特征的步骤,具体包括:
[0032]
对所述超像素点的第二光谱特征进行全局平均池化操作,以获取所述超像素点的第四光谱特征;
[0033]
在光谱维度上对所述超像素点的第四光谱特征进行一维卷积操作,并经过sigmoid激活函数得到各个光谱维度的权重矩阵;
[0034]
将权重矩阵与所述超像素点的第二光谱特征相乘,得到所述超像素点的第五光谱特征;根据所述超像素点的第五光谱特征,使用多层感知机学习以获取变换矩阵;
[0035]
使用所述变换矩阵对所述超像素点的第五光谱特征进行变换,并使用一维卷积对变换后的特征进行压缩,以获取所述超边特征。
[0036]
可选地,通过主成分分析或者卷积神经网络进行图像预处理。
[0037]
另一方面,本技术还提供一种基于动态超图卷积网络的高光谱图像分类装置,包括:
[0038]
图像降维模块,用于获取高光谱图像,并对所述高光谱图像进行图像预处理;
[0039]
像素分割模块,用对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;
[0040]
第一超图构造模块,用于根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;
[0041]
超图卷积模块,用于对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;
[0042]
第二超图构造模块,用于根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;
[0043]
动态图卷积模块,用于对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;
[0044]
特征转换模块,用于根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征;
[0045]
像素点分类模块,用于基于所述原始像素点的光谱特征对所述原始像素点进行分类。
[0046]
上述基于动态超图卷积网络的高光谱图像分类方法及系统,通过获取高光谱图像,并对所述高光谱图像进行图像预处理;对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征;基于所述原始像素点的光谱特征对所述原始像素点进行分类。通过两次超图构造,能够实现超像素点的光谱特征/原始像素点的光谱特征的动态提取,并且由于是将整个高光谱图像作为输入,而不是像多数cnn划分成块,这可以使网络更好的提取全局特征。同时,使用超像素分割技术将原始像素划分成超像素,大幅度减少了计算量,并且通过超像素的标签可以确定一些未知像素点的标签,相当于隐式增加了样本数量,这让本技术提出的模型在少量训练样本的情况下就可以达到较高的分类精度。
附图说明
[0047]
图1本技术的基于动态超图卷积网络的高光谱图像方法的一流程示意图。
[0048]
图2示出了本技术的网络架构示意图。
[0049]
图3示出了本技术的降维和维度重塑的示意图。
[0050]
图4示出了高光谱图像中的目标像素点和相似像素点的位置关系示意图。
[0051]
图5示出了顶点卷积的示意图。
[0052]
图6示出了超边卷积的示意图。
[0053]
图7a-7e分别示出了伪彩色图像、地面真实图、训练样本图、测试样本图及分类结构图。
[0054]
图8示出了本技术的基于动态超图卷积网络的高光谱图像装置的一功能模块图。
具体实施方式
[0055]
以下通过特定的具体实例说明本技术的实施方式,本领域技术人员可由本说明书所揭露的内容轻易地了解本技术的其他优点与功效。本技术还可以通过另外不同的具体实施方式加以实施或应用,本说明书中的各项细节也可以基于不同观点与应用,在没有背离本技术的精神下进行各种修饰或改变。
[0056]
请参阅图1-8。需要说明的是,本实施例中所提供的图示仅以示意方式说明本技术的基本构想,遂图式中仅显示与本技术中有关的组件而非按照实际实施时的组件数目、形状及尺寸绘制,其实际实施时各组件的型态、数量及比例可为一种随意的改变,且其组件布局型态也可能更为复杂。
[0057]
高光谱图像是由高光谱遥感器系统通过获取、记录带有地物信息的太阳辐射信号得到的含有上百通道且波段连续的图像。高光谱图像含有丰富的光谱和空间信息,由于不同成分对光谱吸收程度不同,在某个特定波长下图像对某个缺陷会有较显著的反映,这使得高光谱图像不仅可以反映样本的大小、形状、缺陷等外部品质特征,还可以充分反映样品内部的物理结构、化学成分的差异。这些特点决定了高光谱图像技术在精准农业领域的独特优势。
[0058]
精准农业是一种将信息技术与农业生产全面结合的新型农业。精准农业以信息技术为支撑,采用3s(gps、gis和rs)等高新技术与现代农业技术相结合,根据空间变异,定位、定时、定量地实施一整套现代化农事操作与管理的系统。精准农业可最大限度地提高农业生产力,是实现优质、高产、低耗和环保的可持续发展农业的有效途径。
[0059]
由于高光谱图像的高维特性、波段间高度相关性、光谱混合等特点使得高光谱图像分类面临巨大挑战。现有的高光谱图像分类的方法大多数是基于卷积神经网络的,而卷积神经网络通常需要足够多的标记样本进行训练,这对于高光谱图像分类是非常困难的。同时,大部分的卷积神经网络架构都是将图片进行分批输入网络,这会导致网络不能捕捉到远距离上的像素之间的关系。
[0060]
基于此,本技术公开一种基于动态超图卷积网络的高光谱图像分类方法,可应用于精准农业的高光谱图像分类。图1示出了本技术的一实施例的基于动态超图卷积网络的高光谱图像分类方法。如图1所示,所述基于动态超图卷积网络的高光谱图像分类方法,包括:
[0061]
s10:获取高光谱图像,并对所述高光谱图像进行图像预处理;
[0062]
s20:对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;
[0063]
s30:根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;
[0064]
s40:对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;
[0065]
s50:根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;
[0066]
s60:对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;
[0067]
s70:根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征;
[0068]
s80:基于所述原始像素点的光谱特征对所述原始像素点进行分类。
[0069]
需要说明的是,本实施例的基于动态超图卷积网络的高光谱图像分类方法除了可应用于精准农业领域,还可以应用于航天领域、食品安全、医学诊断、流水线工业分类质检、军事领域、地质勘测等领域。
[0070]
本实施例的基于动态超图卷积网络的高光谱图像分类方法通过图2所示的网络架构实现,下面将结合该网络架构来详细阐述本实施例的技术方案。
[0071]
首先,执行步骤s10:获取高光谱图像,并对所述高光谱图像进行图像预处理。
[0072]
由于高光谱图像与普通图像及高光谱图像不同,高光谱图像光谱波段具有上百维,将主成分分析(pca)应用于高光谱图像上,可以有效的去除光谱冗余波段,方便更好得提取光谱特征。如图3所示,通过pca降维操作,也即图像预处理操作,将原本高光谱图像的光谱波段数由c维降低至d维。
[0073]
需要说明的是,在其他实施例中,也可通过一维卷积神经网络(1d cnn)对所述高光谱图像进行图像预处理来去除高光谱图像中的冗余波段,方便更好得提取光谱特征。譬如利用bn和二维卷积(1
×
1的卷积核)实现)对所述高光谱图像进行图像预处理。
[0074]
其次,执行步骤s20:对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点。
[0075]
由于高光谱图像中的原始像素点有上千个,虽然我们可以将图像中的原始像素视为超图中的节点,但超图的构建需要邻接矩阵,这个矩阵是十分庞大的,这将大幅提高计算成本。
[0076]
为了解决这一问题,如图2和3所示,在申请的网络架构中,采用简单线性迭代聚类(slic)将图像预处理后所述高光谱图像的原始像素点进行分组。大致将n个原始像素点通过k-means聚类的方法生成一个超像素点,将slic生成的m个超像素点作为超图中的节点。
[0077][0078]
h和w分别表示高光谱图像的高度和宽度。超像素分割可以根据光谱和空间的加权欧氏距离将相似的像素分割成均匀的区域,分割后得到的超像素s={s1,s2,

sm}具有局部同质性。需要说明的是,实际处理时,每个超像素点中包含的原始像素的个数并不相同。
[0079]
通过超像素分割,每一个原始像素点都对应一个超像素点,而高光谱图像的最后分类是针对每一个原始像素点进行分类的,需要将得到的超像素点的光谱特征分配给每一个原始像素。
[0080]
由于不同的超像素包含不同数量的原始像素,很难通过网络直接计算,为此,可利用np.reshape对超像素分割后的高光谱图像进行维度重塑,将原三维高光谱图像中的长和
宽相乘转换为二维矩阵,如图3所示。从图3可以看出来维度重塑后的二维矩阵的维度分别是h
×
w和d,d是降维后的高光谱图像的光谱波段数,h
×
w的值表示高光谱图像中所有原始像素点的个数。
[0081]
编码器(encoder)和解码器(decoder)中定义了一个映射矩阵o∈r
hw
×m将超像素与原始像素进行相互转换。其中x∈r
hw
×d表示的是由3d的高光谱图像沿着光谱空间维度重塑成的2d矩阵。映射矩阵o的表示为
[0082][0083]
通过以下公式实现特征转换:
[0084][0085][0086]
其中是按列进行归一化的映射矩阵o,v表示由超像素点构成的节点,为经过特征提取后超像素垫构成的节点,表示经过解码器操作后得到的原始像素点。经过slic操作、维度重塑及编码器(encoder)后得到v,分割后的特征可以看作是一个无向图g=(v,e),其中v和e分别表示节点和边。
[0087]
接着,执行步骤s30:根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图。
[0088]
超图是一种广义的图模型,与所有边的度都是2的简单图相比,超图的特点是一条超边可以连接多个点。超图的描述比较自由化,在多模态和异构的情况下更易于扩展。超图可以结合邻接矩阵,对物体的高阶关系进行建模,这在许多计算机视觉任务中显示出强大的数据表示能力,如语义划分、图数据的分割、三维物体分类、多模态数据分析。超图中的超边可以连接多个顶点,这更有利于网络捕获到更多远距离相似节点的特征,这对高光谱图像分类是很有意义。
[0089]
在一示例性实施例中,根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图的步骤,具体包括:
[0090]
根据图像预处理后的所述高光谱图像的原始像素点的光谱特征,计算获取每个所述超像素点的第一光谱特征,其中,每个所述超像素点的第一光谱特征为对应所述超像素点所包含的所有原始像素点的光谱特征的平均值;
[0091]
基于所述超像素点的第一光谱特征的相似度构建第一超边,以获取超图关联矩阵,完成所述第一超图的构建。
[0092]
本技术将每一个超像素点作为超图的一个节点进行构图,超图的节点特征(也即对应超级像素点的第一光谱特征)是超像素点中所包含的所有像素特征的平均值。假设第i个超像素点是由j个原始像素构成的,每一个原始像素点的光谱向量d可以表示为d
α
∈r1×d(α=1,2,

,j)。节点的特征f
β
(也即对应超级像素点的第一光谱特征)计算公式为
[0093]
[0094]
高光谱图像的一个原始像素点的分类不仅可能受益于邻居像素点,还有可能是远端的像素点。与cnn卷积不同,cnn的感受野较小,无法获取远端像素的有效信息。而图结构不受限于距离,是将整张高光谱图像作为输入,而不是像大部分基于cnn的网络架构要分割成块再输入,这可以使网络更好的提取全局特征。图4说明了高光谱图像中的目标像素点和相似像素点的位置关系,这更能体现基于图结构的网络框架的优势。
[0095]
本技术通过构建超图模型进行特征提取。与普通图一条边只连接两个节点不同,超图中的每一条超边都可以连接多个节点。超边的这种多节点连接方式,可以更好得提取特征。每一个像素点可以与多个相似像素点相连接,提取的特征更全面。
[0096]
为了更好得获取高光谱图像的全局特征,利用knn方法,根据节点之间的光谱特征相似度构建超边。给定一个节点vi,选择与vi最相似的n个节点vj,这与目标节点相近,也即与目标节点相似度满足预设阈值的n个其他节点vj组成了节点vi的邻域超边ei的定义公式如下
[0097][0098]
从而每个节点对应一个超边,也即超边的个数与节点的个数一致,均为m。
[0099]
超图的关联矩阵h∈rm×m可以定义为
[0100][0101]
在构建超图时将每一条超边的权重都设为1,那么超边的权重矩阵w∈rm×m即为单位矩阵。节点度矩阵dv∈rm×m表示为
[0102][0103]
超边度矩阵de∈rm×m表示为
[0104][0105]
接着,执行步骤s40:对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征。
[0106]
超图卷积是一种基于超图模型的卷积方法。超图卷积能够通过超图结构计算复杂的高阶数据相关性,并通过信息传播来更新节点的特征。超图卷积通过节点-边缘-节点变换有效地提取超图的高阶相关性。
[0107]
本技术的架构将超像素点作为节点,用knn的方法获取超边后,得到了超图结构。为了进一步提取特征,我们在构建的超图上进行了两次(当然也可以是一次,三次及三次以上)超图卷积运算,最后一层超图卷积的输出值作为超像素点的第二光谱特征。
[0108]
对于一个超图g=(v,e,w),其中v和e分别表示为顶点集和超边集,对角矩阵w表示为超边权重矩阵。顶点的个数即为划分的超像素的个数m个,超图卷积的公式为
[0109][0110]
其中表示m个节点和c1维的特征,也即输入的超像素点的光谱特征,h∈rm
×m是超图关联矩阵,dv和de分别是节点度矩阵和超边度矩阵,是可训练的参数矩阵,在超图的节点上应用θ过滤器来提取特征。经过卷积操作得到也即y是超图卷积输出的超像素点的光谱特征。经过超图卷积操作节点的特征由转换成l表示层数(对应超图卷积的个数,在图2中取值为2),每一层的超图卷积公式为
[0111][0112]
接着,执行步骤s50:根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图。在第二超图的构造过程中,根据所述超像素点的第二光谱特征,采用knn方法和k-means聚类方法构成第二超边。
[0113]
经过两次超图卷积操作后,得到的顶点特征为x=[x1;x2;

;xm],其中xi(i=1,2,

,m表示的是第i个超像素顶点的光谱特征。通过图2中的动态超图模块dhg,利用这些顶点的特征,重新构建一个超图。在dhg模块中,我们采用knn方法和k-means聚类方法分别基于局部特征和全局特征更新超图结构。采用knn方法生成基本超边,并通过聚类算法扩展相邻超边集,即k-means聚类。对于knn方法,我们计算了每个顶点u和其k-1(k值可配置,例如可以取值为12,18等)个最邻近的顶点,这些邻近顶点与顶点u一起,构成了一条超边。对于k-means方法,我们根据欧氏距离将所有的顶点分成s个簇,对于每个顶点,最近的s-1簇将被分配为该顶点的相邻超边。通过动态超图构造方法可以提取局部和全局特征。
[0114]
接着,执行步骤s60:对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征。
[0115]
在一示例性实施例中,对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征的步骤,包括:
[0116]
对所述第二超图进行顶点卷积,以将每个所述超像素点的第二光谱特征聚合到所述第二超边,以获取超边特征。
[0117]
对所述超边特征进行超边卷积,以将邻近超边特征聚集到所述超像素点,以获取所述超像素点的第三光谱特征。
[0118]
在一示例性实施例中,在对所述第二超图进行顶点卷积,以将每个所述超像素点的第二光谱特征聚合到所述第二超边,以获取超边特征时,首先,对所述超像素点的第二光谱特征进行全局平均池化操作,以获取所述超像素点的第四光谱特征;接着,在光谱维度上对所述超像素点的第四光谱特征进行一维卷积操作,并经过sigmoid激活函数得到各个光谱维度的权重矩阵;将权重矩阵与所述超像素点的第二光谱特征相乘,得到所述超像素点的第五光谱特征;根据所述超像素点的第五光谱特征,使用多层感知机学习以获取变换矩阵;使用所述变换矩阵对所述超像素点的第五光谱特征进行变换,并使用一维卷积对变换后的特征进行压缩,以获取所述超边特征。
[0119]
下面将结合具体示例来分别说明顶点卷积(vertex conv)和超边卷积(hyperedge conv)。
[0120]
顶点卷积(vertex conv)
[0121]
如图5所示,假设第二超图中的每一条超边上有n个顶点,每个顶点的光谱维特征为d维,首先将输入原始顶点特征进行全局平均池化(gap)操作,然后在光谱维度上进行一维卷积操作,卷积核大小为k,并经过sigmoid激活函数得到各个光谱维度的权重矩阵w,公
式为
[0122]
w=σ(conv1k(y))
[0123]
其中,conv1表示一维卷积,k是卷积核的大小,y是全局平均池化操作后的特征(也即所述超像素点的第四光谱特征)。得到的权重矩阵w与原始顶点特征的对应元素相乘,得到一个新的顶点特征(也即所述超像素点的第五光谱特征)。我们从新顶点特征图中使用多层感知机(mlp)学习变换矩阵t(transformmatrix),进行特征排列和加权,并使用一维卷积对变换后的特征进行压缩。公式如下
[0124][0125][0126]
通过以上操作将顶点特征聚合到超边得到超边特征xe,对光谱特征进行进一步提取。
[0127]
超边卷积(hyperedge cony)
[0128]
通过超边卷积将由顶点卷积得到的超边特征聚合为质心顶点特征。如图6所示,对于超边卷积,我们采用注意机制,利用多层感知机(mlp)生成每条超边的权值。输出质心顶点特征计算为输入超边特征的加权和。过程公式为
[0129]
w=softmax(xew b)
[0130][0131]
其中,xe是由顶点卷积得到的质心顶点所在的超边xe组成的,即所有的相邻超边特征。n表示相邻超边集的超边数量,x表示超边卷积后得到的质心顶点特征(也即所述超像素点的第三光谱特征),w和b为可学习参数。
[0132]
接着,s70:根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征。
[0133]
经过超像素的特征提取后,通过解码器(decoder))将超像素点的光谱特征转换成了原始像素的光谱特征。
[0134]
最后,s80:基于所述原始像素点的光谱特征对所述原始像素点进行分类。
[0135]
具体地是将获取的原始像素点的光谱特征输入到全连接层中,实现原始像素的分类。全连接层的表达式为:
[0136][0137]
其中,表示经过解码器操作后得到的原始像素,fc()为全连接层,y为最终像素级分类结果。
[0138]
分类后,可以计算损失函数(loss function)来评价分类的准确定。
[0139]
损失函数采用的是交叉熵损失函数,公式为
[0140]
[0141]
其中,c是高光谱图像中原始像素点的类别总数,t是训练原始像素点的总数,如果第t个原始像素点属于第c个类的真值为是第t个像素的预测输出。
[0142]
数据集分类示意图如图7a-7e所示,其中,图7a-7e分别示出了伪彩色图像、地面真实图、训练样本图、测试样本图及分类结构图。从图中可以看出,训练样本图只采用少量像素作为训练样本(图7e中小点),而测试样本集采用除训练样本之外的其他像素。采用少量训练样本是因为通过超像素的特征标签可以确定一些未知像素点的特征标签,相当于隐式增加了样本数量,这让本技术提出的模型在少量训练样本的情况下就可以达到较高的分类精度。
[0143]
需要说明的是,上面各种方法的步骤划分,只是为了描述清楚,实现时可以合并为一个步骤或者对某些步骤进行拆分,分解为多个步骤,只要包含相同的逻辑关系,都在本专利的保护范围内;对算法中或者流程中添加无关紧要的修改或者引入无关紧要的设计,但不改变其算法和流程的核心设计都在该专利的保护范围内。
[0144]
图8为本技术实施例提供的基于动态超图卷积网络的高光谱图像装置的一个功能模块图。请参阅图5所示,所述基于动态超图卷积网络的高光谱图像装置11,包括图像获取降维模块111,像素分割模块112,第一超图构造模块113,超图卷积模块114,第二超图构造模块115,动态图卷积模块116,特征转换模块117及像素点分类模块118。
[0145]
其中,图像获取降维模块111,用于获取高光谱图像,并对所述高光谱图像进行图像预处理;像素分割模块112,用对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;第一超图构造模块113,用于根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;超图卷积模块114,用于对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;第二超图构造模块115,用于根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;动态图卷积模块116,用于对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;特征转换模块117,用于根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征;像素点分类模块118,用于基于所述原始像素点的光谱特征对所述原始像素点进行分类。
[0146]
需要说明的是,本实施例的基于动态超图卷积网络的高光谱图像装置11是与上述基于动态超图卷积网络的高光谱图像方法相对应的装置,基于动态超图卷积网络的高光谱图像装置11中的功能模块或者分别对应基于动态超图卷积网络的高光谱图像方法中的相应步骤。本实施例的基于动态超图卷积网络的高光谱图像装置11可与基于动态超图卷积网络的高光谱图像方法相互相配合实施。相应地,本实施例的基于动态超图卷积网络的高光谱图像装置11中提到的相关技术细节也可应用在上述基于动态超图卷积网络的高光谱图像方法中。
[0147]
需要说明的是,上述的各功能模块实际实现时可以全部或部分集成到一个物理实体上,也可以物理上分开。且这些模块可以全部以软件通过处理元件调用的形式实现;也可以全部以硬件的形式实现;还可以部分模块通过处理元件调用软件的形式实现,部分模块通过硬件的形式实现。此外这些模块全部或部分可以集成在一起,也可以独立实现。这里所述的处理元件可以是一种集成电路,具有信号的处理能力。在实现过程中,上述方法的部分或全部步骤,或以上的各功能模块可以通过处理器元件中的硬件的集成逻辑电路或者软件
形式的指令完成。
[0148]
综上所述,本技术的基于动态超图卷积网络的高光谱图像分类方法及系统,通过获取高光谱图像,并对所述高光谱图像进行图像预处理;对图像预处理后的所述高光谱图像进行超像素分割,以获得若干超像素点,每个超像素点对应若干原始像素点;根据所述超像素点的第一光谱特征进行超图构造,以获取第一超图;对所述第一超图进行超图卷积处理,得到所述超像素点的第二光谱特征;根据所述超像素点的第二光谱特征重新进行超图构造,以获取第二超图;对所述第二超图进行动态超图卷积处理,以获取每个所述超像素点的第三光谱特征;根据所述超像素点与原始像素点的对应关系,将所述超像素的光谱特征转换成对应原始像素点的光谱特征;基于所述原始像素点的光谱特征对所述原始像素点进行分类。通过两次超图构造,能够实现超像素点的光谱特征/原始像素点的光谱特征的动态提取,并且由于是将整个高光谱图像作为输入,而不是像多数cnn划分成块,这可以使网络更好的提取全局特征。同时,使用超像素分割技术将原始像素划分成超像素,大幅度减少了计算量,并且通过超像素的标签可以确定一些未知像素点的标签,相当于隐式增加了样本数量,这让本技术提出的模型在少量训练样本的情况下就可以达到较高的分类精度。
[0149]
对于本领域技术人员而言,显然本技术不限于上述示范性实施例的细节,而且在不背离本技术的精神或基本特征的情况下,能够以其他的具体形式实现本技术。
[0150]
以上实施例仅用以说明本技术的技术方案而非限制,尽管参照较佳实施例对本技术进行了详细说明,本领域的普通技术人员应当理解,可以对本技术的技术方案进行修改或等同替换,而不脱离本技术技术方案的精神和范围。
再多了解一些

本文用于创业者技术爱好者查询,仅供学习研究,如用于商业用途,请联系技术所有人。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献