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

fix __init__.py

Browse Source
This commit is contained in:
Erwin Coumans
2018-02-21 10:47:24 -08:00
parent 67318f3323
commit 8ba9644e78
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/pybullet/gym/pybullet_envs/prediction/__init__.py
View File

@@ -1,3 +1,3 @@
from . import boxstack_pybullet_simy from . import boxstack_pybullet_sim
from . import pybullet_sim_gym_envy from . import pybullet_sim_gym_env

2
examples/pybullet/gym/pybullet_envs/prediction/pybullet_sim_gym_env.py
View File

@@ -190,7 +190,7 @@ class PyBulletSimGymEnv(gym.Env):
proj_matrix=[1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, -1.0000200271606445, -1.0, 0.0, 0.0, -0.02000020071864128, 0.0] proj_matrix=[1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, -1.0000200271606445, -1.0, 0.0, 0.0, -0.02000020071864128, 0.0]
(_, _, px, _, _) = self._pybullet_client.getCameraImage( (_, _, px, _, _) = self._pybullet_client.getCameraImage(
width=self._render_width, height=self._render_height, viewMatrix=view_matrix, width=self._render_width, height=self._render_height, viewMatrix=view_matrix,
projectionMatrix=proj_matrix, renderer=pybullet.ER_TINY_RENDERER) projectionMatrix=proj_matrix, renderer=pybullet.ER_BULLET_HARDWARE_OPENGL)#ER_TINY_RENDERER)
rgb_array = np.array(px, dtype=np.uint8) rgb_array = np.array(px, dtype=np.uint8)
rgb_array = np.reshape(rgb_array, (self._render_height, self._render_width, 4)) rgb_array = np.reshape(rgb_array, (self._render_height, self._render_width, 4))
rgb_array = rgb_array[:, :, :3] rgb_array = rgb_array[:, :, :3]