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

add step condition to verofy

Browse Source
This commit is contained in:
QuanyiLi
2023-05-07 16:01:20 +01:00
parent ac3ceab1ec
commit 725cc7d2b9
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scenarionet/verifier/utils.py
View File

@@ -9,7 +9,7 @@ from metadrive.policy.replay_policy import ReplayEgoCarPolicy
from metadrive.scenario.utils import get_number_of_scenarios
def verify_loading_into_metadrive(dataset_path, result_save_dir=None):
def verify_loading_into_metadrive(dataset_path, result_save_dir=None, steps_to_run=0):
scenario_num = get_number_of_scenarios(dataset_path)
if result_save_dir is not None:
assert os.path.exists(result_save_dir) and os.path.isdir(
@@ -28,6 +28,8 @@ def verify_loading_into_metadrive(dataset_path, result_save_dir=None):
try:
for i in tqdm.tqdm(range(scenario_num)):
env.reset(force_seed=i)
for i in range(steps_to_run):
env.step([0, 0])
except Exception as e:
file_name = env.engine.data_manager.summary_lookup[i]
file_path = os.path.join(dataset_path, env.engine.data_manager.mapping[file_name], file_name)
@@ -38,6 +40,7 @@ def verify_loading_into_metadrive(dataset_path, result_save_dir=None):
finally:
env.close()
if result_save_dir is not None:
with open(os.path.join(result_save_dir, "error_scenarios_{}.json".format(os.path.basename(dataset_path))), "w+") as f:
with open(os.path.join(result_save_dir, "error_scenarios_{}.json".format(os.path.basename(dataset_path))),
"w+") as f:
json.dump(error_files, f)
return success, error_files