use name instead of label id in tracking evaluation (#244)

Co-authored-by: Redrew <[email protected]>
argoverse · Dec 17, 2023 · 8f90a33 · 8f90a33
1 parent eb62ac1
commit 8f90a33
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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