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

uses fftconvolve instead of convolve2d

Browse Source
This commit is contained in:
scottsievert
2015-11-30 23:25:44 -06:00
parent 3b525987f5
commit 5aabc199aa
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/deconvolution.py
+3 -3
View File
@@ -7,7 +7,7 @@ from __future__ import division
import numpy as np
import numpy.random as npr
from scipy.signal import convolve2d
from scipy.signal import fftconvolve
from . import uft
@@ -370,8 +370,8 @@ def richardson_lucy(image, psf, iterations=50, clip=True):
im_deconv = 0.5 * np.ones(image.shape)
psf_mirror = psf[::-1, ::-1]
for _ in range(iterations):
relative_blur = image / convolve2d(im_deconv, psf, 'same')
im_deconv *= convolve2d(relative_blur, psf_mirror, 'same')
relative_blur = image / fftconvolve(im_deconv, psf, 'same')
im_deconv *= fftconvolve(relative_blur, psf_mirror, 'same')
if clip:
im_deconv[im_deconv > 1] = 1