是否可以在OpenAI体育馆中获取环境图像?

时间:2020-03-29 07:26:12

标签: reinforcement-learning openai-gym

在Atari太空入侵者这样的环境中,环境的状态就是它的图像,因此在下面的代码行中

observation, action, reward, _ = env.step() observation变量保存环境的实际图像,但是对于像Cartpole这样的环境,观察将是一些标量数。

是否可以以某种方式访问​​那些环境中的状态图?

2 个答案:

答案 0 :(得分:1)

env.render(mode="rgb_array")

这将返回您可以存储的渲染图像(数组)。可以使用env.viewer.cam的距离,方位角和高程属性设置摄像机角度

答案 1 :(得分:0)

import gym
env = gym.make('Breakout-v0')
env.reset()
env.env.ale.saveScreenPNG('test_image2.png')