From c6775d719747df42fc1b5fc2aaf0b4da87b360d5 Mon Sep 17 00:00:00 2001 From: Alan <36124025+Alan-LanFeng@users.noreply.github.com> Date: Fri, 29 Mar 2024 14:05:02 +0100 Subject: [PATCH] fix av2 sdc_track_indx (#72) (#76) --- scenarionet/converter/argoverse2/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scenarionet/converter/argoverse2/utils.py b/scenarionet/converter/argoverse2/utils.py index 5131823..b4f7596 100644 --- a/scenarionet/converter/argoverse2/utils.py +++ b/scenarionet/converter/argoverse2/utils.py @@ -224,12 +224,13 @@ def convert_av2_scenario(scenario, version): # === Waymo specific data. Storing them here === md_scenario[SD.METADATA]["current_time_index"] = 49 - md_scenario[SD.METADATA]["sdc_track_index"] = scenario.focal_track_id # obj id obj_keys = list(tracks.keys()) md_scenario[SD.METADATA]["objects_of_interest"] = [obj_keys[idx] for idx, cat in enumerate(category) if cat == 2] + md_scenario[SD.METADATA]["sdc_track_index"] = obj_keys.index('AV') + track_index = [obj_keys.index(scenario.focal_track_id)] track_id = [scenario.focal_track_id] track_difficulty = [0]