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
result[SD.DYNAMIC_MAP_STATES] = {}
track_to_predict = result[SD.TRACKS][instance_token]
result[SD.METADATA]["tracks_to_predict"] = {
instance_token: {
"track_index": list(result[SD.TRACKS].keys()).index(instance_token),
"track_id": instance_token,
"difficulty": 0,
"object_type": track_to_predict['type']
}
}
# track_to_predict = result[SD.TRACKS][instance_token]
# result[SD.METADATA]["tracks_to_predict"] = {
# instance_token: {
# "track_index": list(result[SD.TRACKS].keys()).index(instance_token),
# "track_id": instance_token,
# "difficulty": 0,
# "object_type": track_to_predict['type']
# }
# }
# map
result[SD.MAP_FEATURES] = get_map_features(scene_info, nuscenes, map_center, map_radius, only_lane=only_lane)
# add back map center