pong_wrapper.py 文件源码

def get_screen(self):
    screen = self.env.render(mode='rgb_array')
    screen = color.rgb2gray(screen)
    screen = imresize(screen, (110, 84))
    screen = screen[18:102][:] / 255.0
    return screen.astype(np.float)