Deep learning algorithm for Gaussian noise removal from images

被引:8
作者
Aghajarian, Mickael [1 ]
McInroy, John E. [1 ]
Muknahallipatna, Suresh [1 ]
机构
[1] Univ Wyoming, Coll Engn & Appl Sci, Dept Elect & Comp Engn, Laramie, WY 82071 USA
关键词
deep learning; image denoising; Gaussian noise; convolutional neural networks;
D O I
10.1117/1.JEI.29.4.043005
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
A deep learning algorithm for Gaussian noise removal from both grayscale and color images is developed. As opposed to most existing discriminative methods that train a specific model for each noise level, the proposed method can handle a wide range of noise levels using only two trained models, one for low noise levels and the other for high noise levels. In the proposed algorithm, the training process consists of three successive steps. In the first step, a classifier is trained to classify the noisy and clean images. In the second step, a denoiser network aims to remove the noise in the image features that are extracted by the trained classifier. Finally, a decoder is utilized to map back the denoised images features into images pixels. To evaluate the performance of the model, the Berkeley segmentation dataset of 68 images (BSDS68) and 12 widely used images are used, and the denoising performance for additive white Gaussian noise is compared with several state-of-the-art methods in terms of peak signal-to-noise ratio (PSNR) and visual quality. For grayscale image denoising of BSDS68, our method gives the highest PSNR on all noise levels (significant mean improvement of 0.99). For color image denoising of BSDS68, except for one low noise level, the proposed method gives the highest PSNR on all other noise levels (mean improvement of 0.3). (C) 2020 SPIE and IS&T
引用
收藏
页数:14
相关论文
共 24 条
[1]   A High-Quality Denoising Dataset for Smartphone Cameras [J].
Abdelhamed, Abdelrahman ;
Lin, Stephen ;
Brown, Michael S. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1692-1700
[2]   Salt-and-pepper noise removal using modified mean filter and total variation minimization [J].
Aghajarian, Mickael ;
McInroy, John E. ;
Wright, Cameron H. G. .
JOURNAL OF ELECTRONIC IMAGING, 2018, 27 (01)
[3]  
[Anonymous], 2011, P 14 INT C ARTIFICIA, DOI DOI 10.1177/1753193410395357
[4]  
[Anonymous], 2009, Advances in Neural Information Processing Systems
[5]  
[Anonymous], 2016, P 29 IEEE C COMPUTER
[6]  
[Anonymous], 2015, IEEE C COMPUTER VISI
[7]  
Burger HC, 2012, PROC CVPR IEEE, P2392, DOI 10.1109/CVPR.2012.6247952
[8]   Trainable Nonlinear Reaction Diffusion: A Flexible Framework for Fast and Effective Image Restoration [J].
Chen, Yunjin ;
Pock, Thomas .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) :1256-1272
[9]   Nonlocality-Reinforced Convolutional Neural Networks for Image Denoising [J].
Cruz, Cristovao ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
IEEE SIGNAL PROCESSING LETTERS, 2018, 25 (08) :1216-1220
[10]   Image denoising by sparse 3-D transform-domain collaborative filtering [J].
Dabov, Kostadin ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2007, 16 (08) :2080-2095