chrisjcc commited on
Commit
f2c36a1
·
verified ·
1 Parent(s): cf6cc37
Files changed (1) hide show
  1. 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
- cam = renderer.cam
 
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: