dmytromishkin commited on
Commit
30cf7df
1 Parent(s): 8b90397

Update hoho/wed.py

Browse files
Files changed (1) hide show
  1. hoho/wed.py +1 -1
hoho/wed.py CHANGED
@@ -2,7 +2,7 @@ from scipy.spatial.distance import cdist
2
  from scipy.optimize import linear_sum_assignment
3
  import numpy as np
4
 
5
- def compute_WED(pd_vertices, pd_edges, gt_vertices, gt_edges, cv, ce, normalized=True, squared=False):
6
  pd_vertices = np.array(pd_vertices)
7
  gt_vertices = np.array(gt_vertices)
8
  pd_edges = np.array(pd_edges)
 
2
  from scipy.optimize import linear_sum_assignment
3
  import numpy as np
4
 
5
+ def compute_WED(pd_vertices, pd_edges, gt_vertices, gt_edges, cv=1.0, ce=1.0, normalized=True, squared=False):
6
  pd_vertices = np.array(pd_vertices)
7
  gt_vertices = np.array(gt_vertices)
8
  pd_edges = np.array(pd_edges)