DATA-231: add predicted samples and samples to result

horizon-metrics.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Union
+
 import evaluate
 import datasets
 import numpy as np
@@ -110,12 +112,13 @@ class HorizonMetrics(evaluate.Metric):
     """
 
     def __init__(self,
-                 vertical_fov_degrees,
-                 height,
+                 vertical_fov_degrees: Union[float, int, None],
+                 height: float,
                  roll_threshold=0.5,
                  pitch_threshold=0.1,
                  **kwargs):
 
+        
         super().__init__(**kwargs)
 
         self.slope_threshold = roll_to_slope(roll_threshold)
@@ -203,5 +206,7 @@ class HorizonMetrics(evaluate.Metric):
 
         detection_rate = detected_horizon_count / detected_gt_count
         result['detection_rate'] = detection_rate
+        result['predicted_samples'] = detected_horizon_count,
+        result['samples'] = detected_gt_count
 
         return result