Merge pull request #1559 from shubhtuls/patch-1

minor bugfix in image reshaping in kukaGymEnv.py
2018-03-08 15:16:30 -08:00
parent 94118bb2aa 5fdd7ed3b4
commit d69fefd7e9
1 changed files with 1 additions and 1 deletions
--- 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