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

update get nuplan parameters

Browse Source
This commit is contained in:
QuanyiLi
2023-05-06 21:58:48 +01:00
parent 3963c88e41
commit 85156570b4
2 changed files with 43 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scenarionet/examples/convert_nuplan.py
View File

@@ -5,7 +5,7 @@ MetaDrive.
import os
from scenarionet import SCENARIONET_DATASET_PATH
from scenarionet.converter.nuplan.utils import get_nuplan_scenarios, convert_nuplan_scenario, example_dataset_params
from scenarionet.converter.nuplan.utils import get_nuplan_scenarios, convert_nuplan_scenario
from scenarionet.converter.utils import write_to_directory
if __name__ == "__main__":
@@ -14,7 +14,9 @@ if __name__ == "__main__":
output_path = os.path.join(SCENARIONET_DATASET_PATH, dataset_name)
version = 'v1.1'
scenarios = get_nuplan_scenarios(example_dataset_params)
data_root = os.path.join(os.getenv("NUPLAN_DATA_ROOT"), "nuplan-v1.1/splits/mini")
map_root = os.getenv("NUPLAN_MAPS_ROOT")
scenarios = get_nuplan_scenarios(data_root, map_root, logs=["2021.07.16.20.45.29_veh-35_01095_01486"])
write_to_directory(convert_func=convert_nuplan_scenario,
scenarios=scenarios,