bug

Author: anna-grim

src/segmentation_skeleton_metrics/skeleton_metric.py

@@ -115,12 +115,12 @@ def __init__(
         # Load ground truth
         print("\n(1) Load Ground Truth")
         assert type(valid_labels) is set if valid_labels else True
+        self.label_mask = pred_labels
         self.valid_labels = valid_labels
         self.init_label_map(connections_path)
         self.init_graphs(gt_pointer)
 
         print("\n(2) Load Prediction")
-        self.label_mask = pred_labels
         if fragments_pointer:
             self.load_fragments(fragments_pointer)
         else:

src/segmentation_skeleton_metrics/utils/swc_util.py

@@ -172,7 +172,6 @@ def load_from_local_paths(self, paths):
             threads = list()
             pbar = tqdm(total=len(paths), desc="Read SWCs")
             for path in paths:
-                #self.load_from_local_path(path)
                 threads.append(
                     executor.submit(self.load_from_local_path, path)
                 )