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

get all scenarios

Browse Source
This commit is contained in:
QuanyiLi
2023-05-06 22:06:15 +01:00
parent 85156570b4
commit b101f6fdea
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scenarionet/converter/nuplan/utils.py
View File

@@ -40,7 +40,7 @@ except ImportError as e:
EGO = "ego"
def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None], builder="nuplan_mini"):
def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None] = None, builder="nuplan_mini"):
"""
Getting scenarios. You could use your parameters to get a bunch of scenarios
:param data_root: path contains .db files, like /nuplan-v1.1/splits/mini
@@ -54,7 +54,7 @@ def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None], builder="
logs = logs or [file for file in os.listdir(data_root)]
log_string = ""
for log in logs:
log_string += log
log_string += log[:-3] # remove .db
log_string += ","
log_string = log_string[:-1]

3
scenarionet/examples/convert_nuplan.py
View File

@@ -16,7 +16,8 @@ if __name__ == "__main__":
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"])
# scenarios = get_nuplan_scenarios(data_root, map_root, logs=["2021.07.16.20.45.29_veh-35_01095_01486"])
scenarios = get_nuplan_scenarios(data_root, map_root)
write_to_directory(convert_func=convert_nuplan_scenario,
scenarios=scenarios,