Victoria Oberascher commited on
Commit
d324c80
·
1 Parent(s): 7da121b

fix bug concerning input field

Browse files
Files changed (1) hide show
  1. horizon-metrics.py +4 -0
horizon-metrics.py CHANGED
@@ -176,6 +176,10 @@ class HorizonMetrics(evaluate.Metric):
176
  float: The computed horizon error.
177
  """
178
 
 
 
 
 
179
  # calculate erros and store values in slope_error_list and midpoint_error_list
180
  for annotated_horizon, proposed_horizon in zip(self.ground_truth_det,
181
  self.predictions):
 
176
  float: The computed horizon error.
177
  """
178
 
179
+ if (self.ground_truth_det is None or self.predictions is None):
180
+ self.ground_truth_det = references
181
+ self.predictions = predictions
182
+
183
  # calculate erros and store values in slope_error_list and midpoint_error_list
184
  for annotated_horizon, proposed_horizon in zip(self.ground_truth_det,
185
  self.predictions):