fix typo in results_to_csv.py

bmmtstb · Aug 18, 2024 · f89bb5a · f89bb5a
1 parent a14030b
commit f89bb5a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/helpers/results_to_csv.py b/scripts/helpers/results_to_csv.py
@@ -91,8 +91,8 @@ def replace_dots_with_commas(fp: FilePath) -> None:
             for k, new_k in {"mota": "MOTA", "motp": "MOTP", "pre": "MOT precision", "rec": "MOT recall"}.items():
                 comb = f"{ds_name}_{conf_key}_{new_k}"
                 custom_total = sum(
-                    float(val) for val in ap_data[k] if val not in ["neck", "head_top", "head_bottom"]
-                ) / (len(ap_data[k]) - 2)
+                    float(val) for val in mot_data[k] if val not in ["neck", "head_top", "head_bottom"]
+                ) / (len(mot_data[k]) - 2)
                 csv_writer.writerow([comb, ds_name, conf_key, new_k, custom_total] + mot_data[k])
 
             AP_file = MOT_file.replace("total_MOT_metrics", "total_AP_metrics")