Image Restoration Using Gaussian Mixture Models With Spatially Constrained Patch Clustering
Milad Niknejad, Hossein Rabbani*, Senior Member, IEEE, and Massoud Babaie-Zadeh, Senior Member, IEEE
Abstract:
In this paper we address the problem of recovering degraded images using multivariate Gaussian Mixture Model (GMM) as a prior. The GMM framework in our method for image restoration is based on the assumption that the accumulation of similar patches in a neighborhood are derived from a multivariate Gaussian probability distribution with a specific covariance and mean. Previous methods of image restoration with GMM have not considered spatial (geometric) distance between patches in clustering. Our conducted experiments show that in the case of constraining Gaussian estimates into a finite-sized windows, the patch clusters are more likely to be derived from the estimated multivariate Gaussian distributions, i.e., the proposed statistical patch-based model provides a better goodness-of-fit to statistical properties of natural images. A novel approach for computing aggregation weights for image reconstruction from recovered patches is introduced which is based on similarity degree of each patch to the estimated Gaussian clusters. The results admit that in the case of image denoising, our method is highly comparable to the state-of-the-art methods, and our image interpolation method outperforms previous state-of-the-art methods.
Niknejad, M.; Rabbani, H.; Babaei-Zadeh, M., "Image Restoration Using Gaussian Mixture Models With Spatially Constrained Patch Clustering," IEEE Transactions on Image Processing, vol. 24, no. 11, pp. 3624-3636, Nov. 2015.
URL: http://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=7128671&isnumber=4358840
Codes: The Matlab codes can be downloaded from here.
This is a fast version of our codes: New Codes ( Thanks Prof. Figueridio for his helps!)
* Note: If you use these codes for a paper, thesis,... please cite this paper and mention in "Acknowledgement" that you have received the codes from the authors.