Better handling of labeled pixels in random walker segmentation when we

return the whole probability.
This commit is contained in:
Emmanuelle Gouillart
2012-07-24 23:01:15 +02:00
parent 4ab7d0a4fa
commit aa92d5f0bd
2 changed files with 6 additions and 1 deletions
@@ -58,7 +58,7 @@ 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
return data, labels_bf, full_prob_bf
def test_2d_cg():