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

fix bug

Browse Source
This commit is contained in:
QuanyiLi
2023-05-08 17:07:24 +01:00
parent 63d6cb2c1a
commit b5418ba3f6
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scenarionet/scripts/verify_dataset.py
View File

@@ -4,12 +4,14 @@ MetaDrive.
"""
import argparse
from scenarionet.verifier.utils import verify_loading_into_metadrive
from scenarionet.verifier.utils import verify_loading_into_metadrive, set_random_drop
if __name__ == '__main__':
parser = argparse.ArgumentParser()
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("--num_workers", type=int, default=8, help="number of workers to use")
parser.add_argument("--random_drop", action="store_true", help="Randomly make some scenarios fail. for test only!")
args = parser.parse_args()
verify_loading_into_metadrive(args.dataset_path, args.result_save_dir, num_workers=args.num_workers)
set_random_drop(args.random_drop)
verify_loading_into_metadrive(args.dataset_path, args.result_save_dir, num_workers=args.num_workers)

6
scenarionet/tests/local_test/combine_verify_generate.sh
View File

@@ -1,6 +1,6 @@
#!/usr/bin/env bash
python ../../scripts/combine_dataset.py --to ../tmp/test_combine_dataset --from_datasets ../../../dataset/waymo ../../../dataset/pg ../../../dataset/nuscenes ../../../dataset/nuplan --overwrite
python ../../scripts/verify_dataset.py --dataset_path ../tmp/test_combine_dataset --result_save_dir ../tmp/test_combine_dataset
python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_pass
python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_fail --broken
python ../../scripts/verify_dataset.py --dataset_path ../tmp/test_combine_dataset --result_save_dir ../tmp/test_combine_dataset --random_drop
python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/error_scenarios_for_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_pass
python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_scenarios_for_combine_dataset.json --overwrite --dataset_path ../tmp/verify_fail --broken