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

[BUG] Corrected a bug in the random walker that appeared

Browse Source 
* when returning the full probability instead of the segmentation
    * for three or more labels (two was OK)
This commit is contained in:
Emmanuelle Gouillart
2012-09-18 19:56:52 +02:00
parent 58a57238cc
commit 6635cf16db
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/segmentation/tests/test_random_walker.py
+7 -2
View File
@@ -58,8 +58,13 @@ def test_2d_bf():
return_full_prob=True)
assert (full_prob_bf[1, 25:45, 40:60] >=
full_prob_bf[0, 25:45, 40:60]).all()
return data, labels_bf, full_prob_bf
# Now test with more than two labels
labels[55, 80] = 3
full_prob_bf = random_walker(data, labels, beta=90, mode='bf',
return_full_prob=True)
assert (full_prob_bf[1, 25:45, 40:60] >=
full_prob_bf[0, 25:45, 40:60]).all()
assert len(full_prob_bf) == 3
def test_2d_cg():
lx = 70