一种多头自注意机制的合成孔径雷达图像去噪算法制造技术

本发明专利技术公开了一种多头自注意机制的合成孔径雷达图像去噪算法，具体包括如下步骤：SSDD数据集SAR图像的获取；Gamma变换生成噪点图像；训练图像打包成模型输入；图像网格化处理；编码器计算；迭代模型训练；利用模型进行噪声抑制处理，本发明专利技术在传统噪声抑制的基础上，加入了基于多头自注意力机制神经网络的噪声抑制方式，通过SSDD的开源数据集链接，进行数据集的下载从而获取SSDD数据集的SAR图像，原图片转成灰度图后，在灰度图上进行Gamma变换，生成噪声增强的SAR图像，将SAR图像原图和生成数据作为输入合成DataLoader，在多头自注意机制神经网络中，对DataLoader进行训练，得到权重信息，并利用得到的权重信息，对SAR图像进行噪声抑制以达预期效果。噪声抑制以达预期效果。噪声抑制以达预期效果。

【技术实现步骤摘要】
一种多头自注意机制的合成孔径雷达图像去噪算法


[0001]本专利技术涉及SAR图像噪声抑制
，具体为一种多头自注意机制的合成孔径雷达图像去噪算法。

技术介绍

[0002]合成孔径雷达(SAR)是一种先进的主动式微波对地观测设备，具有全天候、全天时工作的特点，对地面植被有一定穿透能力，能获得类似光学照片的目标图像，在国民经济和军事应用领域有着十分重要的作用，被广泛应用于军事侦察与测绘、海洋的资源监测与探测以及环境监测等领域，SAR成像的系统需要用信号的幅度信息和相干信号的相应位置来提高其分辨率，拥有良好的相干性，SAR图像是接受地物散射回波的相干信号叠加而形成的，因此对噪声高度敏感，噪声影响了对图像的处理和解译，所以去除噪声是SAR图像处理中重要的一个基本步骤，噪声的抑制对SAR图像的后处理极为重要，SAR图像噪声抑制的方法主要有均值滤波、中值滤波、局部滤波、Sigma滤波等等，前者是通用的RGB图像的噪声抑制方式；
[0003]但是由于SAR图像产生的特殊方式，通用的噪声抑制方式并不适用于SAR图像，对SAR图像的噪声抑制效果不好。

技术实现思路

[0004]本专利技术提供一种多头自注意机制的合成孔径雷达图像去噪算法，可以有效解决上述
技术介绍
中提出由于SAR图像产生的特殊方式，通用的噪声抑制方式并不适用于SAR图像，对SAR图像的噪声抑制效果不好的问题。
[0005]为实现上述目的，本专利技术提供如下技术方案：一种多头自注意机制的合成孔径雷达图像去噪算法，在传统噪声抑制的基础上，加入了基于多头自注意力机制神经网络的噪声抑制方式，利用神经网络来实现噪声抑制；
[0006]具体包括如下步骤：
[0007]步骤1，SSDD数据集SAR图像的获取；
[0008]步骤2，Gamma变换生成噪点图像；
[0009]步骤3，训练图像打包成模型输入；
[0010]步骤4，图像网格化处理；
[0011]步骤5，编码器计算；
[0012]步骤6，迭代模型训练；
[0013]步骤7，利用模型进行噪声抑制处理。
[0014]根据上述技术方案，所述步骤1中，SSDD数据集SAR图像的获取主要是指通过SSDD的开源数据集链接，进行数据集的下载需要从数据集SSDD获取1160张SAR图像，其中包括2456个在合成孔径雷达下成像的舰船。
[0015]根据上述技术方案，所述SSDD的开源数据集链接如下：
[0016]https://github.com/TianwenZhang0825/LS
‑
SSDD
‑
v1.0
‑
OPEN。
[0017]根据上述技术方案，所述步骤2中，Gamma变换生成噪点图像主要是指生成训练图像，具体的，主要对原图进行灰度处理得到灰度图，原图片转换成灰度图后，对灰度图进行Gamma增强和变换，生成噪声增强的SAR图像；
[0018]Gamma分布符合以下数学公式：
[0019]根据上述技术方案，在步骤2中，主要采用了numpy.random.gamma函数实现，对Gamma增强图像与原灰度图做相乘处理，得到最终的训练数据集。
[0020]根据上述技术方案，所述步骤3中，训练图像打包成模型输入主要是指将SAR图像原图和生成数据作为输入合成DataLoader，并将训练数据集通过torch中的DataLoader合成固定Batch的数据集合，将加噪图像和原始图像打包作为模型输入。
[0021]根据上述技术方案，所述步骤4中，图像网格化处理主要是指将图像按照预先设定好的进行网格切图，并且按顺序编号，编号图像作为编码器输入。
[0022]根据上述技术方案，所述步骤5中，编码器计算主要是指将打包好的训练图像通过编码器Encoder进行编码，然后将比编码器输出，放入多头自注意机制神经网络中，在多头自注意机制神经网络中，对DataLoader进行训练，得到一组关于噪声增强图像和原图之间的权重信息；
[0023]从而随机生成对应的Key、Question、Value，将结果进行求和归一化后在前馈神经网络中进行计算，输出同样需要进行归一化操作。
[0024]根据上述技术方案，所述步骤6中，迭代模型训练主要是指重复步骤4和步骤5两个步骤，将输入不断的进行迭代计算，利用反向传播机制配合MSLoss进行计算，更新模型计算权重，直到模型训练的Loss下降至稳定不变为止，模型的状态具体包括如下两种情况：
[0025]一、模型处于收敛状态，则训练结束，进入下一步骤；
[0026]二、模型未处于收敛状态，则跳转至步骤4进行重新处理；
[0027]进一步的，利用更新计算所得到的权重信息，对SAR图像进行噪声抑制以达预期效果。
[0028]根据上述技术方案，所述步骤7中，利用模型进行噪声抑制处理是指对图像的噪声进行抑制处理，主要将训练好的pth模型导入torch框架，切换模型状态为eval模式，对SAR需要降噪的图像进行模型推理，推理结果图即为噪声抑制图。
[0029]与现有技术相比，本专利技术的有益效果：
[0030]本专利技术在传统噪声抑制的基础上，加入了基于多头自注意力机制神经网络的噪声抑制方式，提出了一种基于多头自注意机制的SAR图像去噪方法，且利用本方案降噪的SAR图像在其他任务中准确率有明显提升，解决了当前通用的噪声抑制方式并不适用于SAR图像，对SAR图像的噪声抑制效果不好的问题；
[0031]通过SSDD的开源数据集链接，进行数据集的下载从而获取SSDD数据集的SAR图像，原图片转成灰度图后，在灰度图上进行Gamma变换，生成噪声增强的SAR图像，将SAR图像原图和生成数据作为输入合成DataLoader，在多头自注意机制神经网络中，对DataLoader进行训练，得到一组关于噪声增强图像和原图之间的权重信息，并利用得到的权重信息，对SAR图像进行噪声抑制以达预期效果。
附图说明
[0032]附图用来提供对本专利技术的进一步理解，并且构成说明书的一部分，与本专利技术的实施例一起用于解释本专利技术，并不构成对本专利技术的限制。
[0033]在附图中：
[0034]图1是本专利技术去噪算法的流程示意框图；
[0035]图2是本专利技术去噪算法的具体步骤流程图。
具体实施方式
[0036]以下结合附图对本专利技术的优选实施例进行说明，应当理解，此处所描述的优选实施例仅用于说明和解释本专利技术，并不用于限定本专利技术。
[0037]实施例1：如图1
‑
2所示，本专利技术提供一种技术方案，一种多头自注意机制的合成孔径雷达图像去噪算法，在传统噪声抑制的基础上，加入了基于多头自注意力机制神经网络的噪声抑制方式，利用神经网络来实现噪声抑制；
[0038]具体包括如下步骤：
[0039]步骤1，SSDD数据集SAR图像的获取；
[0040]步骤2，Gamm本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：在传统噪声抑制的基础上，加入了基于多头自注意力机制神经网络的噪声抑制方式，利用神经网络来实现噪声抑制；具体包括如下步骤：步骤1，SSDD数据集SAR图像的获取；步骤2，Gamma变换生成噪点图像；步骤3，训练图像打包成模型输入；步骤4，图像网格化处理；步骤5，编码器计算；步骤6，迭代模型训练；步骤7，利用模型进行噪声抑制处理。2.根据权利要求1所述的一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：所述步骤1中，SSDD数据集SAR图像的获取主要是指通过SSDD的开源数据集链接，进行数据集的下载需要从数据集SSDD获取1160张SAR图像，其中包括2456个在合成孔径雷达下成像的舰船。3.根据权利要求2所述的一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：所述SSDD的开源数据集链接如下：https://github.com/TianwenZhang0825/LS
‑
SSDD
‑
v1.0
‑
OPEN。4.根据权利要求1所述的一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：所述步骤2中，Gamma变换生成噪点图像主要是指生成训练图像，具体的，主要对原图进行灰度处理得到灰度图，原图片转换成灰度图后，对灰度图进行Gamma增强和变换，生成噪声增强的SAR图像；Gamma分布符合以下数学公式：5.根据权利要求4所述的一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：在步骤2中，主要采用了numpy.random.gamma函数实现，对Gamma增强图像与原灰度图做相乘处理，得到最终的训练数据集。6.根据权利要求1所述的一种多头自注意机制的合成孔径雷达图像去噪算法，其特征在于：所述步骤3中，训练图像打包成模型输入主要是指将SA...

【专利技术属性】
技术研发人员：翁胧胧，
申请(专利权)人：地卫二空间技术杭州有限公司，
类型：发明
国别省市：