admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-07 01:51:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

depth(ZMQ): excluding ZMQ from depth support

Browse Source
This commit is contained in:
CarolinePascal
2026-06-25 10:47:28 +02:00
parent b42332f43d
commit b95cf4e4c2
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/robots/unitree_g1/unitree_g1.py
+2 -8
View File
@@ -225,10 +225,7 @@ class UnitreeG1(Robot):
features: dict[str, tuple] = {} features: dict[str, tuple] = {}
for cam in self.cameras: for cam in self.cameras:
cfg = self.config.cameras[cam] cfg = self.config.cameras[cam]
if getattr(cfg, "use_rgb", True): features[cam] = (cfg.height, cfg.width, 3)
features[cam] = (cfg.height, cfg.width, 3)
if getattr(cfg, "use_depth", False):
features[f"{cam}_depth"] = (cfg.height, cfg.width, 1)
return features return features
@cached_property @cached_property
@@ -463,10 +460,7 @@ class UnitreeG1(Robot):
# Cameras - read images from ZMQ cameras # Cameras - read images from ZMQ cameras
for cam_name, cam in self._cameras.items(): for cam_name, cam in self._cameras.items():
if getattr(cam, "use_rgb", True): obs[cam_name] = cam.read_latest()
obs[cam_name] = cam.read_latest()
if getattr(cam, "use_depth", False):
obs[f"{cam_name}_depth"] = cam.read_latest_depth()
return obs return obs