一种基于空洞化的核预测网络的单图像去噪方法和装置制造方法及图纸

本发明专利技术公开了一种基于空洞化的核预测网络的单图像去噪方法和装置，包括：构建包含特征提取模块、特征压缩模块、核预测模块以及图像重建模块的基于空洞化的核预测网络；对空洞化的核预测网络进行参数优化后备用；应用时，将噪声图像输入至参数优化后的基于空洞化的核预测网络，利用特征提取单元从噪声图像中提取高级特征图，利用特征压缩模块将高级特征图压缩成压缩特征图，利用核预测模块根据高级特征图提取预测卷积核，利用图像重建模块基于预测卷积核和压缩特征图获得预测图后与噪声图像融合，得到去噪后图像。通过为单图像去噪问题引入核预测网络，并使用空洞卷积、多核通道融合、压缩特征图等技巧实现单图像去噪任务，大大提升了去噪效果。大大提升了去噪效果。大大提升了去噪效果。

【技术实现步骤摘要】
一种基于空洞化的核预测网络的单图像去噪方法和装置


[0001]本专利技术涉及计算机科学图像处理领域，尤其涉及一种基于空洞化的核预测网络的单图像去噪方法和装置。

技术介绍

[0002]图像去噪是图像处理领域的一大基础问题，近年来深度学习网络的快速发展为去噪算法提供了一种高效的解决方案。但是传统卷积网络在全图像素上使用相同的卷积核，如申请公布号为CN109658348A公开的基于深度学习的联合噪声估计和图像去噪方法，没有充分利用像素区域之间存在的特征差异。
[0003]尽管已有一些较新的研究在尝试引入注意力机制来强调这一局部差异性，如申请公布号为CN111260591A公开的一种基于注意力机制的图像自适应去噪方法，但是现有深度学习网络往往仍无法较好地保持精细的图像细节。
[0004]核预测网络(Kernel Prediction Network)作为一种能为每个像素预测其特有卷积核的深度学习网络，充分评估了每个像素特有的邻域特征。但是该方法目前尚未在单图像去噪问题上得到较好应用，其主要难点在于在仅有的一张参考图像中提取到足够的信息用本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于空洞化的核预测网络的单图像去噪方法，其特征在于，包括以下步骤：构建包含特征提取模块、特征压缩模块、核预测模块以及图像重建模块的基于空洞化的核预测网络，其中，核预测模块包括至少3个不同空洞步长且用于产生分支预测卷积核的空洞分支、和用于融合所有分支预测卷积核的核融合操作以得到预测卷积核；对空洞化的核预测网络进行参数优化后备用；应用时，将噪声图像输入至参数优化后的基于空洞化的核预测网络，利用特征提取单元从噪声图像中提取高级特征图，利用特征压缩模块将高级特征图压缩成压缩特征图，利用核预测模块根据高级特征图提取预测卷积核，利用图像重建模块基于预测卷积核和压缩特征图获得预测图后与噪声图像融合，得到去噪后图像。2.如权利要求1所述的基于空洞化的核预测网络的单图像去噪方法，其特征在于，所述特征提取模块包括若干卷积层和若干残差模块，用于从噪声图像中提取包含多个通道的高级特征图，所述残差模块包括卷积层、线性整流函数和点加操作。3.如权利要求1所述的基于空洞化的核预测网络的单图像去噪方法，其特征在于，所述特征压缩模块包括至少一个卷积层，用于对高级特征图进行压缩，得到与噪声图像通道数相同的压缩特征图。4.如权利要求1所述的基于空洞化的核预测网络的单图像去噪方法，其特征在于，所述核预测模块包括3个不同空洞步长的空洞分支，每个空洞分支用于产生各自的分支预测卷积核，还包括核融合操作，用于融合所有分支预测卷积核得到预测卷积核。5.如权利要求1或4所述的基于空洞化的核预测网络的单图像去噪方法，其特征在于，每个空洞分支包括若干个空洞残差块和至少一个...

【专利技术属性】
技术研发人员：田翔，谢才扬，陈耀武，
申请(专利权)人：浙江大学，
类型：发明
国别省市：