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

adds decision when to use fftconvolve/convolve2d

Browse Source
This commit is contained in:
scottsievert
2015-12-10 19:47:02 -06:00
parent 5aabc199aa
commit 1be8336599
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/deconvolution.py
+11 -2
View File
@@ -365,13 +365,22 @@ def richardson_lucy(image, psf, iterations=50, clip=True):
----------
.. [1] http://en.wikipedia.org/wiki/Richardson%E2%80%93Lucy_deconvolution
"""
direct_time = lambda n, m, k, l: k*l * n*m
def fft_time(m, n, k, l):
return m*np.log(m) + n*np.log(n) + k*np.log(k) + l*np.log(l)
time_ratio = 71.468 * fft_time(*image.shape, *psf.shape)
time_ratio /= direct_time(*image.shape, *psf.shape)
convolve_method = fftconvolve if time_ratio <= 1 else convolve2d
image = image.astype(np.float)
psf = psf.astype(np.float)
im_deconv = 0.5 * np.ones(image.shape)
psf_mirror = psf[::-1, ::-1]
for _ in range(iterations):
relative_blur = image / fftconvolve(im_deconv, psf, 'same')
im_deconv *= fftconvolve(relative_blur, psf_mirror, 'same')
relative_blur = image / convolve_method(im_deconv, psf, 'same')
im_deconv *= convolve_method(relative_blur, psf_mirror, 'same')
if clip:
im_deconv[im_deconv > 1] = 1