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

fix a bug in nusc converter (#60)

Browse Source
This commit is contained in:
Yunsong Zhou
2024-02-19 16:30:56 -08:00
committed by GitHub
parent 696b3ca109
commit d0d8224957
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scenarionet/converter/nuscenes/utils.py
View File

@@ -474,15 +474,15 @@ def convert_nuscenes_scenario(
# No traffic light in nuscenes at this stage # No traffic light in nuscenes at this stage
result[SD.DYNAMIC_MAP_STATES] = {} result[SD.DYNAMIC_MAP_STATES] = {}
track_to_predict = result[SD.TRACKS][instance_token] # track_to_predict = result[SD.TRACKS][instance_token]
result[SD.METADATA]["tracks_to_predict"] = { # result[SD.METADATA]["tracks_to_predict"] = {
instance_token: { # instance_token: {
"track_index": list(result[SD.TRACKS].keys()).index(instance_token), # "track_index": list(result[SD.TRACKS].keys()).index(instance_token),
"track_id": instance_token, # "track_id": instance_token,
"difficulty": 0, # "difficulty": 0,
"object_type": track_to_predict['type'] # "object_type": track_to_predict['type']
} # }
} # }
# map # map
result[SD.MAP_FEATURES] = get_map_features(scene_info, nuscenes, map_center, map_radius, only_lane=only_lane) result[SD.MAP_FEATURES] = get_map_features(scene_info, nuscenes, map_center, map_radius, only_lane=only_lane)
# add back map center # add back map center