Joint Demosaicing and Denoising with Perceptual Optimization on a Generative Adversarial Network

Image demosaicing - one of the most important early stages in digital camera pipelines - addressed the problem of reconstructing a full-resolution image from so-called color-filter-arrays. Despite tremendous progress made in the pase decade, a fundamental issue that remains to be addressed is how to assure the visual quality of reconstructed images especially in the presence of noise corruption. Inspired by recent advances in generative adversarial networks (GAN), we present a novel deep learning approach toward joint demosaicing and denoising (JDD) with perceptual optimization in order to ensure the visual quality of reconstructed images. The key contributions of this work include: 1) we have developed a GAN-based approach toward image demosacing in which a discriminator network with both perceptual and adversarial loss functions are used for quality assurance; 2) we propose to optimize the perceptual quality of reconstructed images by the proposed GAN in an end-to-end manner. Such end-to-end optimization of GAN is particularly effective for jointly exploiting the gain brought by each modular component (e.g., residue learning in the generative network and perceptual loss in the discriminator network). Our extensive experimental results have shown convincingly improved performance over existing state-of-the-art methods in terms of both subjective and objective quality metrics with a comparable computational cost.

[1]  Yann LeCun,et al.  Energy-based Generative Adversarial Network , 2016, ICLR.

[2]  Masatoshi Okutomi,et al.  Residual interpolation for color image demosaicking , 2013, 2013 IEEE International Conference on Image Processing.

[3]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[4]  Lei Zhang,et al.  Color demosaicking via directional linear minimum mean square-error estimation , 2005, IEEE Transactions on Image Processing.

[5]  Evgeny Gershikov Optimized Color Transforms for Image Demosaicing , 2013 .

[6]  Oren Kapah,et al.  Demosaicking using artificial neural networks , 2000, Electronic Imaging.

[7]  Soumith Chintala,et al.  Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks , 2015, ICLR.

[8]  Christian Ledig,et al.  Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[9]  Giancarlo Calvagno,et al.  Regularization Approaches to Demosaicking , 2009, IEEE Transactions on Image Processing.

[10]  Jiri Matas,et al.  DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[11]  Dacheng Tao,et al.  DehazeNet: An End-to-End System for Single Image Haze Removal , 2016, IEEE Transactions on Image Processing.

[12]  Kari Pulli,et al.  FlexISP , 2014, ACM Trans. Graph..

[13]  Valero Laparra,et al.  End-to-end Optimized Image Compression , 2016, ICLR.

[14]  Alessandro Foi,et al.  Denoising and interpolation of noisy Bayer data with adaptive cross-color filters , 2008, Electronic Imaging.

[15]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[16]  Ron Kimmel,et al.  Demosaicing: Image Reconstruction from Color CCD Samples , 1998, ECCV.

[17]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[18]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[19]  Andrew W. Fitzgibbon,et al.  Joint Demosaicing and Denoising via Learned Nonparametric Random Fields , 2014, IEEE Transactions on Image Processing.

[20]  Thomas W. Parks,et al.  Joint demosaicing and denoising , 2006, IEEE Transactions on Image Processing.

[21]  Ming Yang,et al.  DeepFace: Closing the Gap to Human-Level Performance in Face Verification , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[22]  Li Fei-Fei,et al.  Perceptual Losses for Real-Time Style Transfer and Super-Resolution , 2016, ECCV.

[23]  Wangmeng Zuo,et al.  COLOR IMAGE DEMOSAICKING VIA DEEP RESIDUAL LEARNING , 2017 .

[24]  Yap-Peng Tan,et al.  Effective use of spatial and spectral correlations for color filter array demosaicking , 2004, IEEE Transactions on Consumer Electronics.

[25]  Léon Bottou,et al.  Wasserstein GAN , 2017, ArXiv.

[26]  Simon Osindero,et al.  Conditional Generative Adversarial Nets , 2014, ArXiv.

[27]  Vishal M. Patel,et al.  Image De-Raining Using a Conditional Generative Adversarial Network , 2017, IEEE Transactions on Circuits and Systems for Video Technology.

[28]  Kwanghoon Sohn,et al.  Interpolation using neural networks for digital still cameras , 2000, IEEE Trans. Consumer Electron..

[29]  Lei Zhang,et al.  Image demosaicing: a systematic survey , 2008, Electronic Imaging.

[30]  Yücel Altunbasak,et al.  Color plane interpolation using alternating projections , 2002, 2002 IEEE International Conference on Acoustics, Speech, and Signal Processing.

[31]  Fan Zhang,et al.  Robust Color Demosaicking With Adaptation to Varying Spectral Correlations , 2009, IEEE Transactions on Image Processing.

[32]  Guillermo Sapiro,et al.  Non-local sparse models for image restoration , 2009, 2009 IEEE 12th International Conference on Computer Vision.

[33]  Masatoshi Okutomi,et al.  Adaptive residual interpolation for color image demosaicking , 2015, 2015 IEEE International Conference on Image Processing (ICIP).

[34]  Aaron C. Courville,et al.  Improved Training of Wasserstein GANs , 2017, NIPS.

[35]  Marshall F. Tappen,et al.  Separable Markov Random Field Model and Its Applications in Low Level Vision , 2013, IEEE Transactions on Image Processing.

[36]  David Zhang,et al.  Color Reproduction From Noisy CFA Data of Single Sensor Digital Cameras , 2007, IEEE Transactions on Image Processing.

[37]  Yoshua Bengio,et al.  Generative Adversarial Networks , 2014, ArXiv.

[38]  Lei Zhang,et al.  Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising , 2016, IEEE Transactions on Image Processing.

[39]  Lei Zhang,et al.  Waterloo Exploration Database: New Challenges for Image Quality Assessment Models , 2017, IEEE Transactions on Image Processing.

[40]  Sergey Levine,et al.  End-to-End Training of Deep Visuomotor Policies , 2015, J. Mach. Learn. Res..

[41]  Takahiro Saito,et al.  Demosaicing approach based on extended color total-variation regularization , 2008, 2008 15th IEEE International Conference on Image Processing.

[42]  Thomas Pock,et al.  Learning joint demosaicing and denoising based on sequential energy minimization , 2016, 2016 IEEE International Conference on Computational Photography (ICCP).

[43]  Karen O. Egiazarian,et al.  Spatially adaptive color filter array interpolation for noiseless and noisy data , 2007, Int. J. Imaging Syst. Technol..

[44]  Frédo Durand,et al.  Deep joint demosaicking and denoising , 2016, ACM Trans. Graph..

[45]  Wei Ye,et al.  Color Image Demosaicing Using Iterative Residual Interpolation , 2015, IEEE Transactions on Image Processing.

[46]  Eero P. Simoncelli,et al.  Image quality assessment: from error visibility to structural similarity , 2004, IEEE Transactions on Image Processing.

[47]  Kai-Lung Hua,et al.  Self-learning approach to color demosaicking via support vector regression , 2012, 2012 19th IEEE International Conference on Image Processing.

[48]  Xin Li,et al.  Demosaicing by successive approximation , 2005, IEEE Transactions on Image Processing.

[49]  Keigo Hirakawa,et al.  An Empirical Bayes Em-Wavelet Unification for Simultaneous Denoising, Interpolation, and/Or Demosaicing , 2006, 2006 International Conference on Image Processing.

[50]  Masatoshi Okutomi,et al.  Minimized-Laplacian residual interpolation for color image demosaicking , 2014, Electronic Imaging.