huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix av2 sdc_track_indx (#72) (#76)

Browse Source
This commit is contained in:
Alan
2024-03-29 14:05:02 +01:00
committed by GitHub
parent ed2a9d3388
commit c6775d7197
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scenarionet/converter/argoverse2/utils.py
View File

@@ -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]