一种针对复杂退化模型下遥感图像的高分辨率重建方法

专利主权项内容

1.一种针对复杂退化模型下遥感图像的高分辨率重建方法，其特征在于，包括以下步骤：S1：获取样本图像数据集，数据集中包括不同场景的M张图像，随机选取图像，将数据集分为训练集和测试集；S2：设计退化模型，将高分辨率遥感图像模糊处理和下采样之后，加入噪声，生成最终的低分辨率遥感图像；S3：训练带掩码的自编码器模型，学习低分辨率遥感图像的先验信息；S4：搭建一个新的重建网络；重建网络和带掩码的自编码器模型同时训练，利用带掩码的自编码器模型学习到的先验信息对低分辨率遥感图像进行高分辨率重建；重建网络结构整体上分为三个部分，具体如下：S41：第一部分进行浅层特征提取；浅层特征提取模块分别利用卷积核大小为3、5、7的卷积层提取多尺度特征，之后将提取到的三种特征在通道维度连结，利用卷积核大小为1×1的卷积层减小通道数，融合多尺度特征；浅层特征提取模块初步提取了遥感图像中的多尺度特征，表示如式(2)：() (2)F0F0=HSFEILR其中表示浅层特征提取的映射函数，表示输入的低分辨率遥感图像，表示提取到的多尺度特征特征；HSFEILRF0S42：第二部分进行深层特征提取，由残差分支和特征分支组成；特征分支将第一部分提取到的浅层特征直接传向深层特征提取网络后方；残差分支使用UNet模型的结构设计，分支中间的一个卷积层将分支分为前后两个部分；前面部分与后面部分均级联r个基础块且结构相互对应，前面部分每个基础块依次由多尺度感受野注意力模块、残差融合块和先验模块组成，后面部分的每个基础块依次由卷积层、多尺度感受野注意力模块、残差融合块和先验模块组成；其中多尺度感受野注意力模块和残差融合块负责解决特征学习子问题，先验模块负责解决先验学习子问题；前面部分基础块的输入来自先验信息以及上一个基础块的输出，前面部分第r个基础块的输出表示为式(3)：(3)其中表示前面部分第r个基础块的映射函数；/>表示第r个基础块的输出，/>表示第r个基础块的前一个基础块的输出；后面部分同样级联多个基础块，后面部分基础块的输入不仅来自先验信息以及上一个基础块的输出，还来自前面部分对应基础块的输出，它们在通道维度上连结，并利用卷积层进行特征融合；后面部分第r个基础块的输出表示为式(4)：/>(4)其中表示后面部分第r个基础块的输出，/>表示后面部分第r个基础块的映射函数，表示通道维度连结操作，/>表示前面部分对应基础块的输出，其中n是网络中模块的总数量，n=前r个基础块+1个中间部分的卷积块+后r个基础块；ConcatS43：第三部分为上采样模块，上采样方式采用亚像素卷积，避免重建过程中伪影的产生，重建出含有丰富细节信息的图像。