Spaces:
Sleeping
Sleeping
Fix error
Browse files- tray_sim.py +3 -2
tray_sim.py
CHANGED
|
@@ -32,7 +32,8 @@ def run_tray_simulation(seed=0, num_objects=N_OBJECTS, azimuth=45, elevation=-25
|
|
| 32 |
renderer = mujoco.Renderer(model, width=480, height=480) # width=640, height=640
|
| 33 |
|
| 34 |
# Custom camera parameters
|
| 35 |
-
|
|
|
|
| 36 |
cam.type = mujoco.mjtCamera.mjCAMERA_FREE
|
| 37 |
cam.lookat[:] = np.array([0.0, 0.0, 0.02]) # Center of the tray
|
| 38 |
cam.distance = distance # Zoom
|
|
@@ -48,7 +49,7 @@ def run_tray_simulation(seed=0, num_objects=N_OBJECTS, azimuth=45, elevation=-25
|
|
| 48 |
data.qvel[pusher_vel_idx:pusher_vel_idx+3] = [-0.05, 0.05, 0.0] # toward center
|
| 49 |
|
| 50 |
mujoco.mj_step(model, data)
|
| 51 |
-
renderer.update_scene(data)
|
| 52 |
img = renderer.render()
|
| 53 |
|
| 54 |
if t in keyframes:
|
|
|
|
| 32 |
renderer = mujoco.Renderer(model, width=480, height=480) # width=640, height=640
|
| 33 |
|
| 34 |
# Custom camera parameters
|
| 35 |
+
scene_option = mujoco.MjvOption()
|
| 36 |
+
cam = mujoco.MjvCamera()
|
| 37 |
cam.type = mujoco.mjtCamera.mjCAMERA_FREE
|
| 38 |
cam.lookat[:] = np.array([0.0, 0.0, 0.02]) # Center of the tray
|
| 39 |
cam.distance = distance # Zoom
|
|
|
|
| 49 |
data.qvel[pusher_vel_idx:pusher_vel_idx+3] = [-0.05, 0.05, 0.0] # toward center
|
| 50 |
|
| 51 |
mujoco.mj_step(model, data)
|
| 52 |
+
renderer.update_scene(data, camera=cam, scene_option=scene_option)
|
| 53 |
img = renderer.render()
|
| 54 |
|
| 55 |
if t in keyframes:
|