Gaussian Kernel Mixture Network for Single Image Defocus Deblurring

Defocus blur is one kind of blur effects often seen in images, which is challenging to remove due to its spatially variant amount. This paper presents an end-to-end deep learning approach for removing defocus blur from a single image, so as to have an all-in-focus image for consequent vision tasks. First, a pixel-wise Gaussian kernel mixture (GKM) model is proposed for representing spatially variant defocus blur kernels in an efficient linear parametric form, with higher accuracy than existing models. Then, a deep neural network called GKMNet is developed by unrolling a fixed-point iteration of the GKM-based deblurring. The GKMNet is built on a lightweight scale-recurrent architecture, with a scale-recurrent attention module for estimating the mixing coefficients in GKM for defocus deblurring. Extensive experiments show that the GKMNet not only noticeably outperforms existing defocus deblurring methods, but also has its advantages in terms of model complexity and computational efficiency.

[1]  Frédo Durand,et al.  Image and depth from a conventional camera with a coded aperture , 2007, ACM Trans. Graph..

[2]  Tae Hyun Kim,et al.  Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurring , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[3]  Yonina C. Eldar,et al.  Efficient and Interpretable Deep Blind Image Deblurring Via Algorithm Unrolling , 2020, IEEE Transactions on Computational Imaging.

[4]  David Zhang,et al.  Learning Iteration-wise Generalized Shrinkage–Thresholding Operators for Blind Deconvolution , 2016, IEEE Transactions on Image Processing.

[5]  Alexander Wong,et al.  Squeeze-and-Attention Networks for Semantic Segmentation , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[6]  Luca Antiga,et al.  Automatic differentiation in PyTorch , 2017 .

[7]  Alexei A. Efros,et al.  The Unreasonable Effectiveness of Deep Features as a Perceptual Metric , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[8]  Yonina C. Eldar,et al.  An Algorithm Unrolling Approach to Deep Image Deblurring , 2019, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[9]  Hui Ji,et al.  Estimating Defocus Blur via Rank of Local Patches , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[10]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[11]  Li Xu,et al.  Discriminative Blur Detection Features , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[12]  Jian Sun,et al.  Poisson matting , 2004, ACM Trans. Graph..

[13]  A. N. Rajagopalan,et al.  Spatially-Attentive Patch-Hierarchical Network for Adaptive Motion Deblurring , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  A. N. Rajagopalan,et al.  Region-Adaptive Dense Network for Efficient Motion Deblurring , 2019, AAAI.

[15]  Hongdong Li,et al.  Deep Stacked Hierarchical Multi-Patch Network for Image Deblurring , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[16]  Fatih Porikli,et al.  Deblur and deep depth from single defocus image , 2021, Mach. Vis. Appl..

[17]  Yong Xu,et al.  Attentive deep network for blind motion deblurring on dynamic scenes , 2021, Comput. Vis. Image Underst..

[18]  Jianping Shi,et al.  Just noticeable defocus blur detection and estimation , 2015, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[19]  Hui Ji,et al.  Deep Learning for Handling Kernel/model Uncertainty in Image Deconvolution , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[20]  Bernhard Schölkopf,et al.  Learning to Deblur , 2014, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[21]  Dit-Yan Yeung,et al.  Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting , 2015, NIPS.

[22]  Sungkil Lee,et al.  Deep Defocus Map Estimation Using Domain Adaptation , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[23]  Hongdong Li,et al.  Deblurring by Realistic Blurring , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[24]  Jonathan T. Barron,et al.  Learning Single Camera Depth Estimation Using Dual-Pixels , 2019, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

[25]  Rynson W. H. Lau,et al.  Learning Fully Convolutional Networks for Iterative Non-blind Deconvolution , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[26]  Yanning Zhang,et al.  Learning Deep Gradient Descent Optimization for Image Deconvolution , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[27]  Michael S. Brown,et al.  Defocus Deblurring Using Dual-Pixel Data , 2020, ECCV.

[28]  Rob Fergus,et al.  Fast Image Deconvolution using Hyper-Laplacian Priors , 2009, NIPS.

[29]  Claudio Rosito Jung,et al.  Edge-Based Defocus Blur Estimation With Adaptive Scale Selection , 2018, IEEE Transactions on Image Processing.

[30]  Yi Wang,et al.  Scale-Recurrent Network for Deep Image Deblurring , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[31]  Guangming Shi,et al.  Denoising Prior Driven Deep Neural Network for Image Restoration , 2018, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[32]  Sabine Süsstrunk,et al.  Non-Parametric Blur Map Regression for Depth of Field Extension , 2016, IEEE Transactions on Image Processing.

[33]  Seungyong Lee,et al.  Iterative Filter Adaptive Network for Single Image Defocus Deblurring , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[34]  Kang Wang,et al.  Robust Image Deblurring With an Inaccurate Blur Kernel , 2012, IEEE Transactions on Image Processing.

[35]  Seungyong Lee,et al.  Convergence Analysis of MAP Based Blur Kernel Estimation , 2016, 2017 IEEE International Conference on Computer Vision (ICCV).

[36]  Yu-Bin Yang,et al.  Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections , 2016, NIPS.

[37]  Rynson W. H. Lau,et al.  Dynamic Scene Deblurring Using Spatially Variant Recurrent Neural Networks , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[38]  Michael S. Brown,et al.  Modeling Defocus-Disparity in Dual-Pixel Sensors , 2020, 2020 IEEE International Conference on Computational Photography (ICCP).

[39]  Yuhui Quan,et al.  Variational-EM-Based Deep Learning for Noise-Blind Image Deblurring , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[40]  Wei Su,et al.  Efficient Dynamic Scene Deblurring Using Spatially Variant Deconvolution Network With Optical Flow Guided Training , 2020, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

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

[42]  Se Young Chun,et al.  Multi-Temporal Recurrent Neural Networks For Progressive Non-Uniform Single Image Deblurring With Incremental Temporal Training , 2019, ECCV.

[43]  Xin Du,et al.  Estimating Generalized Gaussian Blur Kernels for Out-of-Focus Image Deblurring , 2021, IEEE Transactions on Circuits and Systems for Video Technology.

[44]  Marios Savvides,et al.  Douglas-Rachford Networks: Learning Both the Image Prior and Data Fidelity Terms for Blind Image Deconvolution , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[45]  Hongdong Li,et al.  Dual Pixel Exploration: Simultaneous Depth Estimation and Image Restoration , 2020, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[46]  In-So Kweon,et al.  A Unified Approach of Multi-scale Deep and Hand-Crafted Features for Defocus Estimation , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[47]  D. A. Fish,et al.  Blind deconvolution by means of the Richardson-Lucy algorithm. , 1995 .

[48]  Qibin Hou,et al.  Rotate to Attend: Convolutional Triplet Attention Module , 2020, ArXiv.

[49]  Xiaoyong Shen,et al.  Dynamic Scene Deblurring With Parameter Selective Sharing and Nested Skip Connections , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).