prepare to train racecar using ZED camera pixels (CNN+DQN)

This commit is contained in:
Erwin Coumans
2017-06-13 16:04:50 -07:00
parent 0958e8f473
commit ee8fd56c5e
6 changed files with 228 additions and 3 deletions

View File

@@ -0,0 +1,26 @@
import gym
from envs.bullet.racecarZEDGymEnv import RacecarZEDGymEnv
from baselines import deepq
def main():
env = RacecarZEDGymEnv(renders=True)
act = deepq.load("racecar_model.pkl")
print(act)
while True:
obs, done = env.reset(), False
print("===================================")
print("obs")
print(obs)
episode_rew = 0
while not done:
env.render()
obs, rew, done, _ = env.step(act(obs[None])[0])
episode_rew += rew
print("Episode reward", episode_rew)
if __name__ == '__main__':
main()