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

set waymo origin path

Browse Source
This commit is contained in:
QuanyiLi
2023-05-08 15:17:13 +01:00
parent 2f8f9159c2
commit bbef2db107
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -16,3 +16,4 @@ dataset/*
**/tmp/** **/tmp/**
**/failed_scenarios/ **/failed_scenarios/
**/passed_senarios/ **/passed_senarios/
**/waymo_origin

2
scenarionet/scripts/convert_waymo.py
View File

@@ -23,7 +23,7 @@ if __name__ == '__main__':
output_path = args.dataset_path output_path = args.dataset_path
version = args.version version = args.version
waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "waymo_origin") waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "../waymo_origin")
scenarios = get_waymo_scenarios(waymo_data_directory) scenarios = get_waymo_scenarios(waymo_data_directory)
write_to_directory( write_to_directory(

2
scenarionet/tests/local_test/_test_convert_waymo.py
View File

@@ -22,7 +22,7 @@ if __name__ == '__main__':
output_path = args.dataset_path output_path = args.dataset_path
version = args.version version = args.version
waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "waymo_origin") waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "../waymo_origin")
scenarios = get_waymo_scenarios(waymo_data_directory, num_workers=3) scenarios = get_waymo_scenarios(waymo_data_directory, num_workers=3)
write_to_directory( write_to_directory(