From 473e12643c3637dadfda6c6a0e9fb5841d6af87e Mon Sep 17 00:00:00 2001 From: Lauenburg Date: Tue, 8 Feb 2022 19:37:03 -0500 Subject: [PATCH 1/2] added missing bracket --- connectomics/data/utils/data_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/connectomics/data/utils/data_segmentation.py b/connectomics/data/utils/data_segmentation.py index 817f1e8f..4ad5b53a 100755 --- a/connectomics/data/utils/data_segmentation.py +++ b/connectomics/data/utils/data_segmentation.py @@ -302,7 +302,7 @@ def seg_to_targets(label_orig: np.ndarray, if len(topt) == 1: topt = topt + '-2d-0' # 2d DT without padding (default) _, mode, padding = topt.split('-') - out[tid] = edt_instance(label.copy(), mode, padding=bool(int(padding)) + out[tid] = edt_instance(label.copy(), mode, padding=bool(int(padding))) elif topt[0] == '6': # distance transform (semantic) if len(topt) == 1: topt = topt + '-2d-8-50' From 12e4caae4b1a599de92ea1cf4c5a4506ba19131d Mon Sep 17 00:00:00 2001 From: Lauenburg Date: Wed, 9 Feb 2022 10:57:17 -0500 Subject: [PATCH 2/2] Added type casting: Indexing requires int or boolean, mask values where of format float without decimal, like 124432. --- connectomics/data/utils/data_segmentation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/connectomics/data/utils/data_segmentation.py b/connectomics/data/utils/data_segmentation.py index 4ad5b53a..2b907429 100755 --- a/connectomics/data/utils/data_segmentation.py +++ b/connectomics/data/utils/data_segmentation.py @@ -27,7 +27,7 @@ def getSegType(mid): def relabel(seg, do_type=False): # get the unique labels - uid = np.unique(seg) + uid = np.unique(seg).astype(np.int64) # ignore all-background samples if len(uid) == 1 and uid[0] == 0: return seg @@ -41,7 +41,7 @@ def relabel(seg, do_type=False): m_type = getSegType(mid) mapping = np.zeros(mid, dtype=m_type) mapping[uid] = np.arange(1, len(uid) + 1, dtype=m_type) - return mapping[seg] + return mapping[seg.astype(np.int64)] def remove_small(seg, thres=100):