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: