bb403781182c6e31d3bf5de16f42b0cb0d8421f7,baselines/common/atari_wrappers.py,WarpFrame,_observation,#WarpFrame#Any#,127

Before Change

    def _observation(self, obs):
        frame = np.dot(obs.astype("float32"), np.array([0.299, 0.587, 0.114], "float32"))
        frame = np.array(Image.fromarray(frame).resize((self.res, self.res),
            resample=Image.BILINEAR), dtype=np.uint8)
        return frame.reshape((self.res, self.res, 1))

class FrameStack(gym.Wrapper):
    def __init__(self, env, k):
        Buffer observations and stack across channels (last axis).

After Change

    def _observation(self, frame):
        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
        frame = cv2.resize(frame, (self.width, self.height), interpolation=cv2.INTER_AREA)
        return frame[:, :, None]

class FrameStack(gym.Wrapper):
    def __init__(self, env, k):
        Stack k last frames.

Instances