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

Merge pull request #2200 from YungKC/patch-1

Browse Source 
Fix render issue in server render mode (e.g., docker)
This commit is contained in:
erwincoumans
2019-04-16 06:31:58 -07:00
committed by GitHub
parent 2a9bfbdb1c 8f9ff5e674
commit bf0eb4703d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/pybullet/gym/pybullet_envs/env_bases.py
View File

@@ -96,6 +96,7 @@ class MJCFBaseBulletEnv(gym.Env):
renderer=pybullet.ER_BULLET_HARDWARE_OPENGL renderer=pybullet.ER_BULLET_HARDWARE_OPENGL
) )
rgb_array = np.array(px) rgb_array = np.array(px)
rgb_array = np.reshape(np.array(px), (self._render_height, self._render_width, -1))
rgb_array = rgb_array[:, :, :3] rgb_array = rgb_array[:, :, :3]
return rgb_array return rgb_array