Optics and Precision Engineering, Volume. 31, Issue 6, 920(2023)

Deep learning image denoising based on multi-stage supervised with Res2-Unet

Yan LIU1... Gang CHEN1, Chunyu YU1,*, Shiyun WANG2 and Bin SUN3 |Show fewer author(s)
Author Affiliations
  • 1Nanjing University Posts and Telecommunications,College of Electronic and Optical Engineering, Nanjing20023,China
  • 2Jiangsu North Huguang Opto-Electronics Limited Corporation, Wuxi14035,China
  • 3Nanjing University Posts and Telecommunications, School of Automation, Nanjing21002,China
  • show less

    To restore high quality images from different types of noise images, this study developed a multi-stage supervised deep residual (MSDR) neural network based on Res2-Unet-SE. First, using the neural network, the image denoising task was devised as a multi-stage process. Then, in each processing stage, image blocks with different resolutions were input into a Res2-Unet sub-network to obtain feature information at different scales, and an adaptive learning of the feature fusion information was transferred to the next stage through a channel attention mechanism. Finally, the feature information of different scales was superimposed to achieve high-quality image noise reduction. The BSD400 dataset was selected for training in the experiments, and a Gaussian noise reduction test was performed using the Set12 data set. Real noise reduction test was conducted using the SIDD data set. Compared with the common denoising neural network, the peak signal-to-noise ratios (PSNRs) of the proposed denoising convolutional neural network (DnCNN) improved by 0.03 dB, 0.05 dB, and 0.14 dB when Gaussian noises of σ = 15, 25 and 50, respectively, were added to the image data set. Compared with the latest dual residual block network (DuRN) algorithm, the PSNR of the image denoised using the proposed algorithm was higher by 0.06 dB, 0.57 dB, and 0.39 dB, respectively. For images containing real noise, the PSNR of the image denoised by the proposed algorithm was 0.6 dB higher than that by the convolutional blind denoising network (CBDNET) algorithm. The results indicate that the proposed algorithm is highly robust in the task of image denoising, and it can effectively remove noise and restore the details of an image, as well as fully maintain the global dependence of the image.

    Tools

    Get Citation

    Copy Citation Text

    Yan LIU, Gang CHEN, Chunyu YU, Shiyun WANG, Bin SUN. Deep learning image denoising based on multi-stage supervised with Res2-Unet[J]. Optics and Precision Engineering, 2023, 31(6): 920

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: Jun. 30, 2022

    Accepted: --

    Published Online: Apr. 4, 2023

    The Author Email: YU Chunyu (yucy@njupt.edu.cn)

    DOI:10.37188/OPE.20233106.0920

    Topics