OpenAi Gymを録画

せっかくなので動画をとろうと

ググってもコレといった解決策がないなか

参考になる部分はしつつ

video_recorder.pyの中身とか見ながら

たどり着いた回答が以下のコード

pythonはインデントにうるさかった

Atomエディタ終了してた

あと追加で色々いれた

sudo　がないと　not writeable　とか言われた

sudo pip install --upgrade gym

sudo pip install gym[atari]

pip install ffmpeg

sudo pip install ffmpeg

sudo pip install imageio-ffmpeg

sudo pip install moviepy

    moviepyはvideo_recorder.pyの注意書きにあった

以下pythonのコード

内容はともかく

録画できたので良しとする

import gym

from gym.wrappers.monitoring.video_recorder import VideoRecorder

env=gym.make("LunarLander-v2",render_mode="rgb_array")

vid = VideoRecorder(env, "./video.mp4")

env.action_space.seed(42)

observation, info = env.reset(seed=42)

for _ in range(1000):

observation, reward, terminated, truncated, info = env.step(env.action_space.sample())

vid.capture_frame()

if terminated or truncated:

observation, info = env.reset()

env.close()

vid.close()