Cloud merge detection (#52)

* refactor: valid labels

* refactor: valid_id generation

---------

Co-authored-by: anna-grim <[email protected]>

Author: anna-grim

src/segmentation_skeleton_metrics/skeleton_metric.py

@@ -17,8 +17,7 @@
 from scipy.spatial import KDTree
 
 from segmentation_skeleton_metrics import graph_utils as gutils
-from segmentation_skeleton_metrics import split_detection, utils
-from segmentation_skeleton_metrics import swc_utils
+from segmentation_skeleton_metrics import split_detection, swc_utils, utils
 from segmentation_skeleton_metrics.swc_utils import (
     get_xyz_coords,
     save,
@@ -130,7 +129,7 @@ def init_valid_labels(self, valid_size_threshold):
         for path in self.pred_swc_paths:
             contents = swc_utils.read(path)
             if len(contents) > valid_size_threshold:
-                self.valid_labels.add(utils.get_swc_id(path))
+                self.valid_labels.add(int(utils.get_swc_id(path)))
 
     def init_graphs(self, paths, anisotropy):
         """

src/segmentation_skeleton_metrics/swc_utils.py

@@ -177,7 +177,7 @@ def to_graph(path, anisotropy=[1.0, 1.0, 1.0]):
 
     """
     graph = nx.Graph(swc_id=utils.get_swc_id(path))
-    offset = [0, 0, 0]    
+    offset = [0, 0, 0]
     for line in read(path):
         if line.startswith("# OFFSET"):
             parts = line.split()