From cfe1b2c70f40338309989870dfd0b9bc45b9d76b Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Mon, 19 Aug 2024 15:14:49 +0200
Subject: [PATCH] Removed unnecessary np type cast
---
cooperative_cuisine/pygame_2d_vis/drawing.py | 4 +---
.../pygame_2d_vis/drawing_perf_benchmark.py | 9 ++++-----
2 files changed, 5 insertions(+), 8 deletions(-)
diff --git a/cooperative_cuisine/pygame_2d_vis/drawing.py b/cooperative_cuisine/pygame_2d_vis/drawing.py
index 84176c29..88ece8da 100644
--- a/cooperative_cuisine/pygame_2d_vis/drawing.py
+++ b/cooperative_cuisine/pygame_2d_vis/drawing.py
@@ -943,9 +943,7 @@ class Visualizer:
def get_state_image(self, state: dict,
cache_flags: CacheFlags = CacheFlags.COUNTERS | CacheFlags.BACKGROUND) -> npt.NDArray[np.uint8]:
- screen = self.draw_gamescreen(
- state, [0 for _ in state["players"]], cache_flags=cache_flags
- )
+ screen = self.draw_gamescreen(state, [0 for _ in state["players"]], cache_flags=cache_flags)
return pygame.surfarray.pixels3d(screen)
def draw_recipe_image(
diff --git a/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py b/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
index 42cf7423..d12d3c67 100644
--- a/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
+++ b/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
@@ -47,11 +47,10 @@ def run_benchmark(GRID_SIZE, ITERS, OPTIMIZATIONS):
time_start = time.perf_counter()
observation = visualizer.get_state_image(state=json_dict, cache_flags=OPTIMIZATIONS["cache_counters"])
- img = observation.astype(np.uint8).transpose((1, 0, 2))
- img = cv2.resize(img, (400, 400), interpolation=cv2.INTER_NEAREST)
- cv2.imshow("env", img[:, :, ::-1])
- cv2.waitKey(1)
+ # img = cv2.resize(img, (400, 400), interpolation=cv2.INTER_NEAREST)
+ # cv2.imshow("env", img[:, :, ::-1])
+ # cv2.waitKey(1)
durations.append(time.perf_counter() - time_start)
print(f"Mean duration: %.6fs" % np.mean(durations))
@@ -75,7 +74,7 @@ if __name__ == "__main__":
with open(ROOT_DIR / "pygame_2d_vis" / "visualization.yaml", "r") as file:
visualization_config = yaml.safe_load(file)
- GRID_SIZE = 100
+ GRID_SIZE = 64
ITERS = 1000
OPTIMIZATIONS = {"use_array3d": True, "cache_counters": CacheFlags.BACKGROUND | CacheFlags.COUNTERS}
--
GitLab