FFDNet: Toward a Fast and Flexible Solution for CNN-Based Image Denoising

被引:1878
作者
Zhang, Kai [1 ,2 ]
Zuo, Wangmeng [1 ]
Zhang, Lei [2 ]
机构
[1] Harbin Inst Technol, Sch Comp Sci & Technol, Harbin 150001, Heilongjiang, Peoples R China
[2] Hong Kong Polytech Univ, Dept Comp, Hong Kong, Hong Kong, Peoples R China
关键词
Image denoising; convolutional neural networks; Gaussian noise; spatially variant noise; SPARSE; FRAMEWORK; NOISE;
D O I
10.1109/TIP.2018.2839891
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Due to the fast inference and good performance, discriminative learning methods have been widely studied in image denoising. However, these methods mostly learn a specific model for each noise level, and require multiple models for denoising images with different noise levels. They also lack flexibility to deal with spatially variant noise, limiting their applications in practical denoising. To address these issues, we present a fast and flexible denoising convolutional neural network, namely FFDNet, with a tunable noise level map as the input. The proposed FFDNet works on downsampled sub-images, achieving a good trade-off between inference speed and denoising performance. In contrast to the existing discriminative denoisers, FFDNet enjoys several desirable properties, including: 1) the ability to handle a wide range of noise levels (i.e., [0, 75]) effectively with a single network; 2) the ability to remove spatially variant noise by specifying a non-uniform noise level map; and 3) faster speed than benchmark BM3D even on CPU without sacrificing denoising performance. Extensive experiments on synthetic and real noisy images are conducted to evaluate FFDNet in comparison with state-of-the-art denoisers. The results show that FFDNet is effective and efficient, making it highly attractive for practical denoising applications.
引用
收藏
页码:4608 / 4622
页数:15
相关论文
共 63 条
[1]   Fast Image Recovery Using Variable Splitting and Constrained Optimization [J].
Afonso, Manya V. ;
Bioucas-Dias, Jose M. ;
Figueiredo, Mario A. T. .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2010, 19 (09) :2345-2356
[2]  
[Anonymous], P IEEE INT C COMP VI
[3]  
[Anonymous], 2012, ADV NEURAL INFORM PR
[4]  
[Anonymous], 2015, ARXIV PREPRINT ARXIV
[5]  
[Anonymous], 2014, ACM T GRAPHICS TOG, DOI DOI 10.1145/2661229.2661260
[6]  
[Anonymous], 1999, Kodak lossless true color image suite
[7]  
[Anonymous], THESIS
[8]  
[Anonymous], 2016, EFFICIENT ORTHOGONAL
[9]  
[Anonymous], 2008, ADV NEURAL INF PROCE
[10]  
[Anonymous], 2017, IEEE C COMPUT VIS PA