一种遥感图像的融合方法、系统及相关组件与流程

文档序号:16211578发布日期:2018-12-08 07:48阅读:119来源:国知局
一种遥感图像的融合方法、系统及相关组件与流程

本发明涉及图像处理领域,特别涉及一种遥感图像的融合方法、系统、一种计算机可读存储介质及一种电子设备。

背景技术

多传感器遥感系统能够为用户提供高空间分辨率、高光谱分辨率、高时间分辨率的海量图像。但是,由于每一种图像数据都有其固有的特点和特定的应用领域,因此如何融合来自各种类型遥感图像信息,弥补单一遥感图像信息的不足,综合利用各遥感图像的优势,已成为当前迫切需要解决的问题。

现有技术中,全色图像与多光谱图像融合技术主要有基于元素替换的ihs(intensity-hue-saturation)方法,以及基于多分辨分析的小波变换的方法,即,在全色图像上提取高频信息然后将其注入多光谱图像。通过现有技术融合后的图像只能在高空间分辨率和高光谱分辨率之间权衡,无法得到既有高空间分辨率又有高光谱分辨率的融合图像。

因此,如何提高融合全色图像和多光谱图像的质量是本领域技术人员目前需要解决的技术问题。



技术实现要素:

本申请的目的是提供一种遥感图像的融合方法、系统、一种计算机可读存储介质及一种电子设备,能够提高融合全色图像和多光谱图像的质量。

为解决上述技术问题,本申请提供一种遥感图像的融合方法,该融合方法包括:

对训练样本进行预处理得到训练数据集;其中,所述训练数据集包括多张全色图像和多张多光谱图像;

将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息;

利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;其中,所述编码特征图由所述特征融合神经网络对所有所述全色图像和所有所述多光谱图像进行编码操作得到;

将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;

根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

可选的,所述对训练样本进行预处理得到训练数据集包括:

将所述训练样本中对同一区域采集的全色图像与多光谱图像设置为一个图像组;

对每一所述图像组的全色图像和多光谱图像分别下采样得到预设尺寸的下采样全色图像和下采样多光谱图像;

对所述下采样多光谱图像进行逐波段的插值操作,以使插值操作后的下采样多光谱图像中每个波段图像的像素大小与所述下采样多光谱图像中每个波段图像的像素大小相等;

根据所有所述下采样全色图像和所述下采样多光谱图像构建所述训练数据集。

可选的,将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息包括:

将所述下采样全色图像和所述下采样多光谱图像分别输入至特征提取神经网络的第一子卷积神经网络和第二子卷积神经网络得到中间全色特征信息和中间多光谱特征信息;其中,所述第一子卷积神经网络和所述第二子卷积神经网络分别设有非线性激活单元,所述特征提取神经网络包括所述所述第一子卷积神经网络、所述第二子卷积神经网络、所述非线性激活单元和卷积核;

将所述中间全色特征信息和所述中间多光谱特征信息输入所述卷积核进行下采样得到所述全色特征信息和所述多光谱特征信息。

可选的,将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像包括:

将图像重构神经网络中每两层中的所述编码特征图向上采样作为所述图像编码器的对称结构,将所述图像重构神经网络设置为图像解码器,并在所述图像编码器和所述图像解码器之间添加跳转连接;其中,所述图像编码器为所述特征提取神经网络和所述特征融合神经网络的整体连接网络;

将所述图像编码器中的特征映射复制到所述图像解码器中,并建立所述图像解码器与所述图像编码器之间的特征映射级联;

将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像。

可选的,在重构神经网络构建深度神经网络模型之后,还包括:

将所述训练数据集中的所有图像组输入至所述深度神经网络模型,并采用损失函数和随机梯度下降方法对所述深度神经网络模型进行训练;其中,为所述下采样全色图像,为所述下采样多光谱图像,所述损失函数yi为所述图像组对应的高分辨率多光谱图像,n为所述图像组的数量,θ是神经网络参数;

当所述深度神经网络模型的损失函数与区域局部最优解之差小于预设值时,停止训练所述深度神经网络模型。

本申请还提供了一种遥感图像的融合系统,该系统包括:

预处理模块,用于对训练样本进行预处理得到训练数据集;其中,所述训练数据集包括多张全色图像和多张多光谱图像;

第一训练模块,用于将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息;

第二训练模块,用于利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;其中,所述编码特征图由所述特征融合神经网络对所有所述全色图像和所有所述多光谱图像进行编码操作得到;

第三训练模块,用于将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;

模型构建模块,用于根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

可选的,所述预处理模块包括:

图像组确定单元,用于将所述训练样本中对同一区域采集的全色图像与多光谱图像设置为一个图像组;

下采样单元,用于对每一所述图像组的全色图像和多光谱图像分别下采样得到预设尺寸的下采样全色图像和下采样多光谱图像;

插值处理单元,用于对所述下采样多光谱图像进行逐波段的插值操作,以使插值操作后的下采样多光谱图像中每个波段图像的像素大小与所述下采样多光谱图像中每个波段图像的像素大小相等;

构建单元,用于根据所有所述下采样全色图像和所述下采样多光谱图像构建所述训练数据集。

可选的,所述第一训练模块包括:

第一特征提取单元,用于将所述下采样全色图像和所述下采样多光谱图像分别输入至特征提取神经网络的第一子卷积神经网络和第二子卷积神经网络得到中间全色特征信息和中间多光谱特征信息;其中,所述第一子卷积神经网络和所述第二子卷积神经网络分别设有非线性激活单元,所述特征提取神经网络包括所述所述第一子卷积神经网络、所述第二子卷积神经网络、所述非线性激活单元和卷积核;

第二特征提取单元,用于将所述中间全色特征信息和所述中间多光谱特征信息输入所述卷积核进行下采样得到所述全色特征信息和所述多光谱特征信息。

本申请还提供了一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序执行时实现上述遥感图像的融合方法执行的步骤。

本申请还提供了一种电子设备,包括存储器和处理器,所述存储器中存储有计算机程序,所述处理器调用所述存储器中的计算机程序时实现上述遥感图像的融合方法执行的步骤。

本发明提供了一种遥感图像的融合方法,包括对训练样本进行预处理得到训练数据集;将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息;利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

本发明通过利用训练数据集训练特征提取神经网络全色特征信息和多光谱特征信息,并利用全色特征信息和多光谱特征信息训练特征融合神经网络得到融合特征,进而利用构建好的深度神经网络模型从特征级进行图像融合操作。由于深度神经网络可以表示输入的层级特征,且高分辨率多光谱图是从这些特征中进行重建得到的,故本申请能够提高融合全色图像和多光谱图像的质量,减少光谱损失。本申请同时还提供了一种遥感图像的融合系统、一种计算机可读存储介质和一种电子设备,具有上述有益效果,在此不再赘述。

附图说明

为了更清楚地说明本申请实施例,下面将对实施例中所需要使用的附图做简单的介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。

图1为本申请实施例所提供的一种遥感图像的融合方法的流程图;

图2为本申请实施例所提供的另一种遥感图像的融合方法的流程图;

图3为深度神经网络的网络结构图;

图4为本申请实施例所提供的一种遥感图像的融合系统的结构示意图。

具体实施方式

为使本申请实施例的目的、技术方案和优点更加清楚,下面将结合本申请实施例中的附图,对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本申请一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本申请保护的范围。

下面请参见图1,图1为本申请实施例所提供的一种遥感图像的融合方法的流程图。

具体步骤可以包括:

s101:对训练样本进行预处理得到训练数据集;

其中,所述训练数据集包括多张全色图像和多张多光谱图像;

本实施例默认训练样本包括有多张全色图像和多张多光谱图像的,且训练样本中的每一张全色图像都存在一张与其对应的多光谱图像。例如:存在高空间分辨率的全色图像a、b、c,低空间分辨率的多光谱图像a、b、c,将全色图与多光谱图是同一区域的不同传感器成像图片设置为一组,即存在a-a,b-b,c-c的对应关系,一对全色图像和多光谱图像为不同传感器对同一区域拍摄的不同图像。

具体的,对训练样本进行预处理得到训练数据集可以包括以下操作:

步骤1:将所述训练样本中对同一区域采集的全色图像与多光谱图像设置为一个图像组;

其中,可以选取作为训练使用的高空间分辨率的全色图像和低空间分辨率的多光谱图像,构建高空间分辨率与低空间分辨率图像块对的训练集;每一个图像组都有不同传感器对同一区域采集的全色图像和多光谱图像。

步骤2:对每一所述图像组的全色图像和多光谱图像分别下采样得到预设尺寸的下采样全色图像和下采样多光谱图像;

步骤3:对所述下采样多光谱图像进行逐波段的插值操作,以使插值操作后的下采样多光谱图像中每个波段图像的像素大小与所述下采样多光谱图像中每个波段图像的像素大小相等;

具体的,可以使用双立方插值算法对输入的多光谱图像进行逐波段的插值操作,得到初始放大的多光谱图像,得多光谱图像中每个波段图像的像素大小与全色图像的大小保持一致。还可以分别对已知的高分辨全色图像和初始放大的多光谱图像的每个波段像素进行最大最小归一化方法,使每一个像素取值范围在[0,1]之间。

步骤4:根据所有所述下采样全色图像和所述下采样多光谱图像构建所述训练数据集。

s102:将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息;

其中,本步骤相当于利用训练数据集对特征提取神经网络进行训练的过程。由于本实施例的目的在于将多光谱图像和全色图像进行融合,因此在实施本步骤时需要多光谱图像及其对应的全色图像一同输入特征提取神经网络得到全色特征信息和多光谱特征信息,训练数据集可以包括多张全色图像和多张多光谱图像,因此再将训练数据集中的图像输入值特征提取神经网络得到相应的特征信息后,即可完成特征提取神经网络的训练。目前,训练特征提取神经网络的技术手段已经很成熟,此处不赘述具体的训练过程,总之本步骤通过输入训练数据集至特征提取神经网络得到得到全色特征信息和多光谱特征信息实现对于特征提取神经网络的训练。

具体的,特征提取神经网络可以包括所述所述第一子卷积神经网络、所述第二子卷积神经网络、所述非线性激活单元和卷积核;将训练数据集中的两种图像分别输入至特征提取神经网络的第一子卷积神经网络和第二子卷积神经网络得到中间全色特征信息和中间多光谱特征信息,将所述中间全色特征信息和所述中间多光谱特征信息输入所述卷积核进行下采样得到所述全色特征信息和所述多光谱特征信息。其中,所述第一子卷积神经网络和所述第二子卷积神经网络分别设有非线性激活单元,非线性激活单元(relu)的作用在于:加速收敛过程,等效于对网络进行无监督学习的预训练。

s103:利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;其中,所述编码特征图由所述特征融合神经网络对所有所述全色图像和所有所述多光谱图像进行编码操作得到;

其中,本步骤的目的是训练特征融合神经网络,本实施例默认s102中的特征提取神经网络与特征融合神经网络连接,具体的,特征提取神经网络输出的结果是特征融合神经网络的输入数据。

在已经得到全色特征信息和多光谱特征信息的基础上可以利用融合神经网络进行融合操作得到融合特征,融合特征是既包含全色特征信息又包含多光谱特征信息的特征,即可以理解为融合特征包括了全色图像和多光谱图像的互补信息。进一步的在特征融合神经网络内还存在对训练数据集中的多光谱图像和全色图像进行编码的操作,将一张多光谱图像和一张全色图像进行编码得到一张编码特征图。

s104:将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;

其中,本步骤的目的是训练图像重构神经网络,图像重构神经网络可以与特征融合神经网络连接,将特征融合神经网络输出的结果输入图像重构神经网络利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像。

具体的,本步骤中训练图像重构神经网络的步骤可以包括:

步骤1:将图像重构神经网络中每两层中的所述编码特征图向上采样作为所述图像编码器的对称结构,将所述图像重构神经网络设置为图像解码器,并在所述图像编码器和所述图像解码器之间添加跳转连接;其中,所述图像编码器为所述特征提取神经网络和所述特征融合神经网络的整体连接网络;

步骤2:将所述图像编码器中的特征映射复制到所述图像解码器中,并建立所述图像解码器与所述图像编码器之间的特征映射级联;

步骤3:将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像。

s105:根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

其中,本步骤建立在已经对特征提取神经网络、特征融合神经网络和所述图像重构神经网络均已经训练完毕的基础上,构建深度神经网络模型。根据特征提取神经网络、特征融合神经网络和所述图像重构神经网络三者的所起的作用可知,特征提取神经网络与特征融合神经网络连接,特征融合神经网络与图像重构神经网络连接,特征提取神经网络将提取到的图像特征输入特征融合神经网络,特征融合神经网络将得到的融合特征和编码特征图输入图像重构神经网络。

需要说明的是,在构件好深度神经网络模型之后还可以存在对深度神经网络模型进行训练的过程,具体步骤如下:

步骤1:将所述训练数据集中的所有图像组输入至所述深度神经网络模型,并采用损失函数和随机梯度下降方法对所述深度神经网络模型进行训练;其中,为所述下采样全色图像,为所述下采样多光谱图像,所述损失函数yi为所述图像组对应的高分辨率多光谱图像,n为所述图像组的数量,θ是神经网络参数;损失函数是为了评估模型得到的数据与真实数据之间的误差值,这个值越小代表模型越好。

步骤2:当所述深度神经网络模型的损失函数与区域局部最优解之差小于预设值时,停止训练所述深度神经网络模型。

本实施例通过利用训练数据集训练特征提取神经网络全色特征信息和多光谱特征信息,并利用全色特征信息和多光谱特征信息训练特征融合神经网络得到融合特征,进而利用构建好的深度神经网络模型从特征级进行图像融合操作。由于深度神经网络可以表示输入的层级特征,且高分辨率多光谱图是从这些特征中进行重建得到的,故本实施例能够提高融合全色图像和多光谱图像的质量,减少光谱损失。

下面请参见图2,图2为本申请实施例所提供的另一种遥感图像的融合方法的流程图;

具体步骤可以包括:

s201:将所述训练样本中对同一区域采集的全色图像与多光谱图像设置为一个图像组;

s202:对每一所述图像组的全色图像和多光谱图像分别下采样得到预设尺寸的下采样全色图像和下采样多光谱图像;

s203:对所述下采样多光谱图像进行逐波段的插值操作,以使插值操作后的下采样多光谱图像中每个波段图像的像素大小与所述下采样多光谱图像中每个波段图像的像素大小相等;

s204:根据所有所述下采样全色图像和所述下采样多光谱图像构建所述训练数据集。

s205:将所述下采样全色图像和所述下采样多光谱图像分别输入至特征提取神经网络的第一子卷积神经网络和第二子卷积神经网络得到中间全色特征信息和中间多光谱特征信息;其中,所述第一子卷积神经网络和所述第二子卷积神经网络分别设有非线性激活单元,所述特征提取神经网络包括所述所述第一子卷积神经网络、所述第二子卷积神经网络、所述非线性激活单元和卷积核;

s206:将所述中间全色特征信息和所述中间多光谱特征信息输入所述卷积核进行下采样得到所述全色特征信息和所述多光谱特征信息。

s207:利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;其中,所述编码特征图由所述特征融合神经网络对所有所述全色图像和所有所述多光谱图像进行编码操作得到;

s208:将图像重构神经网络中每两层中的所述编码特征图向上采样作为所述图像编码器的对称结构,将所述图像重构神经网络设置为图像解码器,并在所述图像编码器和所述图像解码器之间添加跳转连接;其中,所述图像编码器为所述特征提取神经网络和所述特征融合神经网络的整体连接网络;

s209:将所述图像编码器中的特征映射复制到所述图像解码器中,并建立所述图像解码器与所述图像编码器之间的特征映射级联;

s210:将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;

s211:根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

下面通过在实际应用中的实施例说明上述实施例描述的流程。

步骤a、对训练样本图像进行预处理和图像光谱特征提取,得到训练数据集具体过程为:

选取作为训练使用的高空间分辨率的全色图像和低空间分辨率的多光谱图像构建高空间分辨率与低空间分辨率图像块对的训练集

将全色图像与多光谱图像分别下采样4个宽度和高度,然后将下采样图像用作网络的输入,并将原始多光谱图用作参考;

使用双立方插值算法对输入的多光谱图像进行逐波段的插值操作,得到初始放大的多光谱图像使得多光谱图像中每个波段图像的像素大小与全色图像的大小保持一致;

分别对已知的高分辨全色图像和初始放大的多光谱图像的每个波段像素进行最大最小归一化方法,使每一个像素取值范围在[0,1]之间;

步骤b、构建深度神经网络模型,神经网络模型由三部分组成,包括特征提取、特征融合和图像重构,前两部分的作用类似于从输入图像中提取特征的编码器,最后一部分对融合特征进行解码以重建所需的高分辨率多光谱图像,图像大小的单位均为像素,参数设置见表1深度神经网络模型参数表:

表1深度神经网络模型参数表

构建的过程具体为:

(1)特征提取:使用两个子卷积神经网络φm和φp分别从全色图像和多光谱图像中提取特征,这两个子网具有相似的架构但权重不同,一个子网将四波段的多光谱图作为输入,另一个子网将单波段的全色图作为输入。每个特征提取网络由两个连续的卷积层组成,后面跟着一个非线性激活单元(relu)加速收敛过程,等效于对网络进行无监督学习的预训练,最后是一个步长为2的卷积核进行下采样,从全色图像和多光谱图像卷积神经网络提取的特征可以写成xp和ym,其中上标l表示特征从第l层提取的;

(2)特征融合:经过步骤b(1)的特征提取,我们有了分别代表全色图像和多光谱图像的特征图xp和ym,这两张特征图包含了全色图和多光谱图的互补信息,为了同时获得空间和光谱信息,将两个特征如方程所示连接在一起,其中φf(xp,ym)是融合特征,代表连接操作。应用具有三个卷积层的融合网络对两幅输入图像的光谱信息和空间信息进行编码;

(3)图像重构:经过编码的特征图只占用输入比例的1/4×1/4的宽度和高度,使用反卷积层从融合特征中重建高分辨率多光谱图像,并将每两层中的特征图向上采样作为编码器(即特征提取和融合网络)的对称结构,在每个上采样步骤之后,在网络的编码器和解码器之间添加跳转连接,编码器中的特征映射被复制到解码器并与相应的特征映射级联,以便在下采样过程中注入更多丢失的细节。构建深度神经网络模型的结构请参见图3,图3为深度神经网络的网络结构图。

步骤c、利用训练数据集训练深度神经网络,具体过程为:

将步骤a中得到训练集作为输入数据输入步骤b中的神经网络,采用损失函数为进行训练,其中yi是相应的高分辨率多光谱图像,n是训练样本的数量。

按照(1)中的损失函数和随机梯度下降方法对深度神经网络进行训练,针对不同尺寸的输入图像,需要将其映射为128×128的矩形,以适应网络结构的输入,使用双线性插值法,将图像减去像素均值后利用神经网络进行训练,通过前向传播逐层提取特征,当整个深度神经网络的损失函数区域局部最优解附近时完成训练。

步骤d、基于测试数据集和训练好的深度神经网络模型,得到测试数据集的融合区域,具体过程为:

测试样本图像同样包含两幅待融合的高分辨率全色图像和低分辨率多光谱图像,按照步骤a中获得训练数据集的方法进行图像预处理,得到测试数据集

将(1)中得到的数据集作为步骤c中训练好的深度神经网络的输入,经过该深度神经网络后输出融合好的高分辨率多光谱图像。

本实施例在特征级全色图像与多光谱图像融合,提供了一种特征提取、特征融合和图像重构的端到端神经网络架构,特征提取网络可以同时接受全色图像和多光谱图像两个输入;在网络的编码器和解码器之间添加了跳转连接,减少细节的丢失。

请参见图4,图4为本申请实施例所提供的一种遥感图像的融合系统的结构示意图;

该系统可以包括:

预处理模块100,用于对训练样本进行预处理得到训练数据集;其中,所述训练数据集包括多张全色图像和多张多光谱图像;

第一训练模块200,用于将所述训练数据集输入至特征提取神经网络,对所述特征提取神经网络进行训练得到全色特征信息和多光谱特征信息;

第二训练模块300,用于利用所有所述全色特征信息和所有所述多光谱特征信息训练特征融合神经网络得到融合特征和编码特征图;其中,所述编码特征图由所述特征融合神经网络对所有所述全色图像和所有所述多光谱图像进行编码操作得到;

第三训练模块400,用于将所述融合特征和所述编码特征图输入图像重构神经网络,并利用所述图像重构神经网络的反卷积层进行重建操作得到高分辨率多光谱图像,以便训练所述图像重构神经网络;

模型构建模块500,用于根据训练后的所述特征提取神经网络、训练后的所述特征融合神经网络和训练后的所述图像重构神经网络构建深度神经网络模型,并利用所述深度神经网络模型进行遥感图像融合操作。

可选的,所述预处理模块100包括:

图像组确定单元,用于将所述训练样本中对同一区域采集的全色图像与多光谱图像设置为一个图像组;

下采样单元,用于对每一所述图像组的全色图像和多光谱图像分别下采样得到预设尺寸的下采样全色图像和下采样多光谱图像;

插值处理单元,用于对所述下采样多光谱图像进行逐波段的插值操作,以使插值操作后的下采样多光谱图像中每个波段图像的像素大小与所述下采样多光谱图像中每个波段图像的像素大小相等;

构建单元,用于根据所有所述下采样全色图像和所述下采样多光谱图像构建所述训练数据集。

可选的,所述第一训练模块200包括:

第一特征提取单元,用于将所述下采样全色图像和所述下采样多光谱图像分别输入至特征提取神经网络的第一子卷积神经网络和第二子卷积神经网络得到中间全色特征信息和中间多光谱特征信息;其中,所述第一子卷积神经网络和所述第二子卷积神经网络分别设有非线性激活单元,所述特征提取神经网络包括所述所述第一子卷积神经网络、所述第二子卷积神经网络、所述非线性激活单元和卷积核;

第二特征提取单元,用于将所述中间全色特征信息和所述中间多光谱特征信息输入所述卷积核进行下采样得到所述全色特征信息和所述多光谱特征信息。

由于系统部分的实施例与方法部分的实施例相互对应,因此系统部分的实施例请参见方法部分的实施例的描述,这里暂不赘述。

本申请还提供了一种计算机可读存储介质,其上存有计算机程序,该计算机程序被执行时可以实现上述实施例所提供的步骤。该存储介质可以包括:u盘、移动硬盘、只读存储器(read-onlymemory,rom)、随机存取存储器(randomaccessmemory,ram)、磁碟或者光盘等各种可以存储程序代码的介质。

本申请还提供了一种电子设备,可以包括存储器和处理器,所述存储器中存有计算机程序,所述处理器调用所述存储器中的计算机程序时,可以实现上述实施例所提供的步骤。当然所述电子设备还可以包括各种网络接口,电源等组件。

说明书中各个实施例采用递进的方式描述,每个实施例重点说明的都是与其他实施例的不同之处,各个实施例之间相同相似部分互相参见即可。对于实施例公开的系统而言,由于其与实施例公开的方法相对应,所以描述的比较简单,相关之处参见方法部分说明即可。应当指出,对于本技术领域的普通技术人员来说,在不脱离本申请原理的前提下,还可以对本申请进行若干改进和修饰,这些改进和修饰也落入本申请权利要求的保护范围内。

还需要说明的是,在本说明书中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的状况下,由语句“包括一个……”限定的要素,并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1