Refactor optimize merge (#118)

* refactor: optimized merge detection

* bug: removed subroutine

* removed test tqdm

* bug: close zip writer

* refactor: metrics in pandas df

* refactor: detect all sites

---------

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

Author: anna-grim

src/segmentation_skeleton_metrics/skeleton_metric.py

@@ -565,11 +565,9 @@ def is_fragment_merge(self, key, label, kdtree):
                     voxel = fragment_graph.voxels[leaf]
                     gt_voxel = util.kdtree_query(kdtree, voxel)
                     if self.physical_dist(gt_voxel, voxel) > 50:
-                        has_merge, visited = self.find_merge_site(
+                        visited = self.find_merge_site(
                             key, kdtree, fragment_graph, leaf, visited
                         )
-                        if has_merge:
-                            break
 
                 # Save fragment (if applicable)
                 if self.save_fragments:
@@ -609,8 +607,8 @@ def find_merge_site(self, key, kdtree, fragment_graph, source, visited):
                         gutil.write_graph(
                              self.gt_graphs[key], self.merge_writer
                          )
-                    return True, visited
-        return False, visited
+                    return visited
+        return visited
 
     def process_merge_sites(self):
         # Remove duplicates