Victoria Oberascher commited on
Commit
c575cfd
1 Parent(s): dbcf938

update init parameters

Browse files
Files changed (1) hide show
  1. horizonmetrics.py +6 -5
horizonmetrics.py CHANGED
@@ -64,14 +64,15 @@ class horizonmetrics(evaluate.Metric):
64
  """TODO: Short description of my evaluation module."""
65
 
66
  def __init__(self,
67
- slope_threshold=0.1,
68
- midpoint_threshold=0.1,
69
  vertical_fov_degrees=25.6,
70
  **kwargs):
71
  super().__init__(**kwargs)
72
- self.slope_threshold = slope_threshold
73
- self.midpoint_threshold = midpoint_threshold
74
- self.vertical_fov_degrees = vertical_fov_degrees
 
75
  self.predictions = None
76
  self.ground_truth_det = None
77
  self.slope_error_list = None
 
64
  """TODO: Short description of my evaluation module."""
65
 
66
  def __init__(self,
67
+ roll_threshold=0.5,
68
+ pitch_threshold=0.1,
69
  vertical_fov_degrees=25.6,
70
  **kwargs):
71
  super().__init__(**kwargs)
72
+
73
+ self.slope_threshold = roll_to_slope(roll_threshold)
74
+ self.midpoint_threshold = pitch_to_midpoint(pitch_threshold,
75
+ vertical_fov_degrees)
76
  self.predictions = None
77
  self.ground_truth_det = None
78
  self.slope_error_list = None