問題
由于服務器上沒有圖形化界面,所以在調(diào)用gym中的render()
函數(shù)時,會報錯pyglet.canvas.xlib.NoSuchDisplayException: Cannot connect to "None"
。
思路是:把視頻保存下來,下載到本地再看。
解決方案
直接調(diào)用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函數(shù)
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數(shù),只記錄特定的episode;force
為是否覆蓋之前的視頻。
然后在終端輸入:
xvfb-run -a -s "-screen 0 640x480x24" python test.py
如果不適用xvfb運行的話,還是會報錯
可能遇到的問題
當運行之后,發(fā)現(xiàn)視頻無法打開,那是因為在源碼中有一個bug,參考Fix video recordinggym/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())
應該修改為:文章來源:http://www.zghlxwxcb.cn/news/detail-405808.html
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())
即:找到gym包的安裝目錄,把video_recorder.py
中的self.proc.stdin.write(frame.tobytes())
這一行代碼放到if代碼塊外面來文章來源地址http://www.zghlxwxcb.cn/news/detail-405808.html
總結(jié)
- 調(diào)用Monitor保存錄像
- 要使用
xvfb-run
運行代碼 - 修改gym源碼中的一個bug
參考資料
- Extending OpenAI Gym environments with Wrappers and Monitors [Tutorial]
- Fix video recording.
- Record OpenAI gym Video with Monitor
到了這里,關于服務器無法調(diào)用gym中的render,采用Monitor保存視頻的方法解決的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!