Редактировать: я добился некоторого прогресса, так что вот обновленный код < /p>
Код: Выделить всё
from PIL import ImageGrab, Image
import numpy as np
import torch
from torch import nn
from torchvision import transforms
import torch.optim as optim
# imporot keyboard and pyautogui for playing the game
_bbox=(800, 200, 1200, 900) # (L,T,R,B)
dead = False
def get():
global dead
screenshot = ImageGrab.grab(bbox=_bbox)
# Check if dead
if np.sum(np.array(screenshot)[:,:,0])/(screenshot.size[0]*screenshot.size[1])>=190 and np.sum(np.array(screenshot)[:,:,1])/(screenshot.size[0]*screenshot.size[1])
Подробнее здесь: [url]https://stackoverflow.com/questions/79398991/using-pytorch-sequential-for-reinforcement-model-without-gym[/url]