服务器无法调用gym中的render，采用Monitor保存视频的方法解决

躁动的风儿

已于 2022-06-04 11:25:38 修改

阅读量1.9k

点赞数 2

分类专栏：深度学习文章标签： python 深度学习人工智能

于 2022-05-21 10:25:25 首次发布

本文链接：https://blog.csdn.net/m0_59019651/article/details/124894736

版权

深度学习专栏收录该内容

4 篇文章 0 订阅

订阅专栏

问题

由于服务器上没有图形化界面，所以在调用gym中的render()函数时，会报错pyglet.canvas.xlib.NoSuchDisplayException: Cannot connect to "None"。
思路是：把视频保存下来，下载到本地再看。

解决方案

直接调用gym.wrappers.Monitor把视频保存到本地
代码示例如下：

# test.py

import gym
from gym.wrappers import Monitor

outdir = 'video_record'
env = gym.make('MountainCar-v0')
env = Monitor(env, outdir, video_callable=lambda episode_id: True,  force=True) # 主要是修改这行代码，把env传递给Monitor函数
state = env.reset()
done = False
while not done:
    action = env.action_space.sample()
    state_next, reward, done, info = env.step(action)
env.close()

outdir为视频保存路径；video_callable表示多少个episode记录一次视频，可将True改为指定的episode数，只记录特定的episode；force为是否覆盖之前的视频。

然后在终端输入：

xvfb-run -a -s "-screen 0 640x480x24" python test.py

如果不适用xvfb运行的话，还是会报错

可能遇到的问题

当运行之后，发现视频无法打开，那是因为在源码中有一个bug，参考Fix video recording
gym/wrappers/monitoring/video_recorder.py源代码如下：

if frame.dtype != np.uint8:
	raise error.InvalidFrame("Your frame has data type {}, but we require uint8 (i.e. RGB values from 0-255).".format(frame.dtype))

	self.proc.stdin.write(frame.tobytes())

应该修改为：

if frame.dtype != np.uint8:
	raise error.InvalidFrame("Your frame has data type {}, but we require uint8 (i.e. RGB values from 0-255).".format(frame.dtype))

self.proc.stdin.write(frame.tobytes())