add unique ids

user-friendly-metrics.py

@@ -17,8 +17,7 @@ import datasets
 import motmetrics as mm
 from motmetrics.metrics import (events_to_df_map, 
                                 obj_frequencies,
-                                track_ratios,
-                                num_gt_ids)
+                                track_ratios)
 import numpy as np
 
 _CITATION = """\
@@ -97,6 +96,10 @@ def recognition(track_ratios, th = 0.5):
     """Number of objects tracked for at least 20 percent of lifespan."""
     return track_ratios[track_ratios >= th].count()
 
+def num_gt_ids(df):
+    """Number of unique gt ids."""
+    return df.full["OId"].dropna().unique().shape[0]
+
 def calculate(predictions, 
               references, 
               max_iou: float = 0.5,
@@ -144,7 +147,7 @@ def calculate(predictions,
         summary[key] = summary[key][0]
 
     for th in recognition_thresholds:
-        summary[f'recognition_{th}'] = recognition(tr_ratios, th)
+        summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
     
     return summary