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]