一种高光谱图像空谱联合分类方法及装置制造方法及图纸

技术编号:24170760 阅读:27 留言:0更新日期:2020-05-16 02:52
本发明专利技术公开了一种高光谱图像空谱联合分类方法及装置,该方法包括:对输入的高光谱图像数据进行归一化处理,经过第一预处理后得到提取光谱特征的训练样本集S

【技术实现步骤摘要】
一种高光谱图像空谱联合分类方法及装置
本专利技术涉及一种高光谱图像的分类方法,尤其涉及一种基于可变形卷积神经网络的高光谱图像空谱联合分类方法,属于遥感信息处理

技术介绍
由高光谱传感器的探测感知机理可知,高光谱数据中蕴含了地物丰富的空间及光谱信息,具有光谱分辨率高、成像波段多、信息量大等特点。因此,高光谱图像广泛应用于军事、农业、医学、矿采等诸多领域。对高光谱图像中的地物进行分类识别是辅助研究人员快速理解高光谱图像中所包含信息的重要手段。虽然高光谱图像等光谱分辨率高,但是相对于多光谱、全色等类型的图像而言空间分辨率依然较低。在此情景下,大量城市地物像元是混像元。由于城市区域不同类型的地物高密度的混合在一起,造成了同一类地物的类内差异大,不同地物的类间差距小,这些因素给城市区域的高光谱分类带来了很大挑战。在此背景下,研究了一种基于可变形卷积神经网络的高光谱图像空谱联合分类方法。针对高光谱图像存在类内差异的问题,对每一类样本用k-means算法进行类内聚类选取具有代表性的训练样本;在高光谱图像分类任务中设计了多尺度空间-光谱联合网络和可变形卷积网络,自主提取高光谱图像的空谱特征,使得通过高光谱遥感获取的丰富信息被充分利用,对现实应用有着重要的意义。
技术实现思路
本专利技术所要解决的技术问题在于提供一种基于可变形卷积神经网络的高光谱图像空谱联合分类方法。该方法针对城市区域高光谱图像类内差异大的特点,首先对不同类别的样本作类内聚类,选择具有代表性的训练样本;然后充分结合高光谱图像的光谱信息和多尺度的空间信息,利用可变形卷积神经网络能适应几何形变的特点,提升网络性能及高光谱图像的分类精度。为实现上述的专利技术目的,本专利技术采用下述的技术方案:根据本专利技术的第一方面,一种高光谱图像空谱联合分类方法,包括如下步骤:对输入的高光谱图像数据进行归一化处理;分别将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S1以及经过第二预处理以得到提取空间特征的训练样本集S2;通过多尺度空间-光谱联合网络从所述样本集S1中提取光谱特征及从所述样本集S2中提取空间特征;将提取到的光谱特征和多尺度的空间特征联结起来,输入到能够适应几何形变的可变形卷积神经网络中,以提取高级特征;以及经可变性卷积网络将提取的高级特征输入到全连接网络分类器中进行分类,以得到基于可变形卷积神经网络的高光谱图像空谱联合分类结果。进一步的,所述对输入的高光谱图像数据进行归一化处理的步骤包括通过以下公式进行归一化处理:其中,y(i,j,k)表示高光谱图像中第k个波段中位置坐标为(i,j)的一个像元,max(yk)和min(yk)分别表示三维高光谱图像在第k个波段的最大值和最小值。进一步的,所述将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S1的步骤包括:利用k-means聚类算法对所述高光谱图像中的每一类样本包含的光谱作类内聚类,聚类数目为3簇;从每个簇中选取20%的样本保存其标签,该标签作为训练样本标签Gtr,剩下的作为测试样本标签Gte;提取以带标签Gtr为中心的11×11×L邻域范围内的数据块作为提取光谱特征的训练样本集S1。进一步的,所述经过第二预处理以得到提取空间特征的训练样本集S2的步骤包括:利用主成分分析PCA方法对原始高光谱图像进行降维处理,从降维后的图像中提取以带标签Gtr为中心的11×11×D邻域范围内的数据块作为提取空间特征的训练样本集S2。进一步的,所述通过多尺度空间-光谱联合网络包括光谱特征提取网络和空间特征提取网络,所述光谱特征提取网络结构包括1×1卷积层,归一化层BatchNormalization和激励函数ReLU,所述空间特征提取网络结构包括并联的3×3卷积层和5×5卷积层,卷积层后分别归一化层BatchNormalization和激励函数ReLU,其中,训练样本集S1输入到光谱特征提取网络以提取光谱特征,训练样本集S2输入到空间特征提取网络以获得多尺度的空间特征。进一步的,所述可变形卷积网络从输入到输出包括的卷积层依次包括第一二维卷积层、第一可变形卷积层、第二可变形卷积层、第二二维卷积层以及全局平均池化层。进一步的,每个所述二维卷积层和可变形卷积层的卷积核大小均为3×3,步长为1。进一步的,卷积核个数依次为:64,128,256,512。进一步的,每个所述二维卷积层和可变形卷积层后均使用BatchNormalization层做归一化,并使用ReLU作为激励函数。进一步的,所述全连接网络分类器包括一全连接层和一softmax分类层。根据本专利技术的第二方面,一种高光谱图像空谱联合分类装置,该装置包括:存储器,该存储器中存储有计算机指令;处理器,该处理器与存储器数据连接,其通过执行所述计算机指令,从而执行根据权利要求1-8任一项所述的基于可变形卷积神经网络的高光谱图像空谱联合分类方法。根据本专利技术的第三方面,一种高光谱图像空谱联合分类装置,该装置包括:归一化处理模块,用于对输入的高光谱图像数据进行归一化处理;训练样本集生成模块,用于分别将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S1以及经过第二预处理以得到提取空间特征的训练样本集S2;第一特征提取模块,用于通过多尺度空间-光谱联合网络从所述训练样本集S1中提取光谱特征及从所述训练样本集S2中提取空间特征;第二特征提取模块,用于将提取到的光谱特征和多尺度的空间特征联结起来,输入到能够适应几何形变的可变形卷积神经网络中,以提取高级特征;以及全连接网络分类器,用于根据输入的经可变性卷积网络将提取的高级特征得到基于可变形卷积神经网络的高光谱图像空谱联合分类结果。本专利技术所提供的基于可变形卷积神经网络的高光谱图像空谱联合分类方法的优点包括:本专利技术在样本选择时,针对高光谱图像存在类内差异的问题采用了k-means聚类算法先对每类样本作类内聚类,选择具有代表性的样本。这种方式选取的训练样本比随机选取训练样本能更好的覆盖样本空间;本专利技术设计的多尺度空间-光谱联合网络,同时利用了高光谱图像的光谱特征和多尺度的空间特征,可以提取更加丰富和完善的特征信息;本专利技术采用的可变形卷积网络,可以自适应的提取不同大小和形状的地物特征,能有效提高高光谱图像的分类精度。附图说明图1为本专利技术提出的高光谱图像空谱联合分类方法流程图;图2为本专利技术提出的提取光谱特征网络示意图;图3为本专利技术提出的提取空间特征网络示意图;图4为本专利技术提出的可变形卷积神经网络模块示意图。具体实施方式下面将结合本专利技术实施例中的附图,对本专利技术实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本专利技术一部分实施例,而不是全部的实施例。基于本专利技术中的实本文档来自技高网...

【技术保护点】
1.一种高光谱图像空谱联合分类方法,其特征在于,所述方法包括:/n对输入的高光谱图像数据进行归一化处理;/n分别将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S

【技术特征摘要】
1.一种高光谱图像空谱联合分类方法,其特征在于,所述方法包括:
对输入的高光谱图像数据进行归一化处理;
分别将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S1,以及经过第二预处理以得到提取空间特征的训练样本集S2;
采用多尺度空间-光谱联合网络,从所述训练样本集S1中提取光谱特征以及从所述训练样本集S2中提取空间特征;
将提取到的光谱特征和多尺度的空间特征联结起来,输入到能够适应几何形变的可变形卷积神经网络中,以提取高级特征;以及
经可变性卷积网络将提取的高级特征输入到全连接网络分类器中进行分类,以得到基于可变形卷积神经网络的高光谱图像空谱联合分类结果。


2.根据权利要求1所述的高光谱图像空谱联合分类方法,其特征在于,所述将归一化处理后的高光谱图像经过第一预处理后以得到提取光谱特征的训练样本集S1的步骤包括:
利用k-means聚类算法对所述高光谱图像中的每一类样本包含的光谱作类内聚类;
从每个簇中选取部分样本保存其标签,该标签作为训练样本标签Gtr,剩下的作为测试样本标签Gte;
提取以带标签Gtr为中心的11×11×L邻域范围内的数据块作为提取光谱特征的训练样本集S1。


3.根据权利要求2所述的高光谱图像空谱联合分类方法,其特征在于,所述经过第二预处理以得到提取空间特征的训练样本集S2的步骤包括:利用主成分分析PCA方法对原始高光谱图像进行降维处理,从降维后的图像中提取以带标签Gtr为中心的11×11×D邻域范围内的数据块作为提取空间特征的训练样本集S2。


4.根据权利要求1-3任一项所述的高光谱图像空谱联合分类方法,其特征在于,所述通过多尺度空间-光谱联合网络包括光谱特征提取网络和空间特征提取网络;
所述光谱特征提取网络结构包括1×1卷积层,归一化层BatchNormalization和激励函数ReLU;
所述空间特征提取网络结构包括并联的3×3卷积层和5×5卷积层,卷积层后分别归一化层BatchNormalization和激励函数ReLU;<...

【专利技术属性】
技术研发人员:徐其志聂进焱
申请(专利权)人:北京云智空间科技有限公司
类型:发明
国别省市:北京;11

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

1