add rec and precision

horizon-metrics.py

@@ -180,11 +180,20 @@ class HorizonMetrics(evaluate.Metric):
         Returns:
             float: The computed horizon error.
         """
-
+        tp = 0
+        fp = 0
+        fn = 0
         # calculate erros and store values in slope_error_list and midpoint_error_list
         for annotated_horizon, proposed_horizon in zip(self.ground_truth_det,
                                                        self.predictions):
 
+            if annotated_horizon is None and proposed_horizon != None:
+                fp += 1
+            elif annotated_horizon != None and proposed_horizon != None:
+                tp += 1
+            elif annotated_horizon != None and proposed_horizon is None:
+                fn += 1
+                
             if annotated_horizon is None or proposed_horizon is None:
                 continue
             slope_error, midpoint_error = calculate_horizon_error(
@@ -207,6 +216,8 @@ class HorizonMetrics(evaluate.Metric):
         detection_rate = detected_horizon_count / detected_gt_count
         result['detection_rate'] = detection_rate
         result['predicted_samples'] = detected_horizon_count
+        result['recall'] = tp/(tp+fn+1e-5)
+        result['precision'] = tp/(fp+fn+1e-5)
         result['samples'] = detected_gt_count
 
         return result