一种基于深度学习的自适应图像去噪方法技术

本发明专利技术涉及一种基于深度学习的自适应图像去噪方法，包括如下步骤：步骤1）建立图像集；步骤2)构建自适应深度卷积神经网络；步骤3)训练自适应深度卷积神经网络：设置深度卷积神经网络网络的学习率和动量参数，通过深度学习框架训练所述自适应深度卷积神经网络直至训练达到迭代最大次数，生成训练后的自适应深度卷积神经网络模型；步骤4)图像去噪：将待去噪的图像输入至训练好的自适应深度卷积神经网络模型，得到对应的残差图像，再将待去噪的图像减去残差图像得到去噪后的图像。有益效果：该方法进一步提高和稳定卷积神经网络的训练性能，并且保证去噪性能，节省时间，无需手动调参，能实现未知噪声级图像的盲去噪。

【技术实现步骤摘要】
一种基于深度学习的自适应图像去噪方法
本专利技术属于图像处理
，特别涉及了一种基于深度学习的自适应图像去噪方法。
技术介绍
图像在成像、存储和传输等过程中，由于受传感器缺陷、宽带限制、码流丢失、环境噪声、有损压缩和数模转换等问题影响，导致图像视觉效果变差。那么就会对后续的目标分割、识别、检测和跟踪等深层研究产生影响，进而对图像分析、描述、分类和解释造成干扰。尤其是在近年兴起的模式识别和人工智能领域，图像质量的高低对它们有着决定性的作用。所以，对污染的噪声进行图像去噪得到高质量的图像这一过程的分析是有必要的，同时更是图像处理方向最具基础和前景的学科之一。随着图像技术的广泛使用，图像去噪技术也已经渗透到人类生活的方方面面。近年来，图像去噪方面各种去噪方法。这些去噪方法有三大缺点：第一，这些算法在测试阶段普遍存在复杂优化问题，以牺牲时间来达到高性能去噪效果，使得去噪时间延迟；第二，现有模型普遍是非凸的并且还有一部分是要手动选择参数从而提高去噪性能。第三，训练出的模型都是针对已知具体的噪声值，无法实现未训练图像的盲噪声去噪。无论是以陈运锦等人提出的TNRD(TrainableNonlinearReactionDiffusion，TNRD)[Chen,Y.,&Pock,T.(2016).Trainablenonlinearreactiondiffusion:aflexibleframeworkforfastandeffectiveimagerestoration.IEEETransactionsonPatternAnalysis&MachineIntelligence,39(6),1256-1272.]方法，还是Schmidt和Roth提出的CSF(CascadeofShrinkageFields，CSF)[Schmidt,U.,&Roth,S.(2014).ShrinkageFieldsforEffectiveImageRestoration.IEEEConferenceonComputerVisionandPatternRecognition(pp.2774-2781).IEEEComputerSociety.]方法,它们都较好的平衡了计算效率和去噪质量的关系，得到令人满意的结果。但是由于它们是基于先验知识的分析模型，具体而言，它们在获取图像全部特征结构时受到限制，以及在整个训练阶段都要手动的微调参数，此外这些方法训练出的模型都是针对已知具体的噪声级，无法实现未知噪声级图像的盲去噪。即使目前最优的DnCNN方法[ZhangK,ZuoW,ChenY,etal.BeyondaGaussianDenoiser:ResidualLearningofDeepCNNforImageDenoising.[J].IEEETransactionsonImageProcessing,2017,26(7):3142-3155]打破了传统的去噪方法，利用残差学习完成去噪，也需要预先设定噪声训练图像的标准偏差。如果训练图像的所有标准偏差是相同的，则方法仅适用于特定的噪声级。因此该方法也无法实现未知噪声级图像的盲去噪。
技术实现思路
本专利技术的目的是克服上述
技术介绍
的不足，提供一种基于深度学习的自适应图像去噪方法，保证去噪性能，节省时间，无需手动调参。为了实现上述技术目的，本专利技术的技术方案为：所述基于深度学习的自适应图像去噪方法,包括如下步骤：步骤1)建立图像集：设定有噪声的图像y、对应的理想无噪声图像u以及噪声v，则y＝u+v；步骤2)构建自适应深度卷积神经网络；步骤3)训练自适应深度卷积神经网络：设置深度卷积神经网络网络的学习率和动量参数，通过深度学习框架训练所述自适应深度卷积神经网络直至训练达到迭代最大次数，生成训练后的自适应深度卷积神经网络模型；步骤4)图像去噪：将待去噪的图像输入至训练好的自适应深度卷积神经网络模型，得到对应的残差图像，再将待去噪的图像减去残差图像得到去噪后的图像。所述基于深度学习的自适应图像去噪方法的进一步设计在于，所述步骤1)中噪声v的标准偏差设定为用于表示噪声级的σ。所述基于深度学习的自适应图像去噪方法的进一步设计在于，所述步骤1)中建立图像集的过程具体为：获取伯克利BSDS500图像数据集并在互联网上下载无噪声的图像作为原始无噪声图像集，并采用y＝u+v添加不同σ值的噪声，噪声图像裁剪成n*n大小，处理得到N个噪声-干净训练图像和对应的噪声方标准差其中下标i表示序列号。所述基于深度学习的自适应图像去噪方法的进一步设计在于，所述步骤2)包括：步骤2-1)图像特征提取：对输入的噪声图像yi进行特征提取得到特征图a0,i：a0,i＝fact(W0yi+b0)(1)式(1)中W0是初始层的卷积滤波系数，b0是初始层的偏差，fact为激活层使用的激活函数，fact通过阈值法实现，fact满足：步骤2-2)图像特征变换；将所述特征图进行l次的特征变换，每次变换都是从当前特征图的上一层al-1,i中进行滤波得到当前特征图al,i：al,i＝fact(fbn(Wlal-1,i+bl))(3)式(3)中，fbn是批规范化函数，Wl是第l层的卷积滤波系数，bl是第l层的偏差；步骤2-3)根据式(4)计算残差图像R(yi)；R(yi)＝Wd-1ad-2,i+bd-1(4)式(4)中的Wd-1是第d-1层的卷积滤波系数，ad-,i是d-2层的特征图，bd-1是第d-1层的偏差。所述基于深度学习的自适应图像去噪方法的进一步设计在于，所述步骤2-2)中每一特征图al,i的特征转换层都由一个卷积层、批规范化层和激活层组成，其中卷积核大小为3*3，并采用64个滤波器。所述基于深度学习的自适应图像去噪方法的进一步设计在于，所述步骤3)中训练自适应深度卷积神经网络采用的损失函数如式(5)，式(5)中，i表示序列号，Θ表示训练参数，R为残差图像。有益效果：本专利技术的基于深度学习的自适应图像去噪方法使用阈值法作为激活函数，模型中通过使用批规范化和残差学习进一步提高和稳定卷积神经网络的训练性能，并且保证去噪性能，节省时间，无需手动调参。该方法无论是峰值信噪比PSNR还是结构相似性MSSIM都超过其它去噪方法。该方法既保证了去噪精度和速度，又可以在不降低去噪性能情况下能完成未知噪声级图像的盲去噪，节省了训练成本。附图说明图1为基于深度学习的自适应图像去噪方法的流程图。图2为构造深度残差学习网络的示意图。具体实施方式下面结合附图和具体实施方式对本专利技术的技术方案作进一步详细的说明。本实施例的基于深度学习的自适应图像去噪方法包括如下步骤：步骤1)建立图像集。设有噪声的图像y，对应理想无噪声图像u，噪声v，则y＝u+v。噪声v的标准偏差为σ，σ表示噪声级。获取伯克利BSDS500图像数据集以及在网上下载无噪声的图像作为原始无噪声图像集，并用y＝u+v添加不同σ值的噪声，噪声图像裁剪成n*n大小(n为自然数)，处理得到N(N为自然数)个噪声-干净训练图像和对应的噪声方标准差其中下标i表示序列号。步骤2)构建自适应深度卷积神经网络。步骤2-1)图像特征提取：对输入的第i个噪声图像yi进行特征提取得到特征图a0,i。a0,i＝本文档来自技高网...

【技术保护点】
1.一种基于深度学习的自适应图像去噪方法，其特征在于包括如下步骤：步骤1)建立图像集：设定有噪声的图像y、对应的理想无噪声图像u以及噪声v，则y＝u+v；步骤2)构建自适应深度卷积神经网络；步骤3)训练自适应深度卷积神经网络：设置深度卷积神经网络网络的学习率和动量参数，通过深度学习框架训练所述自适应深度卷积神经网络直至训练达到迭代最大次数，生成训练后的自适应深度卷积神经网络模型；步骤4)图像去噪：将待去噪的图像输入至训练好的自适应深度卷积神经网络模型，得到对应的残差图像，再将待去噪的图像减去残差图像得到去噪后的图像。

【技术特征摘要】
1.一种基于深度学习的自适应图像去噪方法，其特征在于包括如下步骤：步骤1)建立图像集：设定有噪声的图像y、对应的理想无噪声图像u以及噪声v，则y＝u+v；步骤2)构建自适应深度卷积神经网络；步骤3)训练自适应深度卷积神经网络：设置深度卷积神经网络网络的学习率和动量参数，通过深度学习框架训练所述自适应深度卷积神经网络直至训练达到迭代最大次数，生成训练后的自适应深度卷积神经网络模型；步骤4)图像去噪：将待去噪的图像输入至训练好的自适应深度卷积神经网络模型，得到对应的残差图像，再将待去噪的图像减去残差图像得到去噪后的图像。2.根据权利要求1所述的基于深度学习的自适应图像去噪方法，其特征在于所述步骤1)中噪声v的标准偏差设定为用于表示噪声级的σ。3.根据权利要求2所述的基于深度学习的自适应图像去噪方法，其特征在于所述步骤1)中建立图像集的过程具体为：获取伯克利BSDS500图像数据集并在互联网上下载无噪声的图像作为原始无噪声图像集，并采用y＝u+v添加不同σ值的噪声，噪声图像裁剪成n*n大小，处理得到N个噪声-干净训练图像和对应的噪声方标准差其中下标i表示序列号。4.根据权利要求1所述的基于深度学习的自适应图像去噪方法,其特征在于所述步骤2)包括：步骤2-1)图像特征提取：对输入的...

【专利技术属性】
技术研发人员：陈晓，徐畅，
申请(专利权)人：南京信息工程大学，
类型：发明
国别省市：江苏,32