From 2b7bf24c1fb8abfdd32505cf1045d2038a89a1d4 Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Fri, 26 Sep 2014 19:11:32 +1000 Subject: [PATCH] Add shape fix for deconvolution functions --- skimage/restoration/deconvolution.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/skimage/restoration/deconvolution.py b/skimage/restoration/deconvolution.py index 51c2c705..ade2516d 100644 --- a/skimage/restoration/deconvolution.py +++ b/skimage/restoration/deconvolution.py @@ -130,7 +130,8 @@ def wiener(image, psf, balance, reg=None, is_real=True, clip=True): wiener_filter = np.conj(trans_func) / (np.abs(trans_func)**2 + balance * np.abs(reg)**2) if is_real: - deconv = uft.uirfft2(wiener_filter * uft.urfft2(image)) + deconv = uft.uirfft2(wiener_filter * uft.urfft2(image), + shape=image.shape) else: deconv = uft.uifft2(wiener_filter * uft.ufft2(image)) @@ -320,7 +321,7 @@ def unsupervised_wiener(image, psf, reg=None, user_params=None, is_real=True, # Empirical average \approx POSTMEAN Eq. 44 x_postmean = x_postmean / (iteration - params['burnin']) if is_real: - x_postmean = uft.uirfft2(x_postmean) + x_postmean = uft.uirfft2(x_postmean, shape=image.shape) else: x_postmean = uft.uifft2(x_postmean)