【技术实现步骤摘要】
一种图像超分辨率网络训练方法、装置、设备及存储介质
[0001]本申请涉及人工智能中的图像处理
,尤其涉及一种基于光学图像特征融合的图像超分辨率网络训练方法、装置、计算机设备及存储介质。
技术介绍
[0002]深度神经网络在现实生活中已经取得了广泛的应用,例如图像分类、文本转换以及其他的一些应用。
[0003]最近,基于深度学习的图像超分辨率方面的研究也越来越多,这种方法主要是从大量的低分辨率图像与高分辨率图像对中学习一种非线性的映射关系,然后根据这种映射关系从一张低分辨率的图像中恢复重建出对应的高分辨率图像。
[0004]然而,申请人发现传统基于深度学习的图像超分辨率方法都是针对自然图像也就是可见光图像提出的,对于特征信息较少的单通道SAR图像来说,这些已经提出的图像超分辨率网络并不适用,由此可见,传统的图像超分辨率方法存在无法适用于特征信息较少的单通道SAR图像的问题。
技术实现思路
[0005]本申请实施例的目的在于提出一种基于光学图像特征融合的图像超分辨率网络训练方法、装置、计算机设备及存储介质,以解决传统的图像超分辨率方法存在无法适用于特征信息较少的单通道SAR图像的问题。
[0006]为了解决上述技术问题,本申请实施例提供一种基于光学图像特征融合的图像超分辨率网络训练方法,采用了如下所述的技术方案:
[0007]获取原始数据,其中,所述原始数据包括高分辨率SAR图像以及高分辨率可见光图像;
[0008]对所述高分辨率SAR图像进行下采样操作,得 ...
【技术保护点】
【技术特征摘要】
1.一种基于光学图像特征融合的图像超分辨率网络训练方法,其特征在于,包括下述步骤:获取原始数据,其中,所述原始数据包括高分辨率SAR图像以及高分辨率可见光图像;对所述高分辨率SAR图像进行下采样操作,得到低分辨率SAR图像;将所述低分辨率SAR图像以及所述高分辨率可见光图像输入至图像超分辨率网络进行超分辨率转化操作,得到超分辨率转化图像;计算所述超分辨率转化图像以及所述高分辨率SAR图像的平均像素差的绝对值;将所述平均像素差的绝对值作为损失函数对所述图像超分辨率网络进行参数调优操作,得到目标图像超分辨率网络;所述将所述低分辨率SAR图像以及所述高分辨率可见光图像输入至图像超分辨率网络进行超分辨率转化操作,得到超分辨率转化图像的操作,具体包括下述步骤:对所述低分辨率SAR图像进行图像特征提取操作,得到浅层SAR图像特征;根据可见光特征引导模块对所述浅层SAR图像特征以及所述高分辨率可见光图像进行可见光特征融合操作,得到可见光图像信息增强的SAR图像特征;根据SAR图像特征编码模块对所述可见光图像信息增强的SAR图像特征进行特征编码操作,得到深层SAR图像特征;根据特征解码模块对所述深层SAR图像特征进行局部特征重建操作,得到局部重建特征;将所述局部重建特征输入至恢复卷积层进行通道数恢复操作,得到所述超分辨率转化图像。2.根据权利要求1所述的基于光学图像特征融合的图像超分辨率网络训练方法,其特征在于,所述平均像素差的绝对值L
OFFN
表示为:其中,表示为所述低分辨率SAR图像;表示为所述高分辨率可见光图像;表示为所述高分辨率SAR图像;表示图像超分辨率网络映射函数输出的重建的高分辨率SAR图像;表示第m个重构特征损失的值;‖
·
‖1表示L1范数,即求平均绝对值误差。3.根据权利要求2所述的基于光学图像特征融合的图像超分辨率网络训练方法,其特征在于,所述重构特征损失表示为:其中,表示为第i个所述SAR图像特征编码模块输出的特征图;表示为第log2r
‑
i个所述特征解码模块重构的特征图,其对应的是第i个编码模块的特征图,所以计
算重构的特征与原提取到的特征之间的损失;表示为第i个重构损失的权重,所有重构损失的和为1。4.根据权利要求1所述的基于光学图像特征融合的图像超分辨率网络训练方法,其特征在于,所述根据可见光特征引导模块对所述浅层SAR图像特征以及所述高分辨率可见光图像进行可见光特征融合操作,得到可见光图像信息增强的SAR图像特征的步骤,具体包括下述步骤:对所述低分辨率SAR图像进行上采样操作,得到原始高分辨率SAR图像;分别将所述原始高分辨率SAR图像以及所述高分辨率可见光图像输入至卷积层进行浅层特征提取操作,得到SAR图像浅层特征以及可见光浅层特征;将所述可见光浅层特征输入至OFGM模块进行可见光特征提取操作,得到可见光特征图;将所述可见光特征图以及所述SAR图像浅层特征进行像素相乘操作,得到相乘像素特征;对所述相乘像素特征以及所述SAR图像浅层特征进行特征拼接操作,得到拼接特征;将所述拼接特征输入至所述恢复卷积层进行特征通道数恢复操作,得到可见光图像信息增强的SAR图像特征。5.一种基于光学图像特征融合的图像超分辨率网络训练装置,其特征在于,所述装置包括:原始数据获取模块,用于获取原始数据,其中,所述原始数据包括高分辨率SAR图像以及高分辨率可见光图像;下采样模块,用于对所述高分辨率SAR图像进行下采样操作,得到低分辨率SAR图像;超分辨率转化模块,用于将所述低分辨率SAR图像以及所述高分辨率可...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。