一种基于膨胀卷积图像去噪方法技术

本发明专利技术涉及一种基于膨胀卷积图像去噪方法，具体包括以下步骤，步骤一、训练数据的准备；步骤二、模型的建立；步骤三、将步骤二得到的图像进行编译得到编译后的模型；步骤四、将步骤一中得到的块图像加上加性高斯白噪声得到带有噪声的批量图片；步骤五、将步骤四得到的带有噪声的批量图片进行训练，得到训练后的模型；步骤六、测试数据的准备；步骤七、将得到的测试图像导入到步骤五模型的predict预测函数得到去噪后的图；本发明专利技术的去噪方法不仅可以恢复锐利的边缘和精细的细节，而且还能在光滑的区域产生令人愉快的视觉效果，并且本方法的网络结构由14层组成，能够减少需要的时间，提高效率。

【技术实现步骤摘要】
一种基于膨胀卷积图像去噪方法
本专利技术属于图像去噪
，具体涉及一种基于膨胀卷积图像去噪方法。
技术介绍
图像去噪是指减少数字图像中噪声的过程，现实中的数字图像在数字化和传输过程中常受到成像设备与外部环境噪声干扰等影响，称为含噪图像或噪声图像。DnCNN是用残差学习在隐层中去掉干净的图像，得到一个噪声图像，再用有噪声的输入图像减去噪声图像从而得到清晰的图像。DnCNN采用17层网络，其中第一层是膨胀卷积+非线性激活函数（Relu），第2-16层是膨胀卷积+批归一化+非线性激活函数（Relu），第17层是膨胀卷积，此类网络层数过深，需要的时间长，所以会降低效率。
技术实现思路
为了解决现有技术中存在的上述技术问题，本专利技术提供了以下技术方案：一种基于膨胀卷积图像去噪方法，包括以下步骤，步骤一、训练数据的准备，准备若干张灰度训练图片，设置缩放因子，采用立方插值的方法生成若干张图片一；设置块尺寸和步长，对图片一进行块的裁取，对裁取后的每张图片一再做水平翻转或者翻转90°操作，得到数个块图像；步骤二、模型的建立，第一层，将输入图像采用零填充的方式与滤波器进行卷积操作得到图像x，再将图像x进行非线性激活操作得到图像x1；第二层，将图像x1采用零填充的方式与滤波器进行膨胀因子为1的卷积操作得到图像out，再将图像out进行批归一化处理得到图像out1，然后对图像out1进行非线性激活操作得到图像out2；第三层，将图像out2采用零填充的方式与滤波器进行膨胀因子为2的卷积操作得到图像out3，再将图像out3进行批归一化处理得到图像out4，然后对图像out4进行非线性激活操作得到图像out5；第四层，设置膨胀因子为3，将图像out5重复第三层步骤得到图像out6；第五层，将图像out6采用零填充的方式与滤波器进行膨胀因子为4的卷积操作得到图像out7，再将图像out7进行批归一化处理得到图像out8，然后将图像out8与第一层得到的图像x1做跳跃连接，形成一个“残差块”，最后将该图像out8进行非线性激活操作得到图像out9，将图像out9作为下一个输入的图像x2；第六层至第九层，将图像x2分别重复第二层至第五层，得到图像x3；第十层至第十三层，将图像x3分别重复第二层至第五层，得到图像x4；第十四层，将图像x4采用零填充的方式与滤波器进行卷积操作得到图像x5，然后将第一层的输入图像与图像x5做相减处理并赋值给图像x5，即得到图像模型；步骤三、将步骤二得到的图像模型采用Adam优化器和均方误差损失函数进行编译得到编译后的模型；步骤四、将步骤一中得到的数个块图像按批量尺寸分成多代，每代需要多次迭代，给每一次迭代的图像分别加上加性高斯白噪声得到带有噪声的批量图片；步骤五、将步骤四得到的带有噪声的批量图片采用学习率衰减的方法进行训练，得到训练后的模型；步骤六、测试数据的准备，准备多张测试图片，对测试图片分别加上加性高斯白噪声得到测试图像；步骤七、将得到的测试图像导入到步骤五预测函数模型中，求到去噪后的图像。与现有技术相比，本专利技术取得的有益效果为：本专利技术的去噪方法不仅可以恢复锐利的边缘和精细的细节，而且还能在光滑的区域产生令人愉快的视觉效果，并且本方法的网络结构由14层组成，能够减少需要的时间，提高效率。以下将结合附图及实施例对本专利技术做进一步详细说明。附图说明图1是本专利技术去噪方法的网络结构示意图。图2是用BM3D方法处理噪声后的图片。图3是用MLP方法处理噪声后的图片。图4是用EPLL方法处理噪声后的图片。图5是用WNNM方法处理噪声后的图片。图6是用TNRD方法处理噪声后的图片。图7是用DnCNN方法处理噪声后的图片。图8是本专利技术的方法处理噪声后的图片。具体实施方式为进一步阐述本专利技术达成预定目的所采取的技术手段及功效，以下结合附图及实施例对本专利技术的具体实施方式、结构特征及其功效，详细说明如下。在现有技术中不仅存在网络层数深的问题，还存在有“网格问题”，由于膨胀卷积是将零填充在卷积核中的两个像素之间，因此该卷积核的感受野仅覆盖具有棋盘模式的区域--仅采样非零值的位置，所以丢失了一些邻近信息。当膨胀因子增加时，“网格问题”会变得更严重，通常是在感受野较大的较高层：卷积核太稀疏，无法覆盖任何局部信息，因为非零值离得太远了。为解决现有技术中存在的网络层数深，并且会出现“网格问题”，本实施例提供一种基于膨胀卷积图像去噪方法，该方法的网络是由14层组成，第一层是卷积(膨胀因子默认为1)+非线性激活函数（Relu），第2、3、4、5层分别是由膨胀因子为1、2、3、4的层，它们4层组成一个“残差块”，第6、7、8、9层组成一个“残差块”，第10、11、12、13层组成一个“残差块”，在“残差块”里需要注意的是，跳远连接是连在批归一化与非线性激活函数（Relu）之间的，其网络结构如图1所示。上述方法具体包括以下步骤，步骤一、训练数据的准备，准备400张180×180的灰度训练图片，设置缩放因子为[1,0.9,0.8,0.7]，采用立方插值的方法生成400×4张图片一，尺寸分别为180×180，162×162，144×144，126×126；设置块尺寸（patchsize）为40×40，步长为10，对得到的400×4张图片一进行块（patch）的裁取，对裁取后的每张图片一再做水平翻转或者翻转90°操作，这样180×180的图片可裁成225张40×40的，160×160的可裁成169张40×40的，144×144可裁成121张40×40的，126×126可裁成81张40×40的，总共得到400×(225+169+121+81)=238400块（patches）图像；步骤二、模型的建立，第一层、将输入图像（input）采用零填充的方式与64个3×3的滤波器进行卷积操作得到图像x，再将图像x进行非线性激活操作（Relu）得到图像x1；第二层、将图像x1采用零填充的方式与64个3×3的滤波器进行膨胀因子为1的卷积操作得到图像out，再将图像out进行批归一化处理得到图像out1，然后对图像out1进行非线性激活操作（Relu）得到图像out2；第三层、将图像out2采用零填充的方式与64个3×3的滤波器进行膨胀因子为2的卷积操作得到图像out3，再将图像out3进行批归一化处理得到图像out4，然后对图像out4进行非线性激活操作得到图像out5；第四层、设置膨胀因子为3，将图像out5重复第三层步骤得到图像out6；第五层、将图像out6采用零填充的方式与64个3×3的滤波器进行膨胀因子为4的卷积操作得到图像out7，再将图像out7进行批归一化处理得到图像out8，然后将图像out8与第一层得到的图像x1做跳跃连接（跳跃连接可以从某一网络层获取激活，然后迅速反馈给另外一层，甚至是神经网络的更深层，能够节省训练需要的时间，收敛更快），形成一个“残差块”，最后将该图像out8进行非线性激活操作（Relu）得到图像out9，将图像out9作为下一个输入的图像x2；第六层至第九层、将图像x2分别重复第二层至第五层，得到图像x3；第十层至第十三层、将图像x3分别重复第二层至第五层，得到图像x4；第十四层、将图像x4采用零填充的方式与1个3×本文档来自技高网...

【技术保护点】
1.一种基于膨胀卷积图像去噪方法，其特征在于：包括以下步骤，步骤一、训练数据的准备，准备若干张灰度训练图片，设置缩放因子，采用立方插值的方法生成若干张图片一；设置块尺寸和步长，对图片一进行块的裁取，对裁取后的每张图片一再做水平翻转或者翻转90°操作，得到数个块图像；步骤二、模型的建立，第一层，将输入图像采用零填充的方式与滤波器进行卷积操作得到图像x，再将图像x进行非线性激活操作得到图像x1；第二层，将图像x1采用零填充的方式与滤波器进行膨胀因子为1的卷积操作得到图像out，再将图像out进行批归一化处理得到图像out1，然后对图像out1进行非线性激活操作得到图像out2；第三层，将图像out2采用零填充的方式与滤波器进行膨胀因子为2的卷积操作得到图像out3，再将图像out3进行批归一化处理得到图像out4，然后对图像out4进行非线性激活操作得到图像out5；第四层，设置膨胀因子为3，将图像out5重复第三层步骤得到图像out6；第五层，将图像out6采用零填充的方式与滤波器进行膨胀因子为4的卷积操作得到图像out7，再将图像out7进行批归一化处理得到图像out8，然后将图像out8与第一层得到的图像x1做跳跃连接，形成一个“残差块”，最后将该图像out8进行非线性激活操作得到图像out9，将图像out9作为下一个输入的图像x2；第六层至第九层，将图像x2分别重复第二层至第五层，得到图像x3；第十层至第十三层，将图像x3分别重复第二层至第五层，得到图像x4；第十四层，将图像x4采用零填充的方式与滤波器进行卷积操作得到图像x5，然后将第一层的输入图像与图像x5做相减处理并赋值给图像x5，即得到图像模型；步骤三、将步骤二得到的图像模型采用Adam优化器和均方误差损失函数进行编译得到编译后的模型；步骤四、将步骤一中得到的数个块图像按批量尺寸分成多代，每代需要多次迭代，给每一次迭代的图像分别加上加性高斯白噪声得到带有噪声的批量图片；步骤五、将步骤四得到的带有噪声的批量图片采用学习率衰减的方法进行训练，得到训练后的预测函数模型；步骤六、测试数据的准备，准备多张测试图片，对测试图片分别加上加性高斯白噪声得到测试图像；步骤七、将得到的测试图像导入到步骤五中预测函数模型中，求到去噪后的图像。...

【技术特征摘要】
1.一种基于膨胀卷积图像去噪方法，其特征在于：包括以下步骤，步骤一、训练数据的准备，准备若干张灰度训练图片，设置缩放因子，采用立方插值的方法生成若干张图片一；设置块尺寸和步长，对图片一进行块的裁取，对裁取后的每张图片一再做水平翻转或者翻转90°操作，得到数个块图像；步骤二、模型的建立，第一层，将输入图像采用零填充的方式与滤波器进行卷积操作得到图像x，再将图像x进行非线性激活操作得到图像x1；第二层，将图像x1采用零填充的方式与滤波器进行膨胀因子为1的卷积操作得到图像out，再将图像out进行批归一化处理得到图像out1，然后对图像out1进行非线性激活操作得到图像out2；第三层，将图像out2采用零填充的方式与滤波器进行膨胀因子为2的卷积操作得到图像out3，再将图像out3进行批归一化处理得到图像out4，然后对图像out4进行非线性激活操作得到图像out5；第四层，设置膨胀因子为3，将图像out5重复第三层步骤得到图像out6；第五层，将图像out6采用零填充的方式与滤波器进行膨胀因子为4的卷积操作得到图像out7，再将图像out7进行...

【专利技术属性】
技术研发人员：彭亚丽，宁豆，张鲁，
申请(专利权)人：陕西师范大学，
类型：发明
国别省市：陕西,61