Skip to content

Commit 4ab7911

Browse files
anna-grimanna-grim
andauthored
Nonzero misalignments (#42)
* feat: detect nonzero misalignments * bug: fixed alignment detection --------- Co-authored-by: anna-grim <[email protected]>
1 parent c3a4efd commit 4ab7911

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/segmentation_skeleton_metrics/skeleton_metric.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,6 @@ def detect_splits(self):
345345
label_i = pred_graph.nodes[i]["pred_id"]
346346
label_j = pred_graph.nodes[j]["pred_id"]
347347
if is_split(label_i, label_j):
348-
# pred_graph = gutils.remove_edge(pred_graph, i, j)
349348
dfs_edges, pred_graph = self.is_nonzero_misalignment(
350349
target_graph, pred_graph, dfs_edges, i, j
351350
)
@@ -396,7 +395,7 @@ def is_zero_misalignment(
396395
"""
397396
# Search
398397
black_hole = False
399-
collision_labels = set([pred_graph.nodes[nb]["pred_id"]])
398+
collision_labels = set()
400399
queue = [root]
401400
visited = set()
402401
while len(queue) > 0:
@@ -806,7 +805,7 @@ def is_split(a, b):
806805
Indication of whether there is a split.
807806
808807
"""
809-
return (a != 0 and b != 0) and (a != b)
808+
return (a > 0 and b > 0) and (a != b)
810809

811810

812811
def remove_edge(dfs_edges, edge):

0 commit comments

Comments
 (0)