Views
No views yet
1# Register environment
2from road_env import register_road_envs
3register_road_envs()
4
5# Make environment
6import gymnasium as gym
7env = gym.make('urban-road-v0', render_mode='rgb_array')
8
9# Configure parameters (example)
10env.configure({
11 "random_seed": None,
12 "duration": 60,
13})
14
15obs, info = env.reset()
16
17# Graphic display
18import matplotlib.pyplot as plt
19plt.imshow(env.render())
20
21# Execution
22done = truncated = False
23while not (done or truncated):
24 action = ... # Your agent code here
25 obs, reward, done, truncated, info = env.step(action)
26 env.render() # Update graphic