Spaces:
Sleeping
Sleeping
add unique ids
Browse files- user-friendly-metrics.py +2 -0
user-friendly-metrics.py
CHANGED
@@ -150,6 +150,8 @@ def calculate(predictions,
|
|
150 |
for key in summary:
|
151 |
summary[key] = summary[key][0]
|
152 |
|
|
|
|
|
153 |
for th in recognition_thresholds:
|
154 |
summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
|
155 |
|
|
|
150 |
for key in summary:
|
151 |
summary[key] = summary[key][0]
|
152 |
|
153 |
+
summary["num_gt_ids"] = unique_gt_ids
|
154 |
+
|
155 |
for th in recognition_thresholds:
|
156 |
summary[f'recognition_{th}'] = recognition(tr_ratios, th)/unique_gt_ids
|
157 |
|