我想使用tensorflow训练特工打超级马里奥兄弟。因此,我使用Gym程序包来模拟游戏。对于神经网络来说,环境的大小(256 x 240像素)似乎很大。因此,我想将其调整为某个大小(a x b)。如何才能做到这一点?我浏览了gym-git,发现了python脚本resize_observation。我不确定这是否能胜任。我能够成功运行附件中的代码,但是如果不调整大小,则没有明显区别。
'''
from nes_py.wrappers import JoypadSpace
import gym_super_mario_bros
from gym_super_mario_bros.actions import SIMPLE_MOVEMENT
import gym.wrappers.resize_observation as ro
env = gym_super_mario_bros.make('SuperMarioBros-v0')
env = JoypadSpace(env, SIMPLE_MOVEMENT)
env = ro.ResizeObservation(env, 10)
done = True
for step in range(5000):
if done:
state = env.reset()
state, reward, done, info = env.step(env.action_space.sample())
env.render()
env.close()
'''