From 8f9ff5e6744412a359bf79ad98ab141bfb0d7e79 Mon Sep 17 00:00:00 2001 From: YungKC Date: Fri, 12 Apr 2019 17:42:26 -0700 Subject: [PATCH] Fix render issue in server render mode (e.g., docker) px is an 1D array, and cause a runtime error when trying to strip the color channels into RGB. I fixed this by first reshape the px array into am image compatible format first. --- examples/pybullet/gym/pybullet_envs/env_bases.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/pybullet/gym/pybullet_envs/env_bases.py b/examples/pybullet/gym/pybullet_envs/env_bases.py index 861363348..aeef7fda5 100644 --- a/examples/pybullet/gym/pybullet_envs/env_bases.py +++ b/examples/pybullet/gym/pybullet_envs/env_bases.py @@ -96,6 +96,7 @@ class MJCFBaseBulletEnv(gym.Env): renderer=pybullet.ER_BULLET_HARDWARE_OPENGL ) rgb_array = np.array(px) + rgb_array = np.reshape(np.array(px), (self._render_height, self._render_width, -1)) rgb_array = rgb_array[:, :, :3] return rgb_array