Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1559 from shubhtuls/patch-1

Browse Source 
minor bugfix in image reshaping in kukaGymEnv.py
This commit is contained in:
erwincoumans
2018-03-08 15:16:30 -08:00
committed by GitHub
parent 94118bb2aa 5fdd7ed3b4
commit d69fefd7e9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pybullet/gym/pybullet_envs/bullet/kukaGymEnv.py
View File

@@ -205,7 +205,7 @@ class KukaGymEnv(gym.Env):
rgb_array = np.array(px, dtype=np.uint8) 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] rgb_array = rgb_array[:, :, :3]
return rgb_array return rgb_array