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

MAINT: explicitly specify axis argument

Browse Source
This commit is contained in:
Gregory R. Lee
2016-01-23 11:23:21 -05:00
parent e6a1a19337
commit a30592a22a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/_denoise.py
+1 -1
View File
@@ -178,7 +178,7 @@ def _denoise_tv_chambolle_nd(im, weight=0.1, eps=2.e-4, n_iter_max=200):
g[slices_g] = np.diff(out, axis=ax)
slices_g[ax+1] = slice(None)
norm = np.sqrt((g ** 2).sum(0))[np.newaxis, ...]
norm = np.sqrt((g ** 2).sum(axis=0))[np.newaxis, ...]
E += weight * norm.sum()
tau = 1. / (2.*ndim)
norm *= tau / weight