From 8fc8170f4c8bbc88693486292700a2a02965f8c7 Mon Sep 17 00:00:00 2001 From: Redrew Date: Tue, 31 Oct 2023 15:46:35 -0400 Subject: [PATCH] use name instead of label id in tracking evaluation --- src/av2/evaluation/tracking/eval.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/av2/evaluation/tracking/eval.py b/src/av2/evaluation/tracking/eval.py index 56d56b9f..98124789 100644 --- a/src/av2/evaluation/tracking/eval.py +++ b/src/av2/evaluation/tracking/eval.py @@ -83,7 +83,10 @@ def _load_raw_file( raw_data = { f"{source}_ids": [frame["track_id"] for frame in tracks], - f"{source}_classes": [frame["label"] for frame in tracks], + f"{source}_classes": [ + np.array([self.full_class_list.index(n) for n in frame["name"]]) + for frame in tracks + ], f"{source}_dets": [ np.concatenate((frame["translation_m"], frame["size"]), axis=-1) for frame in tracks