一种基于多尺度残差注意力网络的SAR图像去噪方法技术

本发明专利技术公开了一种基于多尺度残差注意力网络的SAR图像去噪方法。该方法步骤如下：构建训练集并对训练样本进行归一化操作；搭建多尺度残差注意力网络模型：通过多尺度卷积组提取图像不同尺度的特征，在特征提取过程中设计主干分支和掩码分支，主干分支用来进行特征处理，掩码分支生成注意力权重以提取图像信息的相关特征来区分噪声，在掩码分支的注意力机制中添加非局部模块以获得图像像素之间的长距离依赖关系，使用跳跃连接用以加快网络收敛速度，结合残差连接学习噪声分布；设置多尺度残差注意力网络训练模型参数并进行模型训练；将噪声图像输入到训练好的模型中，得到去噪图像。本发明专利技术在提高去噪性能的同时，较好地保留了图像的细节信息。

【技术实现步骤摘要】
一种基于多尺度残差注意力网络的SAR图像去噪方法
本专利技术涉及遥感图像处理
，特别是一种基于多尺度残差注意力网络的SAR图像去噪方法。
技术介绍
合成孔径雷达(SAR)是一种主动式成像传感器。相比于红外光、可见光等其他遥感技术相比，SAR具有全天时、全天候、穿透力强的优势。凭借这些优势，使得SAR在军事和民用领域具有非常大的价值。然而，SAR在成像过程中往往受到由于散射现象引起的相干斑噪声的影响。正是由于相干斑噪声的存在严重地影响了SAR图像的质量，为后续SAR图像解译任务如场景分割、目标检测、目标识别等带来极大地干扰，大大降低了SAR图像后续处理的效率。因此，在进行图像解译任务前，SAR图像相干斑噪声的去除成为不可或缺的预处理步骤。目前，国内外很多专家学者在SAR图像去噪方面提出了很多方法包括基于空间域滤波方法、基于变换域滤波方法、非局部均值滤波方法等。基于空间域滤波方法如Lee滤波、Frost滤波等具有操作简单且能够实时处理的优点，但容易丢失一定程度的纹理和细节信息。基于变换域滤波方法如小波变换可以有效的保留一定程度的图像纹理及细节信息，但相干斑噪声的抑制并不理想。非局部均值滤波方法在保留细节信息与相干斑噪声的抑制方面提供了良好的折衷，但相似图像块方面的搜索使得计算量变大，带来了繁重的计算负担，从而限制了这类方法的应用。近年来，深度学习逐渐应用于各种计算机视觉与图像处理任务，并已经证明了在这类领域中取得了良好的效果。与浅层结构的滤波算法相比，深度学习更有利于处理复杂的非线性问题。尽管使用基于深度学习的去噪方法能够获得较好的去噪效果，但其网络结构制约着去噪性能，存在某些细节信息的缺失，获得的去噪性能有待提升。
技术实现思路
本专利技术的目的在于提供一种基于多尺度残差注意力网络的SAR图像去噪方法，以提高SAR图像去噪性能并有效保留图像细节信息。实现本专利技术目的的技术解决方案为：一种基于多尺度残差注意力网络的SAR图像去噪方法，步骤如下：步骤1，构建训练集并对训练样本进行归一化操作；步骤2，搭建多尺度残差注意力网络模型；步骤3，设置多尺度残差注意力网络训练模型参数并进行模型训练；步骤4，将噪声图像输入到训练好的模型中，得到去噪图像。进一步地，步骤1中构建训练集的方法，具体步骤如下：首先，选取光学遥感图像作为训练集，图像变换为灰度图使深度为1用来作为无噪声图像；其次，给所有灰度图像添加不同视数幅度形式的相干斑噪声得到合成SAR噪声图像；然后，将得到的无噪声和合成SAR图像噪声图像作为样本对，以步长为10切割成40×40大小的图像，并将每对大小为40×40的图像块进行数据增强操作以扩大训练样本，最终得到图像块，以形成训练样本对；数据增强操作包括对图像块进行翻转操作、90度以及180度的旋转操作；最后，使用归一化方法将所有的训练样本对归一化到0～1之间。进一步地，步骤2所述搭建多尺度残差注意力网络模型，其中网络模型的结构包括1个多尺度卷积组、1个卷积、1个主干分支、1个掩码分支以及1个跳跃连接，其中主干分支包括6个卷积核大小为3×3卷积层与两个跳跃连接；掩码分支用来生成注意力权重，包括1个非局部模块、1个卷积核大小为3×3的大步长卷积层、1个转置卷积层、1个卷积核大小为1×1的卷积层以及Sigmoid函数层，网络具体结构设置如下：SAR噪声图像首先经过多尺度卷积组，即输入分别经卷积核大小为3×3、5×5以及7×7的3个卷积层，并将这3个卷积层的输出结果进行特征拼接操作，提取不同尺度的图像特征，而且这三种不同尺寸的卷积核数量设置为30、20、12；多尺度卷积组中的所有卷积都会跟着一个ReLU激活函数；在多尺度卷积组后分别接入主干分支和掩码分支；掩码分支用于生成注意力权重并与主干分支进行按元素点乘运算，得到的结果与多尺度卷积组的输出即主干分支与掩码分支的输入进行按元素点加运算，得到的输出经过卷积核大小为3×3，通道数为1的卷积操作后与SAR噪声图像进行按元素相减操作，最终得到去噪图像。进一步地，所述主干分支用来进行特征处理，主干分支由6个卷积层及2个跳跃连接组成；在主干分支的所有卷积后面分别添加一个ReLU激活函数以组成6个卷积层；所述主干分支中，分别在输入与第二个卷积层的输出之间、第三个卷积层输出与第五个卷积层输出之间添加跳跃连接；所述掩码分支按顺序由1个非局部模块、1个卷积层、1个大步长卷积、1个转置卷积、1个卷积核大小为1×1的卷积以及Sigmoid函数组成；所述卷积层由卷积核大小为3×3的卷积及ReLU激活函数组成；所述大步长卷积的卷积核大小为3×3，步长设置为2；所述转置卷积的卷积核大小为6×6，步长为2，padding设为2。进一步地，步骤3所述设置多尺度残差注意力网络训练模型参数并进行模型训练，具体为：训练网络模型过程中，训练轮数60轮，批量大小Batch_size大小为64，采用Adam优化方式，初始学习率设为0.001，而后每经过30轮次学习率除以10，其余卷积核个数即通道数都为64，其余卷积步长都为1，其余padding按公式(1)计算，选取公式(2)作为损失函数，并以最小化损失函数为目标来训练多尺度非局部残差网络模型；其中，padding表示填充值，kernel_size表示卷积核尺寸大小；式(1)中的除法运算表示得到的商值取整；其中Lloss为损失函数，H，W分别表示图像的高和宽，Y，X分别表示去噪图像和含噪图像。本专利技术与现有技术相比，其显著优点为：(1)多尺度卷积可以提取不同尺度的图像特征，提高网络的泛化性能；(2)注意力机制能够更好地学习到图像的特征并区分噪声，进而提高网络性能，跳跃连接则能够加快网络收敛速度并防止梯度消失；(3)残差学习解决了网络的退化问题，非局部模块则可以获得图像像素之间的长距离依赖关系，进而获取更多的上下文信息；(4)能够提高SAR图像去噪性能，并有效保留图像细节信息。附图说明图1是本专利技术基于多尺度残差注意力网络的SAR图像去噪方法的流程图。图2是本专利技术基于多尺度残差注意力网络的SAR图像去噪方法模型结构图。具体实施方式本专利技术一种基于多尺度残差注意力网络的SAR图像去噪方法，步骤如下：步骤1，构建训练集并对训练样本进行归一化操作；步骤2，搭建多尺度残差注意力网络模型；步骤3，设置多尺度残差注意力网络训练模型参数并进行模型训练；步骤4，将噪声图像输入到训练好的模型中，得到去噪图像。进一步地，步骤1中构建训练集的方法，具体步骤如下：首先，选取光学遥感图像作为训练集，图像变换为灰度图使深度为1用来作为无噪声图像；其次，给所有灰度图像添加不同视数幅度形式的相干斑噪声得到合成SAR噪声图像；然后，将得到的无噪声和合成SAR图像噪声图像作为样本对，以步长为10切割成40×40大小的图像，并将每对大小为40×40的图像块进本文档来自技高网...

【技术保护点】
1.一种基于多尺度残差注意力网络的SAR图像去噪方法，其特征在于，步骤如下：/n步骤1，构建训练集并对训练样本进行归一化操作；/n步骤2，搭建多尺度残差注意力网络模型；/n步骤3，设置多尺度残差注意力网络训练模型参数并进行模型训练；/n步骤4，将噪声图像输入到训练好的模型中，得到去噪图像。/n

【技术特征摘要】
1.一种基于多尺度残差注意力网络的SAR图像去噪方法，其特征在于，步骤如下：
步骤1，构建训练集并对训练样本进行归一化操作；
步骤2，搭建多尺度残差注意力网络模型；
步骤3，设置多尺度残差注意力网络训练模型参数并进行模型训练；
步骤4，将噪声图像输入到训练好的模型中，得到去噪图像。


2.根据权利要求1所述的基于多尺度残差注意力网络的SAR图像去噪方法，其特征在于，步骤1中构建训练集的方法，具体步骤如下：
首先，选取光学遥感图像作为训练集，图像变换为灰度图使深度为1用来作为无噪声图像；
其次，给所有灰度图像添加不同视数幅度形式的相干斑噪声得到合成SAR噪声图像；
然后，将得到的无噪声和合成SAR图像噪声图像作为样本对，以步长为10切割成40×40大小的图像，并将每对大小为40×40的图像块进行数据增强操作以扩大训练样本，最终得到图像块，以形成训练样本对；数据增强操作包括对图像块进行翻转操作、90度以及180度的旋转操作；
最后，使用归一化方法将所有的训练样本对归一化到0～1之间。


3.根据权利要求1所述的基于多尺度残差注意力网络的SAR图像去噪方法，其特征在于，步骤2所述搭建多尺度残差注意力网络模型，其中网络模型的结构包括1个多尺度卷积组、1个卷积、1个主干分支、1个掩码分支以及1个跳跃连接，其中主干分支包括6个卷积核大小为3×3卷积层与两个跳跃连接；掩码分支用来生成注意力权重，包括1个非局部模块、1个卷积核大小为3×3的大步长卷积层、1个转置卷积层、1个卷积核大小为1×1的卷积层以及Sigmoid函数层，网络具体结构设置如下：
SAR噪声图像首先经过多尺度卷积组，即输入分别经卷积核大小为3×3、5×5以及7×7的3个卷积层，并将这3个卷积层的输出结果进行特征拼接操作，提取不同尺度的图像特征，而且这三种不同尺寸的卷积核数量设置为30、20、12；多尺度卷积组中的所有卷积都会跟着一个ReLU激活函数；在...

【专利技术属性】
技术研发人员：陶诗飞，申仕煜，叶晓东，王昊，李男，黄鑫宇，李新逸，
申请(专利权)人：南京理工大学，
类型：发明
国别省市：江苏;32