openai-gym - 指定 OpenAi Gym 的渲染窗口大小

Question

调用 env.render() 总是渲染一个充满整个屏幕的窗口。

env = gym.make('FetchPickAndPlace-v1')
env.reset()
for i in range(1000):
    env.render()

Answer 1

你可以试试这个:

import matplotlib 

is_ipython = 'inline' in matplotlib.get_backend()
if is_ipython: from IPython import display

env = gym.make('FetchPickAndPlace-v1')

a = env.render(mode='rgb_array')
plt.figure()
plt.imshow(a)
plt.show()
if is_ipython: display.clear_output(wait=True)