[go: up one dir, main page]

CN112801906B - Loop Iterative Image Denoising Method Based on Recurrent Neural Network - Google Patents

Loop Iterative Image Denoising Method Based on Recurrent Neural Network Download PDF

Info

Publication number
CN112801906B
CN112801906B CN202110146982.5A CN202110146982A CN112801906B CN 112801906 B CN112801906 B CN 112801906B CN 202110146982 A CN202110146982 A CN 202110146982A CN 112801906 B CN112801906 B CN 112801906B
Authority
CN
China
Prior art keywords
image
network
iteration
denoising
loop
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Active
Application number
CN202110146982.5A
Other languages
Chinese (zh)
Other versions
CN112801906A (en
Inventor
牛玉贞
郑路伟
陈钧荣
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Fuzhou University
Original Assignee
Fuzhou University
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Fuzhou University filed Critical Fuzhou University
Priority to CN202110146982.5A priority Critical patent/CN112801906B/en
Publication of CN112801906A publication Critical patent/CN112801906A/en
Application granted granted Critical
Publication of CN112801906B publication Critical patent/CN112801906B/en
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T5/00Image enhancement or restoration
    • G06T5/70Denoising; Smoothing
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10004Still image; Photographic image
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20084Artificial neural networks [ANN]

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • General Physics & Mathematics (AREA)
  • General Health & Medical Sciences (AREA)
  • Molecular Biology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Evolutionary Computation (AREA)
  • Artificial Intelligence (AREA)
  • Biomedical Technology (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Health & Medical Sciences (AREA)
  • Image Analysis (AREA)
  • Image Processing (AREA)

Abstract

The invention relates to a cyclic iterative image denoising method based on a cyclic neural network, which comprises the following steps of S1, obtaining paired original noise images and noiseless images and preprocessing the images to obtain paired image blocks of the noise images and the noiseless images for training; s2, constructing a cyclic iterative image denoising network based on a cyclic neural network, and training by using paired image blocks of a noise image and a noiseless image; and S3, inputting the original noise image to be detected into the trained denoising network to obtain a denoising image. The invention removes noise more cleanly and keeps more image details in a circular iteration mode, thereby effectively reconstructing a de-noised image.

Description

基于循环神经网络的循环迭代图像去噪方法Loop Iterative Image Denoising Method Based on Recurrent Neural Network

技术领域technical field

本发明涉及图像和视频处理技术领域,具体涉及一种基于循环神经网络的循环迭代图像去噪方法。The invention relates to the technical field of image and video processing, in particular to a cyclic iterative image denoising method based on a cyclic neural network.

背景技术Background technique

随着互联网和多媒体技术的快速发展,图像已经成为人类信息交流和信息传递过程中不可或缺的一部分。图像在通信、社交和医学等领域具有研究价值,对现代社会的信息存储和信息交互技术等方面的发展具有实际意义。然而,图像内容不可避免会发生退化,例如相机参数设置引起的图像退化、环境亮度引起的图像退化、图像压缩和解压技术引起的图像质量下降等。退化后的图像将严重影响图像的美感甚至导致图像信息不能被有效地提取出来。一旦图像内容出现物体轮廓不清晰,图像的前景和背景将无法有效地分割出来,甚至更严重地将无法判别图像的内容。因此,需要对退化后的图像进行处理。图像去噪是重建退化后的图像得到更接近无噪声图像内容的必不可少的技术之一。作为一个低级视觉任务,它的计算结果好坏可以直接影响到图像分割、图像分类、目标识别等高级计算机视觉任务。With the rapid development of the Internet and multimedia technology, images have become an indispensable part of the process of human information exchange and information delivery. Image has research value in communication, social and medical fields, and has practical significance for the development of information storage and information interaction technology in modern society. However, image content will inevitably degrade, such as image degradation caused by camera parameter settings, image degradation caused by ambient brightness, image quality degradation caused by image compression and decompression techniques, etc. The degraded image will seriously affect the beauty of the image and even cause the image information cannot be extracted effectively. Once the outline of the object is not clear in the image content, the foreground and background of the image will not be effectively segmented, and even more seriously, the content of the image will not be able to be distinguished. Therefore, it is necessary to process the degraded image. Image denoising is one of the essential techniques to reconstruct the degraded image to get closer to the noise-free image content. As a low-level visual task, its calculation results can directly affect advanced computer vision tasks such as image segmentation, image classification, and target recognition.

图像去噪的目标是重建噪声图像中的图像内容,使得去噪图像中拥有更多的图像细节信息。图像去噪任务具有很长的研究历史,在已经被提出的图像去噪方法中,大致可以将其分为传统方法和基于深度学习的方法。传统方法利用中值滤波、高斯滤波等滤波器对噪声图像进行处理,但由于计算资源的限制和方法本身需要手工提取图像先验的原因,其处理效率较低。这种方法需要经过一些处理和优化,才能应用到实际生活当中。而基于深度学习的方法利用卷积神经网络的自动特征提取能力,并且可以使用传统方法提取的先验信息来帮助卷积神经网络提取图像的特征。因此,基于深度学习的方法在近几年被研究者广泛研究。The goal of image denoising is to reconstruct the image content in the noisy image, so that the denoised image has more image detail information. The image denoising task has a long research history. Among the image denoising methods that have been proposed, they can be roughly divided into traditional methods and methods based on deep learning. Traditional methods use filters such as median filter and Gaussian filter to process noisy images, but due to the limitation of computing resources and the need to manually extract image priors, the processing efficiency is low. This method needs some processing and optimization before it can be applied to real life. The method based on deep learning utilizes the automatic feature extraction ability of the convolutional neural network, and can use the prior information extracted by traditional methods to help the convolutional neural network to extract the features of the image. Therefore, methods based on deep learning have been widely studied by researchers in recent years.

近年来,随着计算机计算能力的提升,基于深度学习的方法飞速发展,基于深度学习的图像去噪方法不断被提出,并且拥有比传统方法更先进的去噪性能。然而目前许多图像去噪方法仍存在一些问题。例如去噪结果过于平滑,纹理损失严重等。若对噪声图像只进行一次性的去噪操作便得到去噪结果,容易造成一次去噪的结果过于平滑而损失图像细节,且结果无法逆转。若对噪声图像进行多次迭代地去噪处理,则可以将噪声图像的噪声每次去除一部分,直到迭代得到的去噪结果可以去除更多的图像噪声并重建出更多图像的纹理为止。并且若在去噪操作前,对噪声图像中的噪声分布进行合理的估计,再将估计的噪声分布信息同时输入去噪网络中。可以对噪声图像的噪声幅度和噪声位置进行更准确地估计和定位,从而更好地对噪声图像实施去噪操作,也能得到性能更好的去噪结果。In recent years, with the improvement of computer computing power, methods based on deep learning have developed rapidly. Image denoising methods based on deep learning have been continuously proposed, and have more advanced denoising performance than traditional methods. However, many current image denoising methods still have some problems. For example, the denoising result is too smooth, the texture loss is serious, etc. If only one denoising operation is performed on the noisy image to obtain the denoising result, it is easy to cause the denoising result to be too smooth and lose image details, and the result cannot be reversed. If multiple iterative denoising processes are performed on the noisy image, a part of the noise of the noisy image can be removed each time until the denoising result obtained by iteration can remove more image noise and reconstruct more image textures. And if the noise distribution in the noisy image is reasonably estimated before the denoising operation, then the estimated noise distribution information is simultaneously input into the denoising network. The noise amplitude and noise position of the noise image can be estimated and positioned more accurately, so that the denoising operation can be performed on the noise image better, and a denoising result with better performance can be obtained.

发明内容Contents of the invention

有鉴于此,本发明的目的在于提供一种基于循环神经网络的循环迭代图像去噪方法,通过循环迭代的方式将噪声去除的更干净的同时保留更多的图像细节,从而有效地重建去噪图像。In view of this, the object of the present invention is to provide a cyclic iterative image denoising method based on a cyclic neural network, which can remove noise more cleanly and retain more image details through cyclic iterations, thereby effectively reconstructing the denoised image. image.

为实现上述目的,本发明采用如下技术方案:To achieve the above object, the present invention adopts the following technical solutions:

一种基于循环神经网络的循环迭代图像去噪方法,包括以下步骤:A cyclic iterative image denoising method based on recurrent neural network, comprising the following steps:

步骤S1:获取成对的原始噪声图像和无噪声图像并预处理,得到用于训练的噪声图像和无噪声图像的成对图像块;Step S1: obtain paired original noise image and noise-free image and preprocess, obtain the paired image blocks of noise image and noise-free image for training;

步骤S2:构建基于循环神经网络的循环迭代图像去噪网络,并使用噪声图像和无噪声图像的成对图像块训练;Step S2: construct the cyclic iterative image denoising network based on the cyclic neural network, and use the paired image block training of the noise image and the noise-free image;

步骤S3:将待测原始噪声图像输入训练后的去噪网络,得到去噪图像。Step S3: Input the original noise image to be tested into the trained denoising network to obtain a denoising image.

所述步骤S1具体为:The step S1 is specifically:

步骤S11:将成对的原始噪声图像和无噪声图像在同样的位置上进行切块,得到多组成对的噪声图像和无噪声图像的图像块;Step S11: cutting the paired original noise image and noise-free image into blocks at the same position to obtain image blocks of pairs of noise images and noise-free images;

步骤S12:将得到的每组成对的图像块进行相同的随机翻转和旋转,对数据进行增强,得到用于训练的成对的噪声图像和无噪声图像的图像块。Step S12: performing the same random flip and rotation on each pair of obtained image blocks, and enhancing the data to obtain a pair of image blocks of a noise image and a noise-free image for training.

进一步的,所述基于循环神经网络的循环迭代图像去噪网络包括噪声估计子网络和去噪子网络。Further, the cyclic iterative image denoising network based on the cyclic neural network includes a noise estimation sub-network and a denoising sub-network.

进一步的,所述噪声估计子网络由一个输入卷积层、m个串联的ResNet残差块,一个GRU模块和一个输出卷积层组成。其中,输入卷积层由卷积核为3×3,步长为1的卷积核组成,输出卷积层由卷积核为1×1,步长为1的卷积核组成;GRU模块的输入为前一次迭代的GRU模块的输出和当前迭代次数i的ResNet残差块的输出经过通道拼接得到的特征z。Further, the noise estimation sub-network consists of an input convolutional layer, m series-connected ResNet residual blocks, a GRU module and an output convolutional layer. Among them, the input convolution layer is composed of a convolution kernel with a convolution kernel of 3×3 and a step size of 1, and the output convolution layer is composed of a convolution kernel with a convolution kernel of 1×1 and a step size of 1; the GRU module The input of is the output of the GRU module of the previous iteration and the output of the ResNet residual block of the current iteration number i, which is the feature z obtained by channel splicing.

进一步的,所述GRU模块的具体计算公式如下:Further, the specific calculation formula of the GRU module is as follows:

a=conv(z),a=conv(z),

b=conv(z),b=conv(z),

c=conv_(z),c = conv_(z),

d=conv(z),d=conv(z),

Figure BDA0002930607880000041
Figure BDA0002930607880000041

Figure BDA0002930607880000042
Figure BDA0002930607880000042

其中,conv(·)包含卷积核为3,步长为1的卷积层和sigmoid激活函数,conv_(·)包含卷积核为3,步长为1的卷积和tanh激活函数,

Figure BDA0002930607880000043
作为输出卷积层的输入,
Figure BDA0002930607880000044
为第i+1次迭代的GRU模块输入的一部分。Among them, conv(·) contains a convolution layer with a convolution kernel of 3 and a step size of 1 and a sigmoid activation function, conv_(·) contains a convolution kernel with a convolution kernel of 3 and a step size of 1 and a convolution and tanh activation function,
Figure BDA0002930607880000043
As input to the output convolutional layer,
Figure BDA0002930607880000044
Part of the input to the GRU module for iteration i+1.

进一步的,所述去噪子网络包含两次循环操作,每次循环操作由相同的编码器、残差模块和解码器组成:Further, the denoising sub-network includes two loop operations, each loop operation is composed of the same encoder, residual module and decoder:

编码器由一个输入卷积层和两个下采样层组成;输入卷积层包含一个卷积核为5×5,步长为1的卷积层和一个GRU模块,下采样层包含一个含有卷积核为5×5、步长为2的卷积层和一个激活函数和一个GRU模块;GRU模块的输入为前一层对应特征尺寸的输出和前一次迭代对应特征尺寸的GRU模块输出经过通道拼接得到的特征,计算方式与噪声估计子网络的GRU模块计算方式一样;编码器部分将网络的特征划分为3种不同的尺度,从大尺度到小尺度分别为F1、F2和F3The encoder consists of an input convolutional layer and two downsampling layers; the input convolutional layer contains a convolutional layer with a convolution kernel of 5×5 and a stride of 1 and a GRU module, and the downsampling layer contains a convolutional layer with convolution A convolutional layer with a product kernel of 5×5 and a step size of 2, an activation function, and a GRU module; the input of the GRU module is the output of the corresponding feature size of the previous layer and the output of the GRU module of the corresponding feature size of the previous iteration through the channel The features obtained by splicing are calculated in the same way as the GRU module of the noise estimation sub-network; the encoder part divides the features of the network into three different scales, from large scale to small scale are F 1 , F 2 and F 3 ;

残差模块由n个串联的ResNet残差块组成,它的输入为编码器部分得到的特征F3,输出的特征为FcThe residual module consists of n series-connected ResNet residual blocks, its input is the feature F 3 obtained by the encoder part, and the output feature is F c ;

解码器由两个上采样层和一个输出卷积层组成,每个上采样层的操作包含一次最近邻插值操作、一个卷积核大小为3×3,步长为1的卷积层和一个ReLU激活函数,输出卷积层是一个卷积核大小为1×1、步长为1的卷积层;第一个上采样层的输入为F3和Fc经过通道拼接操作得到的特征,输出的特征为f3;第二个上采样层的输入为F2和f3经过通道拼接操作得到的特征,输出的特征为f2;输出卷积层的输入为F1和f2经过通道拼接操作得到的特征,输出为去噪图像。The decoder consists of two upsampling layers and an output convolutional layer. The operation of each upsampling layer includes a nearest neighbor interpolation operation, a convolutional layer with a kernel size of 3×3 and a stride of 1, and a ReLU activation function, the output convolutional layer is a convolutional layer with a convolution kernel size of 1×1 and a step size of 1; the input of the first upsampling layer is the feature obtained by F 3 and F c through the channel splicing operation, The output feature is f 3 ; the input of the second upsampling layer is the feature obtained by F 2 and f 3 through the channel splicing operation, and the output feature is f 2 ; the input of the output convolution layer is F 1 and f 2 through the channel The features obtained by the stitching operation are output as a denoised image.

进一步的,所述去噪网络包含t次循环迭代,第一次循环迭代的每个GRU模块输入的一部分为对应特征尺寸的值均为0的特征,从第二次循环迭代开始,每个GRU模块输入的一部分为对应特征尺寸的上一次循环迭代的GRU特征;且噪声估计子网络在第一次循环迭代时的输入为噪声图像Nori,输出为第一次循环迭代的噪声估计图像E1;去噪子网络的在第一次循环迭代的输入为第一次循环迭代的噪声估计图像E1和噪声图像Nori经过通道拼接后的双通道图像,输出为第一次循环迭代的去噪图像De1。噪声估计子网络在从第i(i>1)次循环迭代开始,第i次循环迭代的输入为第i-1次循环迭代的去噪图像Dei-1,输出为第i次循环迭代的噪声估计图像Ei。去噪子网络在从第二次循环迭代开始,第i次循环迭代的输入为第i次循环的噪声估计图像Ei和第i-1次循环迭代的去噪图像Dei-1经过通道拼接后的双通道图像,输出为第i阶段的去噪图像Dei。最终的去噪图像是第t次循环迭代的输出DetFurther, the denoising network includes t loop iterations, and a part of each GRU module input of the first loop iteration is a feature whose corresponding feature size is 0. From the second loop iteration, each GRU A part of the module input is the GRU feature of the last cycle iteration corresponding to the feature size; and the input of the noise estimation subnetwork in the first cycle iteration is the noise image N ori , and the output is the noise estimation image E 1 of the first cycle iteration ; The input of the denoising sub-network in the first loop iteration is the noise estimation image E 1 of the first loop iteration and the noise image N ori after the channel splicing of the dual-channel image, and the output is the denoising of the first loop iteration Image De 1 . The noise estimation sub-network starts from the ith (i>1) loop iteration, the input of the i-th loop iteration is the denoised image De i-1 of the i-1 loop iteration, and the output is the i-th loop iteration’s Noise estimated image E i . The denoising sub-network starts from the second loop iteration, and the input of the i-th loop iteration is the noise estimation image E i of the i-th loop iteration and the denoising image De i-1 of the i-1 loop iteration through channel splicing The final two-channel image is output as the denoising image De i of the i-th stage. The final denoised image is the output De t of the t-th loop iteration.

进一步的,所述基于循环神经网络的循环迭代图像去噪网络模型训练,具体为:Further, the training of the cyclic iterative image denoising network model based on the cyclic neural network is specifically:

(1):将成对的噪声图像和无噪声图像的图像块随机划分为多个批次,每个批次包含N个图像块;(1): Randomly divide the image blocks of the paired noise image and the noise-free image into multiple batches, and each batch contains N image blocks;

(2):以批次为单位,将批次中对应的N个图像块的噪声图像输入步骤S2所述的去噪网络中,得到对应的N个去噪图像;(2): taking the batch as a unit, input the noise images of the corresponding N image blocks in the batch into the denoising network described in step S2, and obtain corresponding N denoising images;

(3):根据基于循环神经网络的循环迭代图像去噪网络的目标损失函数,使用反向传播方法计算网络中各参数的梯度,并利用随机梯度下降方法更新网络的参数;(3): According to the target loss function of the cyclic iterative image denoising network based on the cyclic neural network, use the back propagation method to calculate the gradient of each parameter in the network, and use the stochastic gradient descent method to update the parameters of the network;

(4):以批次为单位重复进行上述(2)-(3),直至基于循环神经网络的循环迭代图像去噪网络的目标损失函数数值趋于平稳,保存网络参数,完成网络的训练过程。(4): Repeat the above (2)-(3) in batches until the value of the target loss function of the cyclic iterative image denoising network based on the cyclic neural network tends to be stable, save the network parameters, and complete the training process of the network .

进一步的,所述基于循环神经网络的循环迭代图像去噪网络的目标损失函数计算如下:Further, the target loss function of the cyclic iterative image denoising network based on the cyclic neural network is calculated as follows:

Loss=λ1Lstg12Lstg2+…+λkLstgk Loss=λ 1 L stg12 L stg2 +...+λ k L stgk

所述目标损失函数中,λ1,λ2,…,λk为每次循环迭代损失的权重,Lstg1为第一次循环迭代的损失函数,具体计算如下:In the target loss function, λ 1 , λ 2 , ..., λ k are the weights of the loss for each loop iteration, and L stg1 is the loss function for the first loop iteration, and the specific calculation is as follows:

Figure BDA0002930607880000061
Figure BDA0002930607880000061

其中,Ngt1表示第一次循环迭代中噪声估计子网络的参考图像,即原始噪声图像与无噪声图像的差值图像,f(·)表示噪声估计子网络,Nori表示第一次循环的输入图像,wf表示噪声估计子网络的模型参数,Igt表示噪声图像的参考图像,g(·)表示去噪子网络,concat(·)表示通道拼接操作,E1表示第一次循环迭代中噪声估计子网络的输出图像,wg表示去噪子网络的模型参数,||·||1表示L1距离,||·||2表示L2距离;Among them, N gt1 represents the reference image of the noise estimation sub-network in the first cycle iteration, that is, the difference image between the original noise image and the noise-free image, f( ) represents the noise estimation sub-network, and N ori represents the The input image, w f represents the model parameters of the noise estimation sub-network, I gt represents the reference image of the noise image, g( ) represents the denoising sub-network, concat( ) represents the channel concatenation operation, E 1 represents the first loop iteration In the output image of the noise estimation sub-network, w g represents the model parameters of the denoising sub-network, ||·|| 1 represents the L 1 distance, ||·|| 2 represents the L 2 distance;

所述目标损失函数中Lstg2,Lstgk为第二次和第k次循环迭代的损失函数,从第i(i>1)次循环迭代开始,第i次循环迭代的损失函数Lstgi具体计算如下:In the target loss function, L stg2 and L stgk are the loss functions of the second and k-th loop iterations, starting from the i-th (i>1) loop iteration, the loss function L stgi of the i-th loop iteration is specifically calculated as follows:

Figure BDA0002930607880000071
Figure BDA0002930607880000071

其中,Ngti表示第i次循环迭代中噪声估计子网络的参考图像,即第i次循环迭代的输出去噪图像与无噪声图像的差值图像,f(·)表示噪声估计子网络,Dei-1表示第i次循环迭代的输出去噪图像,也是第i次循环迭代的输入图像,wf表示噪声估计子网络的模型参数,Igt表示噪声图像的参考图像,g(·)表示去噪子网络,concat(·)表示通道拼接操作,Ei表示第i次循环迭代中噪声估计子网络的输出图像,wg表示去噪子网络的模型参数,||·||1表示L1距离,||·||2表示L2距离。Among them, N gti represents the reference image of the noise estimation sub-network in the i-th loop iteration, that is, the difference image between the output denoised image and the noise-free image of the i-th loop iteration, f( ) denotes the noise estimation sub-network, De i-1 represents the output denoised image of the i-th cycle iteration, which is also the input image of the i-th cycle iteration, w f represents the model parameters of the noise estimation sub-network, I gt represents the reference image of the noise image, and g( ) represents denoising sub-network, concat(·) represents the channel splicing operation, E i represents the output image of the noise estimation sub-network in the i-th loop iteration, w g represents the model parameters of the denoising sub-network, ||·|| 1 represents L 1 distance, || · || 2 means L 2 distance.

本发明与现有技术相比具有以下有益效果:Compared with the prior art, the present invention has the following beneficial effects:

本发明使用多尺度编码器和残差模块,能够有效地提取噪声图像的特征,并通过解码器重建出去噪图像。通过循环迭代的方式将噪声去除的更干净的同时保留更多的图像细节,从而有效地重建去噪图像。The present invention uses a multi-scale encoder and a residual module to effectively extract the features of a noisy image and reconstruct a denoised image through a decoder. By means of loop iteration, the noise is removed more cleanly while retaining more image details, so as to effectively reconstruct the denoised image.

附图说明Description of drawings

图1为本发明方法流程示意图;Fig. 1 is a schematic flow sheet of the method of the present invention;

图2为本发明实施例(k=2)的步骤S2中的网络训练的整体结构示意图;2 is a schematic diagram of the overall structure of the network training in step S2 of the embodiment of the present invention (k=2);

图3为本发明实施例的第一次迭代时噪声估计子网络的网络结构示意图;3 is a schematic diagram of the network structure of the noise estimation sub-network during the first iteration of the embodiment of the present invention;

图4为本发明实施例的第一次迭代时去噪子网络的网络结构示意图。Fig. 4 is a schematic diagram of the network structure of the denoising sub-network in the first iteration of the embodiment of the present invention.

具体实施方式Detailed ways

下面结合附图及实施例对本发明做进一步说明。The present invention will be further described below in conjunction with the accompanying drawings and embodiments.

请参照图1,本发明提供一种基于循环神经网络的循环迭代图像去噪方法,包括以下步骤:Please refer to Fig. 1, the present invention provides a kind of cyclic iteration image denoising method based on cyclic neural network, comprises the following steps:

步骤S1:获取成对的原始噪声图像和无噪声图像并预处理,得到用于训练的噪声图像和无噪声图像的成对图像块;Step S1: obtain paired original noise image and noise-free image and preprocess, obtain the paired image blocks of noise image and noise-free image for training;

步骤S2:构建基于循环神经网络的循环迭代图像去噪网络,并使用噪声图像和无噪声图像的成对图像块训练;Step S2: construct the cyclic iterative image denoising network based on the cyclic neural network, and use the paired image block training of the noise image and the noise-free image;

步骤S3:将待测原始噪声图像输入训练后的去噪网络,得到去噪图像。Step S3: Input the original noise image to be tested into the trained denoising network to obtain a denoising image.

在本实施例中,步骤S1具体为:In this embodiment, step S1 is specifically:

步骤S11:将成对的原始噪声图像和无噪声图像在同样的位置上进行切块,得到多组成对的噪声图像和无噪声图像的图像块;Step S11: cutting the paired original noise image and noise-free image into blocks at the same position to obtain image blocks of pairs of noise images and noise-free images;

步骤S12:将得到的每组成对的图像块进行相同的随机翻转和旋转,对数据进行增强,得到用于训练的成对的噪声图像和无噪声图像的图像块。Step S12: performing the same random flip and rotation on each pair of obtained image blocks, and enhancing the data to obtain a pair of image blocks of a noise image and a noise-free image for training.

在本实施例中,所述基于循环神经网络的循环迭代图像去噪网络包括噪声估计子网络和去噪子网络。In this embodiment, the cyclic iterative image denoising network based on the cyclic neural network includes a noise estimation subnetwork and a denoising subnetwork.

优选的,噪声估计子网络由一个输入卷积层、m个串联的ResNet残差块,一个GRU模块和一个输出卷积层组成。其中,输入卷积层由卷积核为3×3,步长为1的卷积核组成,输出卷积层由卷积核为1×1,步长为1的卷积核组成;GRU模块的输入为前一次迭代的GRU模块的输出和当前迭代次数i的ResNet残差块的输出经过通道拼接得到的特征z。Preferably, the noise estimation sub-network consists of an input convolutional layer, m connected ResNet residual blocks, a GRU module and an output convolutional layer. Among them, the input convolution layer is composed of a convolution kernel with a convolution kernel of 3×3 and a step size of 1, and the output convolution layer is composed of a convolution kernel with a convolution kernel of 1×1 and a step size of 1; the GRU module The input of is the output of the GRU module of the previous iteration and the output of the ResNet residual block of the current iteration number i, which is the feature z obtained by channel splicing.

GRU模块的具体计算公式如下:The specific calculation formula of the GRU module is as follows:

a=conv(z),a=conv(z),

b=comv(z),b=comv(z),

c=conv_(z),c = conv_(z),

d=conv(z),d=conv(z),

Figure BDA0002930607880000091
Figure BDA0002930607880000091

Figure BDA0002930607880000092
Figure BDA0002930607880000092

其中,conv(·)包含卷积核为3,步长为1的卷积层和sigmoid激活函数,conv_(·)包含卷积核为3,步长为1的卷积和tanh激活函数,

Figure BDA0002930607880000101
作为输出卷积层的输入,
Figure BDA0002930607880000102
为第i+1次迭代的GRU模块输入的一部分。Among them, conv(·) contains a convolution layer with a convolution kernel of 3 and a step size of 1 and a sigmoid activation function, conv_(·) contains a convolution kernel with a convolution kernel of 3 and a step size of 1 and a convolution and tanh activation function,
Figure BDA0002930607880000101
As input to the output convolutional layer,
Figure BDA0002930607880000102
Part of the input to the GRU module for iteration i+1.

优选的,去噪子网络包含两次循环操作,每次循环操作由相同的编码器、残差模块和解码器组成:Preferably, the denoising sub-network contains two loop operations, each loop operation consists of the same encoder, residual module and decoder:

编码器由一个输入卷积层和两个下采样层组成;输入卷积层包含一个卷积核为5×5,步长为1的卷积层和一个GRU模块,下采样层包含一个含有卷积核为5×5、步长为2的卷积层和一个激活函数和一个GRU模块;GRU模块的输入为前一层对应特征尺寸的输出和前一次迭代对应特征尺寸的GRU模块输出经过通道拼接得到的特征,计算方式与噪声估计子网络的GRU模块计算方式一样;编码器部分将网络的特征划分为3种不同的尺度,从大尺度到小尺度分别为F1、F2和F3The encoder consists of an input convolutional layer and two downsampling layers; the input convolutional layer contains a convolutional layer with a convolution kernel of 5×5 and a stride of 1 and a GRU module, and the downsampling layer contains a convolutional layer with convolution A convolutional layer with a product kernel of 5×5 and a step size of 2, an activation function, and a GRU module; the input of the GRU module is the output of the corresponding feature size of the previous layer and the output of the GRU module of the corresponding feature size of the previous iteration through the channel The features obtained by splicing are calculated in the same way as the GRU module of the noise estimation sub-network; the encoder part divides the features of the network into three different scales, from large scale to small scale are F 1 , F 2 and F 3 ;

残差模块由n个串联的ResNet残差块组成,它的输入为编码器部分得到的特征F3,输出的特征为FcThe residual module consists of n series-connected ResNet residual blocks, its input is the feature F 3 obtained by the encoder part, and the output feature is F c ;

解码器由两个上采样层和一个输出卷积层组成,每个上采样层的操作包含一次最近邻插值操作、一个卷积核大小为3×3,步长为1的卷积层和一个ReLU激活函数,输出卷积层是一个卷积核大小为1×1、步长为1的卷积层;第一个上采样层的输入为F3和Fc经过通道拼接操作得到的特征,输出的特征为f3;第二个上采样层的输入为F2和f3经过通道拼接操作得到的特征,输出的特征为f2;输出卷积层的输入为F1和f2经过通道拼接操作得到的特征,输出为去噪图像。The decoder consists of two upsampling layers and an output convolutional layer. The operation of each upsampling layer includes a nearest neighbor interpolation operation, a convolutional layer with a kernel size of 3×3 and a stride of 1, and a ReLU activation function, the output convolutional layer is a convolutional layer with a convolution kernel size of 1×1 and a step size of 1; the input of the first upsampling layer is the feature obtained by F 3 and F c through the channel splicing operation, The output feature is f 3 ; the input of the second upsampling layer is the feature obtained by F 2 and f 3 through the channel splicing operation, and the output feature is f 2 ; the input of the output convolution layer is F 1 and f 2 through the channel The features obtained by the stitching operation are output as a denoised image.

去噪网络包含t次循环迭代,第一次循环迭代的每个GRU模块输入的一部分为对应特征尺寸的值均为0的特征,从第二次循环迭代开始,每个GRU模块输入的一部分为对应特征尺寸的上一次循环迭代的GRU特征;且噪声估计子网络在第一次循环迭代时的输入为噪声图像Nori,输出为第一次循环迭代的噪声估计图像E1;去噪子网络的在第一次循环迭代的输入为第一次循环迭代的噪声估计图像E1和噪声图像Nori经过通道拼接后的双通道图像,输出为第一次循环迭代的去噪图像De1。噪声估计子网络在从第i(i>1)次循环迭代开始,第i次循环迭代的输入为第i-1次循环迭代的去噪图像Dei-1,输出为第i次循环迭代的噪声估计图像Ei。去噪子网络在从第二次循环迭代开始,第i次循环迭代的输入为第i次循环的噪声估计图像Ei和第i-1次循环迭代的去噪图像Dei-1经过通道拼接后的双通道图像,输出为第i阶段的去噪图像Dei。最终的去噪图像是第t次循环迭代的输出DetThe denoising network contains t loop iterations. A part of the input of each GRU module in the first loop iteration is a feature whose corresponding feature size is 0. From the second loop iteration, a part of the input of each GRU module is The GRU feature of the last loop iteration corresponding to the feature size; and the input of the noise estimation subnetwork in the first loop iteration is the noise image N ori , and the output is the noise estimation image E 1 of the first loop iteration; the denoising subnetwork The input of the first loop iteration is the noise estimation image E 1 of the first loop iteration and the noise image N ori after channel splicing, and the output is the denoised image De 1 of the first loop iteration. The noise estimation sub-network starts from the ith (i>1) loop iteration, the input of the i-th loop iteration is the denoised image De i-1 of the i-1 loop iteration, and the output is the i-th loop iteration’s Noise estimated image E i . The denoising sub-network starts from the second loop iteration, and the input of the i-th loop iteration is the noise estimation image E i of the i-th loop iteration and the denoising image De i-1 of the i-1 loop iteration through channel splicing The final two-channel image is output as the denoising image De i of the i-th stage. The final denoised image is the output De t of the t-th loop iteration.

优选的,基于循环神经网络的循环迭代图像去噪网络模型训练,具体为:Preferably, the cyclic iteration image denoising network model training based on the cyclic neural network is specifically:

(1):将成对的噪声图像和无噪声图像的图像块随机划分为多个批次,每个批次包含N个图像块;(1): Randomly divide the image blocks of the paired noise image and the noise-free image into multiple batches, and each batch contains N image blocks;

(2):以批次为单位,将批次中对应的N个图像块的噪声图像输入步骤S2所述的去噪网络中,得到对应的N个去噪图像;(2): taking the batch as a unit, input the noise images of the corresponding N image blocks in the batch into the denoising network described in step S2, and obtain corresponding N denoising images;

(3):根据基于循环神经网络的循环迭代图像去噪网络的目标损失函数,使用反向传播方法计算网络中各参数的梯度,并利用随机梯度下降方法更新网络的参数;(3): According to the target loss function of the cyclic iterative image denoising network based on the cyclic neural network, use the back propagation method to calculate the gradient of each parameter in the network, and use the stochastic gradient descent method to update the parameters of the network;

(4):以批次为单位重复进行上述(2)-(3),直至基于循环神经网络的循环迭代图像去噪网络的目标损失函数数值趋于平稳,保存网络参数,完成网络的训练过程。(4): Repeat the above (2)-(3) in batches until the value of the target loss function of the cyclic iterative image denoising network based on the cyclic neural network tends to be stable, save the network parameters, and complete the training process of the network .

优选的,在本实施例中,基于循环神经网络的循环迭代图像去噪网络的目标损失函数计算如下:Preferably, in this embodiment, the target loss function of the cyclic iterative image denoising network based on the cyclic neural network is calculated as follows:

Loss=λ1Lstg12Lstg2+…+λkLstgk Loss=λ 1 L stg12 L stg2 +...+λ k L stgk

所述目标损失函数中,λ1,λ2,…,λk为每次循环迭代损失的权重,Lstg1为第一次循环迭代的损失函数,具体计算如下:In the target loss function, λ 1 , λ 2 , ..., λ k are the weights of the loss for each loop iteration, and L stg1 is the loss function for the first loop iteration, and the specific calculation is as follows:

Figure BDA0002930607880000121
Figure BDA0002930607880000121

其中,Ngt1表示第一次循环迭代中噪声估计子网络的参考图像,即原始噪声图像与无噪声图像的差值图像,f(·)表示噪声估计子网络,Nori表示第一次循环的输入图像,wf表示噪声估计子网络的模型参数,Igt表示噪声图像的参考图像,g(·)表示去噪子网络,concat(·)表示通道拼接操作,E1表示第一次循环迭代中噪声估计子网络的输出图像,wg表示去噪子网络的模型参数,||·||1表示L1距离,||·||2表示L2距离;Among them, N gt1 represents the reference image of the noise estimation sub-network in the first cycle iteration, that is, the difference image between the original noise image and the noise-free image, f( ) represents the noise estimation sub-network, and N ori represents the The input image, w f represents the model parameters of the noise estimation sub-network, I gt represents the reference image of the noise image, g( ) represents the denoising sub-network, concat( ) represents the channel concatenation operation, E 1 represents the first loop iteration In the output image of the noise estimation sub-network, w g represents the model parameters of the denoising sub-network, ||·|| 1 represents the L 1 distance, ||·|| 2 represents the L 2 distance;

所述目标损失函数中Lstg2,Lstgk为第二次和第k次循环迭代的损失函数,从第i(i>1)次循环迭代开始,第i次循环迭代的损失函数Lstgi具体计算如下:In the target loss function, L stg2 and L stgk are the loss functions of the second and k-th loop iterations, starting from the i-th (i>1) loop iteration, the loss function L stgi of the i-th loop iteration is specifically calculated as follows:

Figure BDA0002930607880000131
Figure BDA0002930607880000131

其中,Ngti表示第i次循环迭代中噪声估计子网络的参考图像,即第i次循环迭代的输出去噪图像与无噪声图像的差值图像,f(·)表示噪声估计子网络,Dei-1表示第i次循环迭代的输出去噪图像,也是第i次循环迭代的输入图像,wf表示噪声估计子网络的模型参数,Igt表示噪声图像的参考图像,g(·)表示去噪子网络,concat(·)表示通道拼接操作,Ei表示第i次循环迭代中噪声估计子网络的输出图像,wg表示去噪子网络的模型参数,||·||1表示L1距离,||·||2表示L2距离。Among them, N gti represents the reference image of the noise estimation sub-network in the i-th loop iteration, that is, the difference image between the output denoised image and the noise-free image of the i-th loop iteration, f( ) denotes the noise estimation sub-network, De i-1 represents the output denoised image of the i-th cycle iteration, which is also the input image of the i-th cycle iteration, w f represents the model parameters of the noise estimation sub-network, I gt represents the reference image of the noise image, and g( ) represents denoising sub-network, concat(·) represents the channel splicing operation, E i represents the output image of the noise estimation sub-network in the i-th loop iteration, w g represents the model parameters of the denoising sub-network, ||·|| 1 represents L 1 distance, || · || 2 means L 2 distance.

本领域内的技术人员应明白,本申请的实施例可提供为方法、系统、或计算机程序产品。因此,本申请可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且,本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。Those skilled in the art should understand that the embodiments of the present application may be provided as methods, systems, or computer program products. Accordingly, the present application may take the form of an entirely hardware embodiment, an entirely software embodiment, or an embodiment combining software and hardware aspects. Furthermore, the present application may take the form of a computer program product embodied on one or more computer-usable storage media (including but not limited to disk storage, CD-ROM, optical storage, etc.) having computer-usable program code embodied therein.

本申请是参照根据本申请实施例的方法、设备(系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。The present application is described with reference to flowcharts and/or block diagrams of methods, apparatus (systems), and computer program products according to embodiments of the present application. It should be understood that each procedure and/or block in the flowchart and/or block diagram, and a combination of procedures and/or blocks in the flowchart and/or block diagram can be realized by computer program instructions. These computer program instructions may be provided to a general purpose computer, special purpose computer, embedded processor, or processor of other programmable data processing equipment to produce a machine such that the instructions executed by the processor of the computer or other programmable data processing equipment produce a An apparatus for realizing the functions specified in one or more procedures of the flowchart and/or one or more blocks of the block diagram.

这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中,使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品,该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。These computer program instructions may also be stored in a computer-readable memory capable of directing a computer or other programmable data processing apparatus to operate in a specific manner, such that the instructions stored in the computer-readable memory produce an article of manufacture comprising instruction means, the instructions The device realizes the function specified in one or more procedures of the flowchart and/or one or more blocks of the block diagram.

这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上,使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理,从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。These computer program instructions can also be loaded onto a computer or other programmable data processing device, causing a series of operational steps to be performed on the computer or other programmable device to produce a computer-implemented process, thereby The instructions provide steps for implementing the functions specified in the flow chart or blocks of the flowchart and/or the block or blocks of the block diagrams.

以上所述仅为本发明的较佳实施例,凡依本发明申请专利范围所做的均等变化与修饰,皆应属本发明的涵盖范围。The above descriptions are only preferred embodiments of the present invention, and all equivalent changes and modifications made according to the scope of the patent application of the present invention shall fall within the scope of the present invention.

Claims (6)

1. A cyclic iteration image denoising method based on a cyclic neural network is characterized by comprising the following steps:
s1, acquiring and preprocessing paired original noise images and noiseless images to obtain paired image blocks of the noise images and the noiseless images for training;
s2, constructing a cyclic iterative image denoising network based on a cyclic neural network, and training by using paired image blocks of a noise image and a noiseless image;
s3, inputting the original noise image to be detected into the trained denoising network to obtain a denoising image;
the recurrent iterative image denoising network based on the recurrent neural network comprises a noise estimation sub-network and a denoising sub-network;
the denoising sub-network comprises t times of cyclic operations, and each cyclic operation comprises the same encoder, a residual error module and a decoder:
the noise estimation sub-network consists of an input convolutional layer, m series-connected ResNet residual blocks, a GRU module and an output convolutional layer; wherein, the input convolution layer is composed of convolution kernels with convolution kernel of 3 multiplied by 3 and step length of 1; the output convolution layer is composed of convolution kernels with convolution kernel of 1 multiplied by 1 and step length of 1; the input of the GRU module is a characteristic z obtained by splicing the output of the GRU module of the previous iteration and the output of the ResNet residual block of the current iteration number i through a channel;
the encoder consists of an input convolutional layer and two down-sampling layers; the input convolutional layer comprises a convolutional layer with a convolutional kernel of 5 multiplied by 5 and a step size of 1 and a GRU module, and the downsampling layer comprises a convolutional layer with a convolutional kernel of 5 multiplied by 5 and a step size of 2, an activation function and a GRU module; the input of the GRU module is the characteristics obtained by channel splicing of the output of the corresponding characteristic dimension of the previous layer and the output of the GRU module of the corresponding characteristic dimension of the previous iteration, and the calculation mode is the same as that of the GRU module of the noise estimation subnetwork; the encoder part divides the characteristics of the network into 3 different scales, which are respectively F from large scale to small scale 1 、F 2 And F 3
The residual module is composed of n series ResNet residual blocks, and the input of the residual module is the characteristic F obtained by the encoder part 3 The output is characterized by F c
The decoder consists of two up-sampling layers and an output convolution layer, wherein the operation of each up-sampling layer comprises one-time nearest neighbor interpolation operation, a convolution layer with the convolution kernel size of 3 multiplied by 3 and the step length of 1 and a ReLU activation function, and the output convolution layer is a convolution layer with the convolution kernel size of 1 multiplied by 1 and the step length of 1; the input to the first up-sampling layer is F 3 And F c The output characteristic is f 3 (ii) a The input of the second up-sampling layer is F 2 And f 3 The output characteristic is f 2 (ii) a The input of the output convolutional layer is F 1 And f 2 And outputting the characteristics obtained by the channel splicing operation as a de-noised image.
2. The method for denoising cyclic iterative images based on the recurrent neural network as claimed in claim 1, wherein the step S1 specifically comprises:
step S11: the original noise image and the noiseless image which are paired are cut into blocks at the same position, and a plurality of groups of image blocks of the noise image and the noiseless image which are paired are obtained;
step S12: and carrying out the same random inversion and rotation on each group of paired image blocks, and enhancing the data to obtain the image blocks of paired noisy images and noiseless images for training.
3. The method of claim 1, wherein the GRU module has a specific calculation formula as follows:
a=conv(z),
b=conv(z),
c=conv_(z),
d=conv(z),
Figure FDA0003952509110000031
Figure FDA0003952509110000032
wherein conv (·) comprises a convolution layer with convolution kernel of 3 and step size of 1 and a sigmoid activation function, conv _ (·) comprises a convolution layer with convolution kernel of 3 and step size of 1 and a tanh activation function,
Figure FDA0003952509110000033
as an input to the output convolutional layer,
Figure FDA0003952509110000034
is part of the GRU module input for the (i + 1) th iteration.
4. The recurrent neural network-based recurrent iterative image denoising method of claim 1, wherein the denoising network comprises t recurrent iterations, and a portion of each GRU module input of a first recurrent iteration is a value of a corresponding feature sizeA feature of 0, starting from the second iteration of the loop, a portion of each GRU module input is the GRU feature of the last iteration of the loop corresponding to the feature size; and the input of the noise estimation subnetwork at the first iteration of the loop is a noise image N ori Output as a noise estimate image E for the first iteration of the loop 1 (ii) a The input of the denoising subnetwork in the first iteration of the loop is a noise estimation image E of the first iteration of the loop 1 And a noise image N ori Outputting the two-channel image after channel splicing as a De-noised image De of the first loop iteration 1 (ii) a The noise estimation sub-network starts from the ith (i is more than 1) loop iteration, and the input of the ith loop iteration is the denoised image De of the (i-1) loop iteration i-1 And outputting a noise estimation image E of the ith loop iteration i (ii) a The denoising subnetwork starts from the second loop iteration, and the input of the ith loop iteration is a noise estimation image E of the ith loop i And the De-noised image De of the i-1 th loop iteration i-1 Outputting the two-channel image spliced by the channels as a De-noised image De of the ith stage i (ii) a The final denoised image is the output De of the t-th iteration of the loop t
5. The method for denoising cyclic iterative images based on the cyclic neural network as claimed in claim 1, wherein the training of the cyclic iterative image denoising network model based on the cyclic neural network is specifically:
(1) Randomly dividing image blocks of a pair of a noisy image and a noiseless image into a plurality of batches, each batch comprising N image blocks;
(2): inputting the noise images of the corresponding N image blocks in the batch into the denoising network in the step S2 by taking the batch as a unit to obtain corresponding N denoising images;
(3): calculating the gradient of each parameter in the network by using a back propagation method according to a target loss function of a cyclic iterative image denoising network based on a cyclic neural network, and updating the parameters of the network by using a random gradient descent method;
(4): and (3) repeating the steps (2) to (3) by taking batches as units until the target loss function value of the recurrent iterative image denoising network based on the recurrent neural network tends to be stable, storing the network parameters and finishing the training process of the network.
6. The method of claim 5, wherein the objective loss function of the recurrent neural network-based image denoising network is calculated as follows:
Loss=λ 1 L stg12 L stg2 +...+λ k L stgk
in said objective loss function, λ 1 ,λ 2 ,...,λ k Weight lost for each iteration of the loop, L stg1 For the loss function of the first loop iteration, the specific calculation is as follows:
Figure FDA0003952509110000051
wherein, N gt1 Representing the reference image of the noise estimation sub-network in the first iteration of the loop, i.e. the difference image of the original noisy image and the clean image, f (-) representing the noise estimation sub-network, N ori Representing the input image of the first cycle, w f Model parameters representing a noise estimation sub-network, I gt A reference image representing a noisy image, g (-) representing a denoising subnetwork, concat (-) representing a channel splicing operation, E 1 Output image, w, representing a noise estimation sub-network in a first iteration of the loop g Model parameters representing a de-noising subnetwork, | ·| non-woven phosphor 1 Represents L 1 Distance, | · | luminance 2 Represents L 2 A distance;
l in the objective loss function stg2 ,L stgk For the loss function of the second and kth loop iteration, starting from the ith (i > 1) loop iteration, the loss function L of the ith loop iteration stgi The specific calculation is as follows:
Figure FDA0003952509110000052
wherein N is gti A reference image representing the noise estimation subnetwork in the ith iteration of the loop, i.e. the difference image of the output denoised image and the noiseless image of the ith iteration of the loop, f (-) represents the noise estimation subnetwork, de i-1 The output denoised image representing the i-1 st iteration of the loop, which is also the input image for the i-th iteration of the loop, w f Model parameters representing noise estimation sub-network, I gt A reference image representing a noisy image, g (-) representing a denoising subnetwork, concat (-) representing a channel splicing operation, E i Representing the output image of the noise estimation sub-network in the ith iteration of the loop, w g Model parameters representing a de-noising subnetwork, | ·| non-woven phosphor 1 Represents L 1 Distance, | · | luminance 2 Represents L 2 Distance.
CN202110146982.5A 2021-02-03 2021-02-03 Loop Iterative Image Denoising Method Based on Recurrent Neural Network Active CN112801906B (en)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202110146982.5A CN112801906B (en) 2021-02-03 2021-02-03 Loop Iterative Image Denoising Method Based on Recurrent Neural Network

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110146982.5A CN112801906B (en) 2021-02-03 2021-02-03 Loop Iterative Image Denoising Method Based on Recurrent Neural Network

Publications (2)

Publication Number Publication Date
CN112801906A CN112801906A (en) 2021-05-14
CN112801906B true CN112801906B (en) 2023-02-21

Family

ID=75813924

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110146982.5A Active CN112801906B (en) 2021-02-03 2021-02-03 Loop Iterative Image Denoising Method Based on Recurrent Neural Network

Country Status (1)

Country Link
CN (1) CN112801906B (en)

Families Citing this family (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US11540798B2 (en) 2019-08-30 2023-01-03 The Research Foundation For The State University Of New York Dilated convolutional neural network system and method for positron emission tomography (PET) image denoising
CN113658118B (en) * 2021-08-02 2024-08-27 维沃移动通信有限公司 Image noise level estimation method, device, electronic device and storage medium
CN114119428B (en) * 2022-01-29 2022-09-23 深圳比特微电子科技有限公司 Image deblurring method and device
CN114972981B (en) * 2022-04-19 2024-07-05 国网江苏省电力有限公司电力科学研究院 Power grid power transmission environment observation image denoising method, terminal and storage medium
CN115393227B (en) * 2022-09-23 2023-06-06 南京大学 Self-adaptive enhancement method and system for low-light full-color video images based on deep learning

Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111145123A (en) * 2019-12-27 2020-05-12 福州大学 Detail-preserving image denoising method based on U-Net fusion
CN111192211A (en) * 2019-12-24 2020-05-22 浙江大学 Multi-noise type blind denoising method based on single deep neural network
CN111861925A (en) * 2020-07-24 2020-10-30 南京信息工程大学滨江学院 Image rain removing method based on attention mechanism and gate control circulation unit

Family Cites Families (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US10102613B2 (en) * 2014-09-25 2018-10-16 Google Llc Frequency-domain denoising
CN106559669B (en) * 2015-09-29 2018-10-09 华为技术有限公司 Prognostic chart picture decoding method and device
CN111754438B (en) * 2020-06-24 2021-04-27 安徽理工大学 Underwater image restoration model and restoration method based on multi-branch gated fusion

Patent Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111192211A (en) * 2019-12-24 2020-05-22 浙江大学 Multi-noise type blind denoising method based on single deep neural network
CN111145123A (en) * 2019-12-27 2020-05-12 福州大学 Detail-preserving image denoising method based on U-Net fusion
CN111861925A (en) * 2020-07-24 2020-10-30 南京信息工程大学滨江学院 Image rain removing method based on attention mechanism and gate control circulation unit

Non-Patent Citations (4)

* Cited by examiner, † Cited by third party
Title
Scale-Recurrent Network for Deep Image Deblurring;X.Tao 等;《2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition》;20180623;全文 *
Toward Convolutional Blind Denoising of Real Photographs;S.Guo 等;《2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)》;20190620;全文 *
基于深度学习的交通路标图像识别研究;张文乐;《中国优秀硕士学位论文全文数据库(工程科技II辑)》;20201215;全文 *
盲去模糊的多尺度编解码深度卷积网络;贾瑞明 等;《计算机应用》;20190910;第39卷(第9期);全文 *

Also Published As

Publication number Publication date
CN112801906A (en) 2021-05-14

Similar Documents

Publication Publication Date Title
CN112801906B (en) Loop Iterative Image Denoising Method Based on Recurrent Neural Network
Shen et al. Deep semantic face deblurring
Wang et al. Blur image identification with ensemble convolution neural networks
US20220036167A1 (en) Sorting method, operation method and operation apparatus for convolutional neural network
CN111145123B (en) Detail-preserving image denoising method based on U-Net fusion
CN112733929B (en) Improved Yolo underwater image small target and shielding target detection method
CN111462012A (en) SAR image simulation method for generating countermeasure network based on conditions
CN110675339A (en) Image inpainting method and system based on edge inpainting and content inpainting
CN113657532B (en) Motor magnetic shoe defect classification method
CN110189260B (en) An Image Noise Reduction Method Based on Multi-scale Parallel Gated Neural Network
CN111626960A (en) Image defogging method, terminal and computer storage medium
CN111007566A (en) Curvature-driven diffusion full-convolution network seismic data bad channel reconstruction and denoising method
CN110782406B (en) Image denoising method and device based on information distillation network
CN110969089A (en) Lightweight face recognition system and recognition method under noise environment
CN111368602A (en) Face image blurring degree evaluation method and device, readable storage medium and equipment
CN115205148A (en) Image Deblurring Method Based on Dual Path Residual Network
CN113554084A (en) Vehicle re-identification model compression method and system based on pruning and light-weight convolution
CN116309202A (en) An Unsupervised Low Light Enhancement Method Based on Histogram Equalization Prior
CN113642581A (en) Image semantic segmentation method and system based on coding multipath semantic cross network
CN118172543A (en) A method and system for detecting small targets in infrared images based on deep expansion of transform domain tensor
CN112990215A (en) Image denoising method, device, equipment and storage medium
CN112836602A (en) Behavior recognition method, device, equipment and medium based on spatiotemporal feature fusion
CN115456908A (en) A Robust Self-Supervised Image Denoising Method
CN114841895A (en) Image shadow removing method based on bidirectional mapping network
CN112801909B (en) Image fusion denoising method and system based on U-Net and pyramid module

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant