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

Merge pull request #1449 from stefanv/clear_border_non_binary

Browse Source 
Allow clear_border to operate on labeled images
This commit is contained in:
Juan Nunez-Iglesias
2015-12-09 11:18:40 +11:00
parent c133d5fd59 6fc4e11b9a
commit 1a05b130e8
2 changed files with 61 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/segmentation/tests/test_clear_border.py
+33 -2
View File
@@ -1,5 +1,5 @@
import numpy as np
from numpy.testing import assert_array_equal
from numpy.testing import assert_array_equal, assert_
from skimage.segmentation import clear_border
@@ -24,9 +24,40 @@ def test_clear_border():
assert_array_equal(result, np.zeros(result.shape))
# test background value
result = clear_border(image.copy(), 1, 2)
result = clear_border(image.copy(), buffer_size=1, bgval=2)
assert_array_equal(result, 2 * np.ones_like(image))
def test_clear_border_non_binary():
image = np.array([[1, 2, 3, 1, 2],
[3, 3, 5, 4, 2],
[3, 4, 5, 4, 2],
[3, 3, 2, 1, 2]])
result = clear_border(image)
expected = np.array([[0, 0, 0, 0, 0],
[0, 0, 5, 4, 0],
[0, 4, 5, 4, 0],
[0, 0, 0, 0, 0]])
assert_array_equal(result, expected)
assert_(not np.all(image == result))
def test_clear_border_non_binary_inplace():
image = np.array([[1, 2, 3, 1, 2],
[3, 3, 5, 4, 2],
[3, 4, 5, 4, 2],
[3, 3, 2, 1, 2]])
result = clear_border(image, in_place=True)
expected = np.array([[0, 0, 0, 0, 0],
[0, 0, 5, 4, 0],
[0, 4, 5, 4, 0],
[0, 0, 0, 0, 0]])
assert_array_equal(result, expected)
assert_array_equal(image, result)
if __name__ == "__main__":
np.testing.run_module_suite()