Feat localize merge (#100)

* feat: localize merges

* feat: save projections

---------

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

Author: anna-grim

src/segmentation_skeleton_metrics/skeleton_metric.py

@@ -543,6 +543,7 @@ def count_merges(self, key, kdtree):
         if self.save_projections:
             zip_path = os.path.join(self.projections_dir, key + ".zip")
             zip_writer = ZipFile(zip_path, "w")
+            #self.graphs[key].to_zipped_swc(zip_writer)
 
         # Iterate over fragments that intersect with GT skeleton
         for label in self.get_node_labels(key):

src/segmentation_skeleton_metrics/split_detection.py

@@ -46,15 +46,15 @@ def run(process_id, graph):
         label_i = int(graph.labels[i])
         label_j = int(graph.labels[j])
         if is_split(label_i, label_j):
-            graph.remove_edge(i, j)
+            graph.remove_edge(i, j) temp
             split_cnt += 1
         elif label_j == 0:
             check_misalignment(graph, visited_edges, i, j)
         visited_edges.add(frozenset({i, j}))
 
     # Finish
     split_percent = split_cnt / graph.graph["n_edges"]
-    graph.remove_nodes_with_label(0)
+    graph.remove_nodes_with_label(0) temp
     return process_id, graph, split_percent