A recorder for open ai gym. you can easily add a text on the frame
Project description
gym_recorder
A recorder for open ai gym.
you can easily add a text on the frame like this.
install
pip install gym_record
usage
import gym
from gym_recorder import Recorder
src_env = gym.make("CartPole-v1")
env = Recorder(src_env, episode_num=10)
for ep in range(10):
obs = env.reset()
done = False
while not done:
action = env.action_space.sample()
n_obs, reward, done, info = env.step(action)
env.txtqueue.append(f"episode:{ep}")
env.txtqueue.append(f"obs:{obs}")
env.txtqueue.append(f"action:{action}")
env.txtqueue.append(f"reward:{reward}")
env.txtqueue.append(f"next obs:{n_obs}")
tips
this library depends on opencv-python. so you may need some library to use opencv. this is .Dockerfile example to resolve it.
RUN apt-get update && apt-get upgrade -y
RUN apt-get install -y libgl1-mesa-dev
if you use docker, you have to set "SDL_VIDEODRIVER" to "dummy".
add this to your .Dockerfile
ENV SDL_VIDEODRIVER=dummy
or simply add this to your script.
import os
os.environ["SDL_VIDEODRIVER"] = "dummy"
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
gym-record-0.0.3.tar.gz
(3.5 kB
view hashes)
Built Distribution
Close
Hashes for gym_record-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 45d8f592db8c5a02f94a7903146fc04b7e47441249643c5de159fbd5136e86c2 |
|
MD5 | d1add8567cfea30883b0d92f85d1382f |
|
BLAKE2b-256 | 825e4b38eb5664590700d74689b20822e5ae49dd375989c27a8a4a8ff0d1bc18 |