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

Cast label_field to integer dtype for float intput

Browse Source
This commit is contained in:
Johannes Schönberger
2013-11-09 21:56:18 +01:00
parent c9a0e530f1
commit 615866ae16
2 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/segmentation/_join.py
+2
View File
@@ -114,6 +114,8 @@ def relabel_sequential(label_field, offset=1):
>>> relab
array([5, 5, 6, 6, 7, 9, 8])
"""
if not np.issubdtype(label_field.dtype, np.int):
label_field = label_field.astype(np.int)
labels = np.unique(label_field)
labels0 = labels[labels != 0]
m = labels.max()
skimage/segmentation/tests/test_join.py
+12
View File
@@ -61,5 +61,17 @@ def test_relabel_sequential_offset5_with0():
assert_array_equal(inv, inv_ref)
def test_relabel_sequential_dtype():
ar = np.array([1, 1, 5, 5, 8, 99, 42, 0], dtype=float)
ar_relab, fw, inv = relabel_sequential(ar, offset=5)
ar_relab_ref = np.array([5, 5, 6, 6, 7, 9, 8, 0])
assert_array_equal(ar_relab, ar_relab_ref)
fw_ref = np.zeros(100, int)
fw_ref[1] = 5; fw_ref[5] = 6; fw_ref[8] = 7; fw_ref[42] = 8; fw_ref[99] = 9
assert_array_equal(fw, fw_ref)
inv_ref = np.array([0, 0, 0, 0, 0, 1, 5, 8, 42, 99])
assert_array_equal(inv, inv_ref)
if __name__ == "__main__":
np.testing.run_module_suite()