Denoising convolutional neural network with mask for salt and pepper noise

被引:20
作者
Chen, Jiuning [1 ]
Li, Fang [1 ,2 ]
机构
[1] East China Normal Univ, Sch Math Sci, Shanghai, Peoples R China
[2] East China Normal Univ, Shanghai Key Lab PMMP, Shanghai, Peoples R China
基金
中国国家自然科学基金;
关键词
image denoising; image restoration; convolutional neural nets; usual SPN-denoising restoration equation; perfect restoration condition; clean image; mask-involved loss function; general DnCNN; salt-and-pepper noise; convolutional neural network denoising; SPN denoising methods; SWITCHING MEDIAN FILTER; IMPULSE NOISE; RESTORATION; FRAMEWORK; REMOVAL;
D O I
10.1049/iet-ipr.2019.0096
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In this study, the authors propose a new loss function for denoising convolutional neural network (DnCNN) for salt-and-pepper noise (SPN). Based on the motivation of utilising the mask of SPN, firstly from the usual SPN-denoising restoration equation, the authors establish a perfect restoration condition; the restored image is precisely the clean image if this condition holds. Then they design a mask-involved loss function to encourage the network to satisfy this condition in training progress. Experimental results demonstrate that compared with general DnCNN and other state-of-the-art SPN denoising methods, DnCNN equipped with the proposed loss function involving mask (MaskDnCNN) is more effective, robust and efficient.
引用
收藏
页码:2604 / 2613
页数:10
相关论文
共 32 条
[1]  
[Anonymous], 2017, P IEEE C COMPUTER VI
[2]   Contour Detection and Hierarchical Image Segmentation [J].
Arbelaez, Pablo ;
Maire, Michael ;
Fowlkes, Charless ;
Malik, Jitendra .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (05) :898-916
[3]   Fast Two-Phase Image Deblurring Under Impulse Noise [J].
Cai, Jian-Feng ;
Chan, Raymond H. ;
Nikolova, Mila .
JOURNAL OF MATHEMATICAL IMAGING AND VISION, 2010, 36 (01) :46-53
[4]   Salt-and-pepper noise removal by median-type noise detectors and detail-preserving regularization [J].
Chan, RH ;
Ho, CW ;
Nikolova, M .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2005, 14 (10) :1479-1485
[5]   Space variant median filters for the restoration of impulse noise corrupted images [J].
Chen, T ;
Wu, HR .
IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II-ANALOG AND DIGITAL SIGNAL PROCESSING, 2001, 48 (08) :784-789
[6]   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
[7]  
Chen YJ, 2015, PROC CVPR IEEE, P5261, DOI 10.1109/CVPR.2015.7299163
[8]   Wavelet frame based blind image inpainting [J].
Dong, Bin ;
Ji, Hui ;
Li, Jia ;
Shen, Zuowei ;
Xu, Yuhong .
APPLIED AND COMPUTATIONAL HARMONIC ANALYSIS, 2012, 32 (02) :268-279
[9]   Noise adaptive soft-switching median filter [J].
Eng, HL ;
Ma, KK .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2001, 10 (02) :242-251
[10]   Removal of High Density Salt and Pepper Noise Through Modified Decision Based Unsymmetric Trimmed Median Filter [J].
Esakkirajan, S. ;
Veerakumar, T. ;
Subramanyam, Adabala N. ;
PremChand, C. H. .
IEEE SIGNAL PROCESSING LETTERS, 2011, 18 (05) :287-290