shakenes / vizdoomgym Goto Github PK

def show_video():
  mp4list = glob.glob('video/*.mp4')
  if len(mp4list) > 0:
    mp4 = mp4list[0]
    video = io.open(mp4, 'r+b').read()
    encoded = base64.b64encode(video)
    ipythondisplay.display(HTML(data='''<video alt="test" autoplay 
                loop controls style="height: 400px;">
                <source src="data:video/mp4;base64,{0}" type="video/mp4" />
             </video>'''.format(encoded.decode('ascii'))))
  else: 
    print("Could not find video")
    

def wrap_env(env):
  env = Monitor(env, './video', force=True)
  return env
environment = wrap_env(env)
done = False
observation = environment.reset()
new_observation = observation

prev_input = None


environment = wrap_env(env)
done = False
observation = environment.reset()
new_observation = observation

prev_input = None
with tf.compat.v1.Session() as sess:
    init.run()
    observation, stacked_frames = stack_frames(stacked_frames, observation, True)
    
    while True:
       
    
        #set input to network to be difference image
  
        
        #print(observation.shape)
        
        

        # feed the game screen and get the Q values for each action
        actions = mainQ_outputs.eval(feed_dict={X:[observation], in_training_mode:False})

        # get the action
        action = np.argmax(actions, axis=-1)
        actions_counter[str(action)] += 1 

        # select the action using epsilon greedy policy
        action = epsilon_greedy(action, global_step)
        environment.render()
        new_observation, stacked_frames = stack_frames(stacked_frames, new_observation, False)
        
        observation = new_observation        
        # now perform the action and move to the next state, next_obs, receive reward
        new_observation, reward, done, _ = environment.step(action)
        
       
        if done: 
          
          break
      
    environment.close()
    show_video()

Exception ignored in: <bound method SimpleImageViewer.__del__ of <gym.envs.classic_control.rendering.SimpleImageViewer object at 0x7fa15f523a90>>
Traceback (most recent call last):
  File "/home/colin/py3/lib/python3.6/site-packages/gym/envs/classic_control/rendering.py", line 359, in __del__
  File "/home/colin/py3/lib/python3.6/site-packages/gym/envs/classic_control/rendering.py", line 355, in close
  File "/home/colin/py3/lib/python3.6/site-packages/pyglet/window/xlib/__init__.py", line 480, in close
  File "/home/colin/py3/lib/python3.6/site-packages/pyglet/gl/xlib.py", line 345, in destroy
  File "/home/colin/py3/lib/python3.6/site-packages/pyglet/gl/base.py", line 334, in destroy
  File "/home/colin/py3/lib/python3.6/site-packages/pyglet/gl/xlib.py", line 335, in detach
  File "/home/colin/py3/lib/python3.6/site-packages/pyglet/gl/lib.py", line 97, in errcheck
ImportError: sys.meta_path is None, Python is likely shutting down

import gym
import vizdoomgym

env = gym.make('VizdoomBasic-v0')