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

TST: tests wavelet denoising

Browse Source
This commit is contained in:
Scott Sievert
2016-07-16 11:03:58 -05:00
parent a1cc31e47f
commit fb8f09263c
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/tests/test_denoise.py
+16
View File
@@ -308,5 +308,21 @@ def test_no_denoising_for_small_h():
assert np.allclose(denoised, img)
def test_wavelet_denoising():
img = astro_gray.copy() + 0.1 * np.random.randn(*(astro_gray.shape))
img = np.clip(img, 0, 1)
# less energy in signal
assert restoration.denoise_wavelet(img).sum() <= img.sum()
# test changing noise_std (higher threshold, so less energy in signal)
assert (restoration.denoise_wavelet(img, noise_stdev=0.2).sum() <=
restoration.denoise_wavelet(img, noise_stdev=0.1).sum())
# This works for this particular image (probably more wavelet theory here)
assert (np.sum(restoration.denoise_wavelet(img, wavelet='db2')) <
np.sum(restoration.denoise_wavelet(img, wavelet='db1')))
if __name__ == "__main__":
run_module_suite()