wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-08-20 12:50:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added return_full_prob kwarg to solve call if pyamg not present.

Browse Source
This commit is contained in:
JDWarner
2012-08-27 11:48:32 -05:00
parent 8955dad32e
commit feca48cc49
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/segmentation/random_walker_segmentation.py
+2 -1
View File
@@ -366,7 +366,8 @@ def random_walker(data, labels, beta=130, depth=1., mode='bf', tol=1.e-3,
"""pyamg (http://code.google.com/p/pyamg/)) is needed to use
this mode, but is not installed. The 'cg' mode will be used
instead.""")
X = _solve_cg(lap_sparse, B, tol=tol)
X = _solve_cg(lap_sparse, B, tol=tol,
return_full_prob=return_full_prob)
else:
X = _solve_cg_mg(lap_sparse, B, tol=tol,
return_full_prob=return_full_prob)