wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-27 11:27:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add shape fix for deconvolution functions

Browse Source
This commit is contained in:
Juan Nunez-Iglesias
2014-09-26 19:13:33 +10:00
parent d50afed18e
commit 2b7bf24c1f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/deconvolution.py
+3 -2
View File
@@ -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)