From 5fdd7ed3b47f2d428b33741c46d206354d87c71c Mon Sep 17 00:00:00 2001 From: Shubham Tulsiani Date: Sun, 11 Feb 2018 21:47:57 -0800 Subject: [PATCH] minor bugfix in image reshaping in kukaGymEnv Reshaped image array should be of size (RENDER_HEIGHT, RENDER_WIDTH, 4) instead of (RENDER_WIDTH, RENDER_HEIGHT, 4). --- examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py b/examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py index 848f10a90..373ca0839 100644 --- a/examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py +++ b/examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py @@ -205,7 +205,7 @@ class KukaGymEnv(gym.Env): rgb_array = np.array(px, dtype=np.uint8) - rgb_array = np.reshape(rgb_array, (RENDER_WIDTH, RENDER_HEIGHT, 4)) + rgb_array = np.reshape(rgb_array, (RENDER_HEIGHT, RENDER_WIDTH, 4)) rgb_array = rgb_array[:, :, :3] return rgb_array