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.
- .gitignore +1 -0
- 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=
|
| 1940 |
-
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
|