Feature/recording

.gitignore

@@ -1,3 +1,4 @@
 .idea/ 
 venv/
-__pycache__/
+__pycache__/
+video/

golem_driving/envs/__init__.py

@@ -16,7 +16,7 @@ def add_env_args(parser: ArgumentParser) -> type(None):
     parser.add_argument('--domain-rand', action='store_true', help='enable domain randomization')
     parser.add_argument('--frame-skip', default=1, type=int, help='number of frames to skip')
     parser.add_argument('--frame-rate', default=None, help='number of frames per second')
-
+    parser.add_argument('--dir', default='./video', help='directory in which recordings are stored')
 
 def get_env_from_args(args, discrete: bool=False) -> Callable[[], Simulator]:
     def builder():
@@ -28,6 +28,7 @@ def builder():
                 domain_rand=args.domain_rand,
                 frame_skip=args.frame_skip,
                 distortion=args.distortion,
+
             )
         else:
             env = gym.make(args.env_name)

golem_driving/run/show.py

@@ -1,12 +1,10 @@
 import pyglet
+from gym.wrappers import Monitor
 from pyglet.window import key
-from gym_duckietown.simulator import Simulator
 
-from golem_driving.agents.agent import Agent
-from golem_driving.config import ShowConfig
 
-
-def show(env: Simulator, agent: Agent, config: ShowConfig) -> type(None):
+def show(env, agent, config, directory):
+    env = Monitor(env, directory, video_callable=lambda episode_id: True, force=True)
     obs = env.reset()
     env.render()
 

main.py

@@ -17,12 +17,13 @@
     args = parser.parse_args()
 
     mode = modes.get(args.mode)
-
+    directory = args.dir
+    print(directory)
     with open(args.config, 'r') as config_file:
         config = mode.config()
         config.load(config_file)
 
     with get_env_from_args(args, discrete=False) as env:
         agent = config.build_agent()
 
-        mode.run(env=env, agent=agent, config=config)
+        mode.run(env=env, agent=agent, config=config, directory=directory)

requirements.txt

@@ -1,3 +1,4 @@
+numpy
 gym
 tensorflow==1.*
 baselines