From 72577e9764fe1cf1920f2f394d0de64c7be71fb7 Mon Sep 17 00:00:00 2001 From: "Gregory R. Lee" Date: Thu, 11 Aug 2016 10:52:37 -0400 Subject: [PATCH] FIX: correct channel range in the multichannel case --- skimage/restoration/_denoise.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skimage/restoration/_denoise.py b/skimage/restoration/_denoise.py index 377cefc3..7f6c6b7b 100644 --- a/skimage/restoration/_denoise.py +++ b/skimage/restoration/_denoise.py @@ -463,7 +463,7 @@ def denoise_wavelet(img, sigma=None, wavelet='db1', mode='soft', if multichannel: out = np.empty_like(img) - for c in range(img.ndim): + for c in range(img.shape[-1]): out[..., c] = _wavelet_threshold(img[..., c], wavelet=wavelet, mode=mode, sigma=sigma) else: