多源遥感数据分类模型训练方法、分类方法和电子设备技术

技术编号:37330156 阅读:20 留言:0更新日期:2023-04-21 23:08
本申请提供多源遥感数据分类模型训练方法、分类方法和电子设备,属于图像处理技术领域,其中,分类模型包括特征生成网络和地物分类器,特征生成网络用于提取多源湿地遥感数据的共性特征和特异性特征,训练方法包括:将判别器和特征生成网络进行至少一轮交替训练,在判别器训练阶段最小化判别器损失函数,使判别器能够对共性特征进行模态分类;在特征生成网络训练阶段最大化判别器损失函数,共性特征和特异性特征线性独立。通过对抗学习保证共性特征不能被模态分类器区分,同时特异性特征与共性特征线性独立,与特异性特征优势互补,一定程度避免了多源特征的冗余性。程度避免了多源特征的冗余性。程度避免了多源特征的冗余性。

【技术实现步骤摘要】
多源遥感数据分类模型训练方法、分类方法和电子设备


[0001]本专利技术涉及图像处理
,具体涉及多源遥感数据分类模型训练方法、分类方法和电子设备。

技术介绍

[0002]随着高分系列卫星的发射,遥感数据储量日益丰富,为开展滨海湿地遥感监测提供了有利条件。遥感数据往往具有多种异质图像,例如光谱图像、激光雷达图像、合成孔径雷达图像等。以光谱图像为例,光谱图像可以包括高光谱图像(Hyperspectral Image, HSI)和多光谱图像(Multispectral Image, MSI),其中,高光谱图像能够提供从可见光谱到红外光谱的光谱信息。因此,高光谱图像能够反映被观测物体的物质特性,从而有利于地物的精细识别。然而,高光谱成像系统为了满足信噪比要求,其成像的空间分辨率相对较低。相比之下,多光谱图像的光谱分辨率相对较低,而其空间纹理信息更为丰富,能够与高光谱图像形成互补。因此,在海量的遥感数据投入应用的同时,亟需自动、快速、大范围的多源遥感数据需要进行协同分类。
[0003]近年来,兴起了很多有关于多源遥感数据协同分类方法研究,主要分为数据融合方法和特征融合方法,考虑到数据融合方法对于异质数据的局限性,特征融合方法成为当前最热门领域。特别是,近年来基于深度学习的特征融合方法能够通过特征自动学习获取多源数据的优势特征,最后通过特征堆叠、加和或者注意力机制算法进行特征融合。然而多源遥感数据具有较高的冗余性,从而导致负增益,难以实现湿地地物的精准分类。
[0004]因此,如何进行提升湿地地物分类的精度成为亟待解决的技术问题。

技术实现思路

[0005]为解决上述
技术介绍
中阐述的现有技术中如何进行提升湿地地物分类的精度的技术问题,本申请提出了一种多源遥感数据分类模型训练方法、分类方法和电子设备。
[0006]根据第一方面本申请实施例提供了一种多源遥感数据分类模型训练方法,所述分类模型包括特征生成网络和地物分类器,所述特征生成网络用于提取多源湿地遥感数据的共性特征和特异性特征,所述训练方法包括:将判别器和所述特征生成网络进行至少一轮交替训练,得到训练后的特征生成网络,其中,每一轮交替训练中,包括至少一次判别器训练阶段和至少一次特征生成网络训练阶段,在判别器训练阶段最小化判别器损失函数,使所述判别器能够对共性特征进行模态分类,且使所述共性特征和特异性特征距离小于预设距离;在特征生成网络训练阶段最大化所述判别器损失函数,直至所述特征生成网络生成的共性特征不能被预所述判别器进行模态分类,且所述共性特征和所述特异性特征线性独立。
[0007]可选地,轮次交替训练中的判别器的训练样本为上一轮交替训练完成后所述特征生成网络提取的共性特征和特异性特征。
[0008]可选地,对所述判别器进行第一次训练的过程包括:利用初始的特征生成网络提
取多源湿地遥感数据的初始共性特征和初始特异性特征;利用初始共性特征和初始特异性特征对所述判别器进行训练,使训练后的判别器能够对初始共性特征进行模态分类,且使所述共性特征和特异性特征距离小于预设距离。
[0009]可选地,所述将判别器和所述特征生成网络进行至少一轮交替训练包括:在判别器训练阶段最小化模态分类器损失函数,使得模态分类器能够对共性特征进行模态分类,同时,最小化均方误差损失函数,使所述共性特征和所述特异性特征的距离小于预设距离;在特征生成网络训练阶段最大化所述模态分类器损失函数,直至所述特征生成网络生成的共性特征不能够被当前轮次训练后的模态分类器进行模态分类,同时,最大化均方误差损失函数,直至所述共性特征和所述特异性特征线性独立。
[0010]可选地,多源遥感数据分类模型训练方法还包括:构建特征重建网络,所述特征重建网络用于对所述共性特征和所述特异性特征向所述多源湿地遥感数据原始特征进行重建,得到重建共性特征和重建特异性特征;在将判别器和所述特征生成网络进行至少一轮交替训练时,利用所述重建网络进行约束,使所述特征生成网络提取的共性特征和特异性特征能够重建多源湿地遥感数据原始特征。
[0011]可选地,得到训练后的特征生成网络之后还包括:利用训练后的特征生成网络提取的共性特征和特异性特征对所述地物分类器进行训练,得到训练后的地物分类器。
[0012]根据第二方面,本申请实施例提供了一种多源遥感数据分类方法,包括:获取待分类多源遥感数据;利用上述第一方面任意一项所述多源遥感数据分类模型训练方法训练的多源遥感数据分类模型对所述待分类多源遥感数据进行分类,得到分类结果。
[0013]根据第三方面,本申请实施例提供了一种多源遥感数据分类模型训练装置,包括:网络生成模块,用于构建特征生成网络,所述特征生成网络用于生成多源湿地遥感数据的共性特征和特异性特征;训练模块,用于将判别器和所述特征生成网络进行至少一轮交替训练,得到训练后的特征生成网络,其中,每一轮交替训练中,包括至少一次判别器训练阶段和至少一次特征生成网络训练阶段,在判别器训练阶段最小化判别器损失函数,使所述判别器能够对共性特征进行模态分类,且使所述共性特征和特异性特征距离小于预设距离;在特征生成网络训练阶段最大化所述判别器损失函数,直至所述特征生成网络生成的共性特征不能被预所述判别器进行模态分类,且所述共性特征和所述特异性特征线性独立。
[0014]根据第四方面,本申请实施例提供了一种电子设备,包括处理器、通信接口、存储器和通信总线,其中,所述处理器、所述通信接口和所述存储器通过所述通信总线完成相互间的通信,所述存储器,用于存储计算机程序;所述处理器,用于通过运行所述存储器上所存储的所述计算机程序来执行第一方面中任一项所述的多源遥感数据分类模型训练方法和/或第二方面所述的多源遥感数据分类方法。
[0015]根据第五方面,本申请实施例提供了一种计算机可读的存储介质,其特征在于,所述存储介质中存储有计算机程序,其中,所述计算机程序被设置为运行时执行第一方面中任一项所述的多源遥感数据分类模型训练方法和/或第二方面所述的多源遥感数据分类方法。
[0016]本申请通过构建特征生成网络用于对特异性特征和共性特征进行提取,将判别器和所述特征生成网络进行至少一轮交替训练,在判别器训练阶段最小化判别器损失函数,
使所述判别器能够对共性特征进行模态分类,且,使所述共性特征和特异性特征距离小于预设距离;在特征生成网络训练阶段最大化所述判别器损失函数,直至所述特征生成网络生成的共性特征不能被预所述判别器进行模态分类,且所述共性特征和所述特异性特征线性独立充分考虑高光谱图像和多光谱图像的数据特性,针对性地构建特异性和共性特征生成网络,通过最大

最小的对抗学习策略保证共性特征不能被模态分类器区分,同时特异性特征与共性特征线性独立,也就是说共性不包含模态相关信息,与特异性特征优势互补征,一定程度避免了多源特征的冗余性。
[0017]进一步,设计重建网络对共性特征和特异性特征进行重建,保持共性特征和特异性特征的完整性,而且避免了网络模型的捷径优化。...

【技术保护点】

【技术特征摘要】
1.一种多源遥感数据分类模型训练方法,其特征在于,所述分类模型包括特征生成网络和地物分类器,所述特征生成网络用于提取多源湿地遥感数据的共性特征和特异性特征,所述训练方法包括:将判别器和所述特征生成网络进行至少一轮交替训练,得到训练后的特征生成网络,其中,每一轮交替训练中,包括至少一次判别器训练阶段和至少一次特征生成网络训练阶段,在判别器训练阶段最小化判别器损失函数,使所述判别器能够对共性特征进行模态分类,且使所述共性特征和特异性特征距离小于预设距离;在特征生成网络训练阶段最大化所述判别器损失函数,直至所述特征生成网络生成的共性特征不能被预所述判别器进行模态分类,且所述共性特征和所述特异性特征线性独立。2.如权利要求1所述的多源遥感数据分类模型训练方法,其特征在于,当前轮次交替训练中的判别器的训练样本为上一轮交替训练完成后所述特征生成网络提取的共性特征和特异性特征。3.如权利要求2所述多源遥感数据分类模型训练方法,其特征在于,对所述判别器进行第一次训练的过程包括:利用初始的特征生成网络提取多源湿地遥感数据的初始共性特征和初始特异性特征;利用初始共性特征和初始特异性特征对所述判别器进行训练,使训练后的判别器能够对初始共性特征进行模态分类,且,使所述共性特征和特异性特征距离小于预设距离。4.如权利要求1所述的多源遥感数据分类模型训练方法,其特征在于,所述将判别器和所述特征生成网络进行至少一轮交替训练包括:在判别器训练阶段最小化模态分类器损失函数,使得模态分类器能够对共性特征进行模态分类,同时,最小化均方误差损失函数,使所述共性特征和所述特异性特征的距离小于预设距离;在特征生成网络训练阶段最大化所述模态分类器损失函数,直至所述特征生成网络生成的共性特征不能够被当前轮次训练后的模态分类器进行模态分类,同时,最大化均方误差损失函数,直至所述共性特征和所述特异性特征线性独立。5.如权利要求1所述的多源遥感数据分类模型训练方法,其特征在于,还包括:构建特征重建网络,所述特征重建网络用于对所述共性特征和所述特异性特征向所述多源湿地遥感数据原始特征进行重建,得到重建共性特征和重建特异性特征;在将判别器和所述特征生成网络进行至少一轮交替训...

【专利技术属性】
技术研发人员:王建步高云浩朱文博马元庆胡亚斌秦华伟宋秀凯李伟宋莎莎隋傅王玮云
申请(专利权)人:山东省海洋资源与环境研究院山东省海洋环境监测中心山东省水产品质量检验中心北京理工大学
类型:发明
国别省市:

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

1