是否可以在训练前和训练期间修改OpenAI健身房状态？

Question

我想要做的是修改一个环境，例如，以Super Mario Bros健身房环境为例，模糊智能体训练的图像，看看强化学习智能体是否仍然能够在这些“模糊”状态上学习。

OpenAI使这样的事情成为可能吗？如何添加健身房环境预处理步骤？

Answer 1

我建议您为健身房环境制作一个包装器，在step()和reset()函数中添加处理

下面是一些代码来说明这个想法：

class EnvWrapper(gym.Env):
    def __init__(self, config):
        self.env = gym.make("Your-Env-Name")    # The wrapper encapsulates the gym env
    
    def step(self, action):
        obs, reward, done, info = self.env.step(action)   # calls the gym env methods
        obs = self._blur(obs)                             # applies your specific treatment
        return obs, reward, done, info

    def reset(self):
        obs = self.env.reset()    # same for reset
        return self._blur(obs)

    def _blur(self):
        do_whatever_you_need

使用这种方法，您不需要对原始环境进行任何更改，这通常是一个好主意