steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-07-11 15:32:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Doors now Working,

Browse Source 
Pypi install adjustments.
This commit is contained in:
Steffen Illium
2023-06-16 09:58:28 +02:00
parent 45cb83746c
commit b4326b514c
12 changed files with 108 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
environment/utils/renderer.py
View File

@ -3,12 +3,12 @@ import sys
from pathlib import Path
from collections import deque
from itertools import product
import numpy as np
import pygame
from typing import Tuple, Union
import time
import torch
from environment.utils.render import RenderEntity
AGENT: str = 'agent'
@ -133,7 +133,8 @@ class Renderer:
pygame.display.flip()
self.clock.tick(self.fps)
rgb_obs = pygame.surfarray.array3d(self.screen)
return torch.from_numpy(rgb_obs).permute(2, 0, 1)
return np.transpose(rgb_obs, (2, 0, 1))
# return torch.from_numpy(rgb_obs).permute(2, 0, 1)
if __name__ == '__main__':