FIX: correct channel range in the multichannel case

2026-06-27 18:44:19 +08:00 · 2016-08-11 10:52:37 -04:00
parent 6f3520b83e
commit 72577e9764
1 changed files with 1 additions and 1 deletions
@@ -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: