Merge branch 'main' into feat-localize-merge

Author: anna-grim

src/segmentation_skeleton_metrics/__init__.py

@@ -2,4 +2,4 @@
 Package to evaluate a predicted segmentation.
 """
 
-__version__ = "4.14.3"
+__version__ = "4.15.1"

src/segmentation_skeleton_metrics/skeleton_metric.py

@@ -138,6 +138,11 @@ def __init__(
         # Initialize writers
         self.init_zip_writers()
 
+        # Initialize fragment projections directory
+        if self.save_projections:
+            self.projections_dir = os.path.join(output_dir, "projections")
+            util.mkdir(self.projections_dir)
+
     # --- Load Data ---
     def load_groundtruth(self, swc_pointer):
         """
@@ -555,6 +560,9 @@ def count_merges(self, key, kdtree):
                 for label in self.label_handler.get_class(label):
                     if label in self.fragment_ids:
                         self.is_fragment_merge(key, label, kdtree)
+                        if self.save_projections:
+                            fragment_graph = self.find_graph_from_label(label)[0]
+                            fragment_graph.to_zipped_swc(zip_writer)
 
     def is_fragment_merge(self, key, label, kdtree):
         """

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