Merge pull request #849 from samuelstjean/patch-5

nlmeans use num threads option in 3d
dipy · Jan 29, 2016 · 6fecc49 · 6fecc49
2 parents 2f588db + a7c95a7
commit 6fecc49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dipy/denoise/nlmeans.py b/dipy/denoise/nlmeans.py
@@ -36,7 +36,7 @@ def nlmeans(arr, sigma, mask=None, patch_radius=1, block_radius=5,
         sigma = np.ones(arr.shape, dtype=np.float64) * sigma
         return nlmeans_3d(arr, mask, sigma,
                           patch_radius, block_radius,
-                          rician).astype(arr.dtype)
+                          rician, num_threads).astype(arr.dtype)
 
     elif arr.ndim == 4:
         denoised_arr = np.zeros_like(arr)