(self, env, k)
Buffer observations and stack across channels (last axis).
Buffer observations and stack across channels (last axis).
Buffer observations and stack across channels ( last axis ) .
def __init__(self, env, k):
        """Buffer observations and stack across channels (last axis)."""
        gym.Wrapper.__init__(self, env)
        self.k = k
        self.frames = deque([], maxlen=k)
        shp = env.observation_space.shape
        assert shp[2] == 1  # can only stack 1-channel frames
        self.observation_space = spaces.Box(low=0, high=255, shape=(shp[0], shp[1], k))
def __init__ ( self , env , k ) : gym . Wrapper . __init__ ( self , env ) self . k = k self . frames = deque ( [ ] , maxlen = k ) shp = env . observation_space . shape assert shp [ 2 ] == 1 # can only stack 1-channel frames self . observation_space = spaces . Box ( low = 0 , high = 255 , shape = ( shp [ 0 ] , shp [ 1 ] , k ) )