增强边缘特征的CBDNet去噪网络优化方法及系统技术方案

本发明专利技术公开了一种增强边缘特征的CBDNet去噪网络优化方法及系统，所述方法包括：获取真实图像数据集和合成图像数据集；将原图像交替输入噪声估计网络，输出噪声估计图；对原图像和噪声估计图进行拼接；将拼接后图像输入非盲去噪网络中，输出去噪后的图像，非盲去噪网络采用U

【技术实现步骤摘要】
增强边缘特征的CBDNet去噪网络优化方法及系统


[0001]本专利技术涉及图像处理
，尤其涉及一种增强边缘特征的CBDNet去噪网络优化方法及系统。

技术介绍

[0002]图像去噪是数字图像处理的重要手段，旨在减少成像过程引入或是人为添加的各种噪声。去噪算法通常作为图像预处理阶段的重要环节，亦或是用于增强图像清晰度。当今，去噪技术追求尽可能地消除所有噪声，并且保证图像中的信息不会因过度去噪而受到损坏。
[0003]目前的图像去噪算法都有着良好的去噪效果。如CBDNet采用了串行分支结构，先通过噪声估计网络来估计噪声水平，再通过非盲去噪网络根据噪声水平来进行去噪操作。然而，虽然CBDNet的去噪效果尚佳，但由于池化过程可能导致细节丢失，且U
‑
Net存在语义鸿沟问题，从而导致图像边缘模糊。
[0004]因此，亟需一种增强边缘特征的CBDNet去噪网络优化方法及系统。

技术实现思路

[0005]本专利技术的目的是提供一种增强边缘特征的CBDNet去噪网络优化方法及系统，以解决上述现有技术中的问题，能够对CBDNet中非盲去噪网络模型进行优化，尽可能地保留边缘细节。
[0006]本专利技术提供了一种增强边缘特征的CBDNet去噪网络优化方法，其中，包括：
[0007]分别获取真实图像数据集和合成图像数据集；
[0008]将选自所述真实图像数据集和所述合成图像数据集中的原图像交替输入噪声估计网络，输出噪声估计图；
[0009]对所述原图像和所述噪声估计图进行拼接；
[0010]将拼接后的所述原图像和所述噪声估计图输入非盲去噪网络中，输出去噪后的图像，其中，所述非盲去噪网络采用U
‑
Net架构，包括若干卷积层、两个双线性插值上采样层和两个IndexNet模块。
[0011]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，所述分别获取真实图像数据集和合成图像数据集，具体包括：
[0012]采用真实成像设备拍摄得到所述真实图像数据集；
[0013]获取无噪图像数据集，并在无噪图像数据集中的各无噪图像中添加泊松
‑
高斯噪声，然后对添加有泊松
‑
高斯噪声的无噪图像进行去马赛克和伽马矫正过程，得到所述合成图像数据集。
[0014]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，所述将选自所述真实图像数据集和所述合成图像数据集中的原图像交替输入噪声估计网络，输出噪声估计图，具体包括：
[0015]输入所述噪声估计网络的原图像与所述噪声估计网络输出的噪声估计图的尺寸相同，所述噪声估计网络采用全卷积网络，且没有池化层和BN层。
[0016]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，输入所述噪声估计网络的原图像和所述噪声估计网络输出的噪声估计图的分辨率大小均为H
×
W
×
3；所述噪声估计网络包括5个ReLU激活的标准卷积层，前4个卷积层包含32个卷积核，最后一个卷积层包含3个卷积核，且所有卷积层的卷积核大小均为3
×
3。
[0017]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，所述对所述原图像和所述噪声估计图进行拼接，具体包括：
[0018]采用concatenate函数将所述原图像与所述噪声估计图拼接成一个分辨率大小为H
×
W
×
6的向量。
[0019]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，所述非盲去噪网络包括5个卷积块、1个卷积层、2个平均池化层、两个双线性插值上采样层和两个IndexNet模块，其中，第一个卷积块由2个卷积层组成，第二个卷积块由3个卷积层组成，第三个卷积块由6个卷积层组成，第四个卷积块由3个卷积层组成，第五个卷积块由2个卷积层组成；第一个IndexNet模块添加在第一个卷积块之后，用于指导第一个平均池化层和第二个双线性插值上采样层，第二个IndexNet模块添加在第二个卷积块之后，用于指导第二个平均池化层和第一个双线性插值上采样层。
[0020]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，各所述IndexNet模块包括1个IndexBlock、Sigmoid激活函数和Softmax激活函数，其中，IndexBlock作为学习模块，由1个卷积层和1个像素重构层构成，
[0021]所述将拼接后的所述原图像和所述噪声估计图输入非盲去噪网络中，输出去噪后的图像，具体包括：
[0022]将拼接后的所述原图像和所述噪声估计图，送入IndexBlock，首先经过卷积核步长为2、输出通道数为4的卷积层，将图像长宽减半，再经过像素重构层将四个特征图合并成一个特征图，其尺寸也恢复为原来的大小；
[0023]IndexBlock输出的图像经过Sigmoid激活函数后用于指导上采样过程，将IndexBlock输出的图像与上采样之前的图像进行元素乘法，再进行上采样操作；
[0024]IndexBlock输出的图像经过Sigmoid函数后再经过Softmax激活函数得到的图像则用于指导下采样过程，将IndexBlock输出的图像先与下采样之前的图像进行元素乘法，再进行下采样操作，并在经过下采样后将图像中每个元素乘4。
[0025]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，所述将拼接后的所述原图像和所述噪声估计图输入非盲去噪网络中，输出去噪后的图像，具体包括：
[0026]将分辨率为H
×
W
×
6的拼接后的所述原图像和所述噪声估计图输入非盲去噪网络，经过2次IndexNet指导的平均池化层后像素逐渐减小，再经过2次IndexNet指导的双线性插值上采样层后像素恢复为原来的尺寸，输出分辨率为H
×
W
×
3的图像。
[0027]如上所述的增强边缘特征的CBDNet去噪网络优化方法，其中，优选的是，根据总损失函数训练所述噪声估计网络和所述非盲去噪网络，并采用随机梯度下降方法进行参数更新，共同训练所述噪声估计网络和所述非盲去噪网络两个网络，其中，所述总损失函数由所述噪声估计网络输出位置的非对称损失函数和全变差损失函数、和所述非盲去噪网络输出
位置的像素级均方误差损失函数加权构成，并且，
[0028]通过以下公式计算所述噪声估计网络输出位置的所述非对称损失函数L
asymm
，
[0029][0030]其中，y
i
表示输入到噪声估计网络的图像，表示噪声强度估计值，σ(y
i
)表示Ground
‑
Truth无噪图像的噪声强度；对于I
e
，e为真时，I
e
＝1本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种增强边缘特征的CBDNet去噪网络优化方法，其特征在于，包括：分别获取真实图像数据集和合成图像数据集；将选自所述真实图像数据集和所述合成图像数据集中的原图像交替输入噪声估计网络，输出噪声估计图；对所述原图像和所述噪声估计图进行拼接；将拼接后的所述原图像和所述噪声估计图输入非盲去噪网络中，输出去噪后的图像，其中，所述非盲去噪网络采用U
‑
Net架构，包括若干卷积层、两个双线性插值上采样层和两个IndexNet模块。2.根据权利要求1所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，所述分别获取真实图像数据集和合成图像数据集，具体包括：采用真实成像设备拍摄得到所述真实图像数据集；获取无噪图像数据集，并在无噪图像数据集中的各无噪图像中添加泊松
‑
高斯噪声，然后对添加有泊松
‑
高斯噪声的无噪图像进行去马赛克和伽马矫正过程，得到所述合成图像数据集。3.根据权利要求1所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，所述将选自所述真实图像数据集和所述合成图像数据集中的原图像交替输入噪声估计网络，输出噪声估计图，具体包括：输入所述噪声估计网络的原图像与所述噪声估计网络输出的噪声估计图的尺寸相同，所述噪声估计网络采用全卷积网络，且没有池化层和BN层。4.根据权利要求3所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，输入所述噪声估计网络的原图像和所述噪声估计网络输出的噪声估计图的分辨率大小均为H
×
W
×
3；所述噪声估计网络包括5个ReLU激活的标准卷积层，前4个卷积层包含32个卷积核，最后一个卷积层包含3个卷积核，且所有卷积层的卷积核大小均为3
×
3。5.根据权利要求4所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，所述对所述原图像和所述噪声估计图进行拼接，具体包括：采用concatenate函数将所述原图像与所述噪声估计图拼接成一个分辨率大小为H
×
W
×
6的向量。6.根据权利要求1所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，所述非盲去噪网络包括5个卷积块、1个卷积层、2个平均池化层、两个双线性插值上采样层和两个IndexNet模块，其中，第一个卷积块由2个卷积层组成，第二个卷积块由3个卷积层组成，第三个卷积块由6个卷积层组成，第四个卷积块由3个卷积层组成，第五个卷积块由2个卷积层组成；第一个IndexNet模块添加在第一个卷积块之后，用于指导第一个平均池化层和第二个双线性插值上采样层，第二个IndexNet模块添加在第二个卷积块之后，用于指导第二个平均池化层和第一个双线性插值上采样层。7.根据权利要求6所述的增强边缘特征的CBDNet去噪网络优化方法，其特征在于，各所述IndexNet模块包括1个IndexBlock、Sigmoid激活函数和Softmax激活函数，其中，IndexBlock作为学习模块，由1个卷积层和1个像素重构层构成，所述将拼接后的所述原图像和所述噪声估计图输入非盲去噪网络中，输出去噪后的图像，具体包括：
将拼接后的所述原图像和所述噪声估计图，送入IndexBlock，首先经过卷积核步长为2、输出通道数为4的卷积层，将图像长宽减半，再经过像素重构层将四个特征图合并成一个特征图，其尺寸也恢复为原来的大小；IndexBlock输出的图像经过Sigmoid激活函数后用于指导上采样过程，将IndexBlock输出的图像与上采样之前的图像进行元素乘法，再进行上采样操作；IndexBlock输出的图像经过Sigmoid函数后再经过Soft...

【专利技术属性】
技术研发人员：单宇翔，高扬华，郁钢，
申请(专利权)人：浙江中烟工业有限责任公司，
类型：发明
国别省市：