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

add num works to script arguments

Browse Source
This commit is contained in:
QuanyiLi
2023-05-08 16:26:26 +01:00
parent 53abbc0152
commit 3c6d562718
5 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scenarionet/scripts/convert_nuplan.py
View File

@@ -17,6 +17,7 @@ if __name__ == '__main__':
help="The path of the dataset") help="The path of the dataset")
parser.add_argument("--version", "-v", default='v1.1', help="version") parser.add_argument("--version", "-v", default='v1.1', help="version")
parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it") parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it")
parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use")
args = parser.parse_args() args = parser.parse_args()
force_overwrite = args.overwrite force_overwrite = args.overwrite
@@ -36,4 +37,5 @@ if __name__ == '__main__':
dataset_version=version, dataset_version=version,
dataset_name=dataset_name, dataset_name=dataset_name,
force_overwrite=force_overwrite, force_overwrite=force_overwrite,
num_workers=args.num_workers
) )

4
scenarionet/scripts/convert_nuscenes.py
View File

@@ -17,6 +17,7 @@ if __name__ == '__main__':
help="The path of the dataset") help="The path of the dataset")
parser.add_argument("--version", "-v", default='v1.0-mini', help="version") parser.add_argument("--version", "-v", default='v1.0-mini', help="version")
parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it") parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it")
parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use")
args = parser.parse_args() args = parser.parse_args()
force_overwrite = args.overwrite force_overwrite = args.overwrite
@@ -34,5 +35,6 @@ if __name__ == '__main__':
dataset_version=version, dataset_version=version,
dataset_name=dataset_name, dataset_name=dataset_name,
force_overwrite=force_overwrite, force_overwrite=force_overwrite,
nuscenes=nusc nuscenes=nusc,
num_workers=args.num_workers
) )

4
scenarionet/scripts/convert_pg.py
View File

@@ -17,6 +17,7 @@ if __name__ == '__main__':
help="The path of the dataset") help="The path of the dataset")
parser.add_argument("--version", "-v", default=metadrive.constants.DATA_VERSION, help="version") parser.add_argument("--version", "-v", default=metadrive.constants.DATA_VERSION, help="version")
parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it") parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it")
parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use")
args = parser.parse_args() args = parser.parse_args()
force_overwrite = args.overwrite force_overwrite = args.overwrite
@@ -33,5 +34,6 @@ if __name__ == '__main__':
dataset_version=version, dataset_version=version,
dataset_name=dataset_name, dataset_name=dataset_name,
force_overwrite=force_overwrite, force_overwrite=force_overwrite,
env=env env=env,
num_workers=args.num_workers
) )

4
scenarionet/scripts/convert_waymo.py
View File

@@ -16,6 +16,7 @@ if __name__ == '__main__':
help="The path of the dataset") help="The path of the dataset")
parser.add_argument("--version", "-v", default='v1.2', help="version") parser.add_argument("--version", "-v", default='v1.2', help="version")
parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it") parser.add_argument("--overwrite", action="store_true", help="If the dataset_path exists, overwrite it")
parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use")
args = parser.parse_args() args = parser.parse_args()
force_overwrite = args.overwrite force_overwrite = args.overwrite
@@ -32,5 +33,6 @@ if __name__ == '__main__':
output_path=output_path, output_path=output_path,
dataset_version=version, dataset_version=version,
dataset_name=dataset_name, dataset_name=dataset_name,
force_overwrite=force_overwrite force_overwrite=force_overwrite,
num_workers=args.num_workers
) )

3
scenarionet/scripts/verify_dataset.py
View File

@@ -10,5 +10,6 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--dataset_path", required=True, help="Dataset path, a directory") parser.add_argument("--dataset_path", required=True, help="Dataset path, a directory")
parser.add_argument("--result_save_dir", required=True, help="Dataset path, a directory") parser.add_argument("--result_save_dir", required=True, help="Dataset path, a directory")
parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use")
args = parser.parse_args() args = parser.parse_args()
verify_loading_into_metadrive(args.dataset_path, args.result_save_dir) verify_loading_into_metadrive(args.dataset_path, args.result_save_dir, num_workers=args.num_workers)