OpenAI 健身房:如何在 CartPole-v0 中获取像素

Tok*_*rby 6 reinforcement-learning deep-learning openai-gym

我想在CartPole-v0不打开渲染窗口的情况下访问 OpenAI 健身房环境中的原始像素。我该怎么做呢?

示例代码:

import gym
env = gym.make("CartPole-v0")
env.reset()
img = env.render(mode='rgb_array', close=True) # Returns None
print(img)
img = env.render(mode='rgb_array', close=False) 
          # Opens annoying window, but gives me the array that I want
print(img.shape)
Run Code Online (Sandbox Code Playgroud)

附注。我很难为 OpenAI 健身房找到好的文档。它只是我,还是根本不存在?

编辑:我不需要打开渲染视频。

Sta*_*low -1

有人在这里给出了答案:

https://github.com/openai/gym/issues/374

“atari 和 doom 环境在其观察中给出了像素(即步骤的返回值)。我认为其他环境不会这样做。

render 在不同的操作系统上产生不同的结果,因此它们不属于任何用于基准测试目的的官方环境。但如果您想创建一个以像素为单位进行观察的新环境,您可以通过包装现有环境并调用渲染来实现它。”

我也在努力获取原始像素,并且我正在尝试找到一种方法来查看返回的内容是否是我所期望的。

该文档可以找到: https: //gym.openai.com/docs

还有一个讨论 OpenAI 的论坛:discuss.openai.com

虽然不是很热闹。