Сохранение видео / GIF-файла для среды Open AI GymTaxi

Question

Я пытаюсь создать GIF-файл для среды Open AI Gym Taxi. Я могу видеть довольно много кодов для сред, которые являются объектами изображения. Но среда Такси - это String, и я не уверен, как сохранить визуализированные среды в виде файлов jpg или png.

Вот что у меня есть:

import gym
env = gym.make('Taxi-v2')
state=env.reset()
while True:
    t += 1
    env.render()
    action = env.action_space.sample()
    state, reward, done, info = env.step(action)
    if done:
        print("Episode finished after {} timesteps".format(t+1))
        break
env.close()

Я пробовал:

env = gym.wrappers.Monitor(env, './video/',video_callable=lambda episode_id: True,force = True)

но это просто создает файл json.

Любая помощь искренне приветствуется. Спасибо

Answer 1

У вас установлен FFmpeg? Если нет, просто установите его с помощью:

apt-get install ffmpeg

для Debian или:

brew install ffmpeg

в OS X.

Редактировать (уточнение):

Видеорегистратор тренажерного зала начинает с создания кадров в секунду и выгрузки их метаданных в формате json, но перед рендерингом этих кадров требуется кодер для сжатия, поэтому инициируется подпроцесс поиска кодера ffmpeg, который может привести к ошибке зависимости если не было найдено и метаданных первых кадров. Для дальнейшего понимания проверьте исходный код, указанный в gym/wrappers/monitoring/video_recorder.py.