state.py 文件源码

def getstate(self):
        win32gui.SetForegroundWindow(self.hwnd)
        bbox = win32gui.GetWindowRect(self.hwnd)

        img = ImageGrab.grab(bbox)
        img = img.resize([84, 84], Image.ANTIALIAS)
        pilim = np.array(img, dtype=np.uint8)
        pilim = rgb2gray(pilim)

        return pilim