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

add dataset to gitignore

Browse Source
This commit is contained in:
QuanyiLi
2023-05-08 16:31:35 +01:00
parent bf53687cc3
commit 37a7d8608e
5 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -18,3 +18,4 @@ dataset/*
**/failed_scenarios/ **/failed_scenarios/
**/passed_scenarios/ **/passed_scenarios/
**/waymo_origin **/waymo_origin
/dataset/

4
scenarionet/tests/local_test/_test_combine_dataset_local.py
View File

@@ -1,6 +1,6 @@
import os import os
from scenarionet import SCENARIONET_DATASET_PATH from scenarionet import SCENARIONET_DATASET_PATH, SCENARIONET_PACKAGE_PATH
from scenarionet.builder.utils import combine_multiple_dataset from scenarionet.builder.utils import combine_multiple_dataset
from scenarionet.verifier.utils import verify_loading_into_metadrive from scenarionet.verifier.utils import verify_loading_into_metadrive
@@ -13,7 +13,7 @@ def _test_combine_dataset():
os.path.join(SCENARIONET_DATASET_PATH, "pg") os.path.join(SCENARIONET_DATASET_PATH, "pg")
] ]
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset") combine_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "tmp", "combine")
combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True) combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)
# os.makedirs("verify_results", exist_ok=True) # os.makedirs("verify_results", exist_ok=True)
# verify_loading_into_metadrive(combine_path, "verify_results") # verify_loading_into_metadrive(combine_path, "verify_results")

4
scenarionet/tests/local_test/_test_filter_local.py
View File

@@ -3,7 +3,7 @@ import os.path
from metadrive.type import MetaDriveType from metadrive.type import MetaDriveType
from scenarionet import SCENARIONET_DATASET_PATH from scenarionet import SCENARIONET_DATASET_PATH, SCENARIONET_PACKAGE_PATH
from scenarionet.builder.filters import ScenarioFilter from scenarionet.builder.filters import ScenarioFilter
from scenarionet.builder.utils import combine_multiple_dataset from scenarionet.builder.utils import combine_multiple_dataset
@@ -17,7 +17,7 @@ def test_filter_dataset():
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo")) dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg")) dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg"))
output_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset") output_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "tmp", "combine")
# ========================= test 1 ========================= # ========================= test 1 =========================
# nuscenes data has no light # nuscenes data has no light

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

@@ -24,7 +24,7 @@ def test_generate_from_error():
os.path.join(SCENARIONET_DATASET_PATH, "pg") os.path.join(SCENARIONET_DATASET_PATH, "pg")
] ]
dataset_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset") dataset_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "tmp", "combine")
combine_multiple_dataset(dataset_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True) combine_multiple_dataset(dataset_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)
summary, sorted_scenarios, mapping = read_dataset_summary(dataset_path) summary, sorted_scenarios, mapping = read_dataset_summary(dataset_path)

4
scenarionet/tests/script/run_env.py
View File

@@ -4,7 +4,7 @@ from metadrive.envs.scenario_env import ScenarioEnv
from metadrive.policy.replay_policy import ReplayEgoCarPolicy from metadrive.policy.replay_policy import ReplayEgoCarPolicy
from metadrive.scenario.utils import get_number_of_scenarios from metadrive.scenario.utils import get_number_of_scenarios
from scenarionet import SCENARIONET_DATASET_PATH from scenarionet import SCENARIONET_DATASET_PATH, SCENARIONET_PACKAGE_PATH
from scenarionet.builder.utils import combine_multiple_dataset from scenarionet.builder.utils import combine_multiple_dataset
if __name__ == '__main__': if __name__ == '__main__':
@@ -13,7 +13,7 @@ if __name__ == '__main__':
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo")) dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg")) dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg"))
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset") combine_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "tmp", "combine")
combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True) combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)
env = ScenarioEnv( env = ScenarioEnv(