40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
# Supported DonkeyCar tracks detected in code:
|
|
available_tracks = [
|
|
"generated_road",
|
|
"warehouse",
|
|
"sparkfun_avc",
|
|
"generated_track",
|
|
"mountain_track",
|
|
"roboracingleague_1",
|
|
"waveshare",
|
|
"mini_monaco",
|
|
"warren",
|
|
"thunderhill",
|
|
"circuit_launch",
|
|
]
|
|
from gym_donkeycar.envs.donkey_env import DonkeyEnv
|
|
import sys
|
|
|
|
print("Available tracks:")
|
|
for i, t in enumerate(available_tracks):
|
|
print(f"[{i}] {t}")
|
|
|
|
choice = input("Enter the number of the track to use: ")
|
|
try:
|
|
idx = int(choice.strip())
|
|
track = available_tracks[idx]
|
|
print(f"Loading track: {track}")
|
|
except Exception as e:
|
|
print(f"Invalid selection ({e}), using default track: {available_tracks[0]}")
|
|
track = available_tracks[0]
|
|
|
|
env = DonkeyEnv(level=track)
|
|
obs, info = env.reset()
|
|
for t in range(10):
|
|
action = env.action_space.sample()
|
|
obs, reward, terminated, truncated, info = env.step(action)
|
|
print(f"Step {t}: reward {reward}, done {terminated or truncated}")
|
|
if terminated or truncated:
|
|
obs, info = env.reset()
|
|
env.close()
|