Georg commited on
Commit
bec2a82
·
1 Parent(s): 7e77839

Update .gitignore and enhance depth snapshot rendering in mujoco_server.py

Browse files

- Added MUJOCO_LOG.TXT to .gitignore to prevent logging files from being tracked.
- Modified the capture_depth_snapshot function to dynamically set render height and width based on the camera name, improving flexibility in depth rendering.

Files changed (2) hide show
  1. .gitignore +1 -0
  2. mujoco_server.py +9 -2
.gitignore CHANGED
@@ -58,3 +58,4 @@ mujoco_server.py.backup
58
  .pid-*
59
  .dev-*.log
60
  server.log
 
 
58
  .pid-*
59
  .dev-*.log
60
  server.log
61
+ MUJOCO_LOG.TXT
mujoco_server.py CHANGED
@@ -1934,10 +1934,17 @@ def capture_depth_snapshot(camera_name: str) -> Optional[bytes]:
1934
  # Create a temporary renderer for depth capture
1935
  with mujoco_lock:
1936
  try:
 
 
 
 
 
 
 
1937
  depth_renderer = mujoco.Renderer(
1938
  env.model,
1939
- height=env.height,
1940
- width=env.width
1941
  )
1942
 
1943
  # Enable depth rendering
 
1934
  # Create a temporary renderer for depth capture
1935
  with mujoco_lock:
1936
  try:
1937
+ if camera_name == "main":
1938
+ render_height = env.height
1939
+ render_width = env.width
1940
+ else:
1941
+ render_height = OVERLAY_RENDER_HEIGHT
1942
+ render_width = OVERLAY_RENDER_WIDTH
1943
+
1944
  depth_renderer = mujoco.Renderer(
1945
  env.model,
1946
+ height=render_height,
1947
+ width=render_width
1948
  )
1949
 
1950
  # Enable depth rendering