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

format

Browse Source
This commit is contained in:
QuanyiLi
2023-08-08 17:51:04 +01:00
parent fac13e2e5f
commit 52ff95a078
1 changed files with 35 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
scenarionet/tests/script/run_ckpt.py
View File

@@ -7,41 +7,41 @@ from scenarionet_training.scripts.train_nuplan import config
if __name__ == "__main__": if __name__ == "__main__":
initialize_ray(test_mode=False, num_gpus=1) initialize_ray(test_mode=False, num_gpus=1)
env = createGymWrapper(ScenarioEnv)({ env = createGymWrapper(ScenarioEnv)(
# "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False), {
"data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw", # "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False),
"use_render": True, "data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw",
# "agent_policy": ReplayEgoCarPolicy, "use_render": True,
"show_interface": False, # "agent_policy": ReplayEgoCarPolicy,
"image_observation": False, "show_interface": False,
"show_logo": False, "image_observation": False,
"no_traffic": False, "show_logo": False,
"no_static_vehicles": False, "no_traffic": False,
"drivable_region_extension": 15, "no_static_vehicles": False,
"sequential_seed": True, "drivable_region_extension": 15,
"reactive_traffic": True, "sequential_seed": True,
"show_fps": False, "reactive_traffic": True,
"render_pipeline": True, "show_fps": False,
"daytime": "07:10", "render_pipeline": True,
"max_lateral_dist": 2, "daytime": "07:10",
"window_size": (1200, 800), "max_lateral_dist": 2,
"camera_dist": 9, "window_size": (1200, 800),
"start_scenario_index": 5, "camera_dist": 9,
"num_scenarios": 4000, "start_scenario_index": 5,
"horizon": 1000, "num_scenarios": 4000,
"store_map": False, "horizon": 1000,
"vehicle_config": dict( "store_map": False,
show_navi_mark=True, "vehicle_config": dict(
# no_wheel_friction=True, show_navi_mark=True,
use_special_color=False, # no_wheel_friction=True,
image_source="depth_camera", use_special_color=False,
lidar=dict(num_lasers=120, distance=50), image_source="depth_camera",
lane_line_detector=dict(num_lasers=0, distance=50), lidar=dict(num_lasers=120, distance=50),
side_detector=dict(num_lasers=0, distance=50) lane_line_detector=dict(num_lasers=0, distance=50),
), side_detector=dict(num_lasers=0, distance=50)
} ),
) }
)
# env.reset() # env.reset()
# #