wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-09-10 12:35:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Some improvements to new deconvolution code

Browse Source
This commit is contained in:
Christoph Deil
2013-12-10 22:45:09 +01:00
committed by François Orieux
parent 17cc22802c
commit e25ccf61f6
4 changed files with 37 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/deconvolution/tests/test_deconvolution.py
+12 -17
View File
@@ -1,38 +1,33 @@
import warnings
from os.path import abspath, dirname, join as pjoin
import numpy as np
import numpy.testing.assert_array_almost_equal
from scipy.signal import convolve2d
from skimage.data import camera
from skimage import deconvolution
from scipy.signal import convolve2d as conv2
from skimage import data, deconvolution
# Test deconvolution
# ===========================
test_img = data.camera().astype(np.float)
test_img = camera().astype(np.float)
def test_wiener():
psf = np.ones((5, 5))
data = conv2(test_img, psf, 'same')
data = convolve2d(test_img, psf, 'same')
np.random.seed(0)
data += 0.1 * data.std() * np.random.standard_normal(data.shape)
deconvolued = deconvolution.wiener(data, psf, 25)
numpy.testing.assert_array_almost_equal(deconvolued,
np.load("./camera_wiener.npy"))
path = pjoin(dirname(abspath(__file__)), 'camera_wiener.npy')
np.testing.assert_array_almost_equal(deconvolued, np.load(path))
def test_unsupervised_wiener():
psf = np.ones((5, 5))
data = conv2(test_img, psf, 'same')
data = convolve2d(test_img, psf, 'same')
np.random.seed(0)
data += 0.1 * data.std() * np.random.standard_normal(data.shape)
deconvolued, _ = deconvolution.unsupervised_wiener(data, psf)
numpy.testing.assert_array_almost_equal(deconvolued,
np.load("./camera_unsup.npy"))
path = pjoin(dirname(abspath(__file__)), 'camera_unsup.npy')
np.testing.assert_array_almost_equal(deconvolued, np.load(path))
def test_rychardson_lucy():
def test_richardson_lucy():
return True