Readded tranpose

cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py

@@ -47,10 +47,9 @@ 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 = cv2.resize(img, (400, 400), interpolation=cv2.INTER_NEAREST)
-        # cv2.imshow("env", img[:, :, ::-1])
-        # cv2.waitKey(1)
+        img = cv2.resize(observation.transpose((1, 0, 2)), (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))