usm3d
/

tools

Model card Files Files and versions

jacklangerman commited on Jun 7, 2024

Commit

b513ce3

·

verified ·

1 Parent(s): 0614af0

fix typo

Files changed (1) hide show

hoho/wed.py +1 -1

hoho/wed.py CHANGED Viewed

@@ -95,7 +95,7 @@ def compute_WED(pd_vertices, pd_edges, gt_vertices, gt_edges, cv_ins=-1/2, cv_de
     edges_to_insert = gt_edges_set - pd_edges_set
     insertion_edge_costs = ce * sum(np.linalg.norm(gt_vertices[edge[0]] - gt_vertices[edge[1]]) for edge in edges_to_insert if len(edge) == 2)
-    # Step 5: Calculation of WED
     WED = translation_costs + deletion_costs + insertion_costs + deletion_edge_costs + insertion_edge_costs

     edges_to_insert = gt_edges_set - pd_edges_set
     insertion_edge_costs = ce * sum(np.linalg.norm(gt_vertices[edge[0]] - gt_vertices[edge[1]]) for edge in edges_to_insert if len(edge) == 2)
+    # Step 6: Calculation of WED
     WED = translation_costs + deletion_costs + insertion_costs + deletion_edge_costs + insertion_edge_costs