mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-11 03:52:02 +00:00
feat(robots): natively integrate Seeed Studio reBot B601-DM arm (#3624)
* feat(robots): natively integrate Seeed Studio reBot B601-DM arm Add first-class LeRobot support for the Seeed Studio reBot arm, replacing the out-of-tree `lerobot-robot-seeed-b601` / `lerobot-teleoperator-rebot-arm-102` plugin packages. New devices: - robot `rebot_b601_follower` — single-arm B601-DM follower (6-DOF + gripper, Damiao CAN motors via `motorbridge`) - robot `bi_rebot_b601_follower` — bimanual follower composing two single arms - teleoperator `rebot_102_leader` — single-arm StarArm102 / reBot Arm 102 leader (FashionStar UART servos via `motorbridge-smart-servo`) - teleoperator `bi_rebot_102_leader` — bimanual leader composing two single arms The bimanual variants reuse the single-arm classes and namespace each arm's observation/action keys with `left_` / `right_` prefixes, so a bimanual StarArm102 leader can teleoperate a bimanual reBot B601 follower. Optional SDK imports are guarded; a `rebot` extra installs `motorbridge` and `motorbridge-smart-servo`. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * docs: add reBot B601-DM calibration & dual-arm teleoperation guide Add docs/source/rebot_b601.mdx covering single-arm and bimanual calibration and teleoperation for the reBot B601-DM follower and reBot Arm 102 leader, with zero-position reference images from the Seeed Studio wiki. Register the page in the docs toctree. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * docs: fix reBot B601 MDX build (move JSON example out of <Tip>) The doc-builder parses `{...}` inside MDX component children as a Svelte expression, so the joint_directions JSON example broke the build. Move it into a top-level fenced code block. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * docs: apply prettier formatting to reBot B601 page Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * docs: remove duplicate colocated reBot B601 page docs/source/rebot_b601.mdx is the canonical, toctree-registered page; the colocated rebot_b601.md was a redundant thinner copy. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * docs: clarify 6-DOF leader fallback comment in reBot B601 follower Explain that holding wrist_yaw at zero is what lets a 6-DOF leader (e.g. so100_leader / so101_leader) teleoperate the 7-DOF follower. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * refactor: address Caroline's PR review on reBot B601 integration - leader: remove _validate_config (no other lerobot device validates its config; a key mismatch now surfaces as a plain KeyError) - leader: simplify _round_to_valid_range to direct modular arithmetic instead of a bidirectional search loop - leader: inline the single-use _clamp helper - follower & leader: write MotorCalibration range_min/range_max from the configured joint_limits / joint_ranges instead of a fixed [-90, 90] - docs: add a "Find the USB ports" section (lerobot-find-port) and move the brltty/permissions tip there; link the OpenArm page for SocketCAN adapter configuration Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,20 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from .bi_rebot_b601_follower import BiRebotB601Follower
|
||||
from .config_bi_rebot_b601_follower import BiRebotB601FollowerConfig
|
||||
|
||||
__all__ = ["BiRebotB601Follower", "BiRebotB601FollowerConfig"]
|
||||
@@ -0,0 +1,150 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import logging
|
||||
from functools import cached_property
|
||||
|
||||
from lerobot.types import RobotAction, RobotObservation
|
||||
from lerobot.utils.decorators import check_if_already_connected, check_if_not_connected
|
||||
|
||||
from ..rebot_b601_follower import RebotB601Follower, RebotB601FollowerRobotConfig
|
||||
from ..robot import Robot
|
||||
from .config_bi_rebot_b601_follower import BiRebotB601FollowerConfig
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class BiRebotB601Follower(Robot):
|
||||
"""Bimanual Seeed Studio reBot B601-DM follower.
|
||||
|
||||
Composes two single-arm :class:`RebotB601Follower` instances. Observation and
|
||||
action keys of each arm are namespaced with a ``left_`` / ``right_`` prefix.
|
||||
"""
|
||||
|
||||
config_class = BiRebotB601FollowerConfig
|
||||
name = "bi_rebot_b601_follower"
|
||||
|
||||
def __init__(self, config: BiRebotB601FollowerConfig):
|
||||
super().__init__(config)
|
||||
self.config = config
|
||||
|
||||
left_arm_config = RebotB601FollowerRobotConfig(
|
||||
id=f"{config.id}_left" if config.id else None,
|
||||
calibration_dir=config.calibration_dir,
|
||||
port=config.left_arm_config.port,
|
||||
can_adapter=config.left_arm_config.can_adapter,
|
||||
dm_serial_baud=config.left_arm_config.dm_serial_baud,
|
||||
disable_torque_on_disconnect=config.left_arm_config.disable_torque_on_disconnect,
|
||||
max_relative_target=config.left_arm_config.max_relative_target,
|
||||
cameras=config.left_arm_config.cameras,
|
||||
motor_can_ids=config.left_arm_config.motor_can_ids,
|
||||
pos_vel_velocity=config.left_arm_config.pos_vel_velocity,
|
||||
gripper_torque_ratio=config.left_arm_config.gripper_torque_ratio,
|
||||
joint_limits=config.left_arm_config.joint_limits,
|
||||
)
|
||||
|
||||
right_arm_config = RebotB601FollowerRobotConfig(
|
||||
id=f"{config.id}_right" if config.id else None,
|
||||
calibration_dir=config.calibration_dir,
|
||||
port=config.right_arm_config.port,
|
||||
can_adapter=config.right_arm_config.can_adapter,
|
||||
dm_serial_baud=config.right_arm_config.dm_serial_baud,
|
||||
disable_torque_on_disconnect=config.right_arm_config.disable_torque_on_disconnect,
|
||||
max_relative_target=config.right_arm_config.max_relative_target,
|
||||
cameras=config.right_arm_config.cameras,
|
||||
motor_can_ids=config.right_arm_config.motor_can_ids,
|
||||
pos_vel_velocity=config.right_arm_config.pos_vel_velocity,
|
||||
gripper_torque_ratio=config.right_arm_config.gripper_torque_ratio,
|
||||
joint_limits=config.right_arm_config.joint_limits,
|
||||
)
|
||||
|
||||
self.left_arm = RebotB601Follower(left_arm_config)
|
||||
self.right_arm = RebotB601Follower(right_arm_config)
|
||||
|
||||
# Only for compatibility with parts of the codebase that expect `robot.cameras`.
|
||||
self.cameras = {**self.left_arm.cameras, **self.right_arm.cameras}
|
||||
|
||||
@property
|
||||
def _motors_ft(self) -> dict[str, type]:
|
||||
return {
|
||||
**{f"left_{k}": v for k, v in self.left_arm._motors_ft.items()},
|
||||
**{f"right_{k}": v for k, v in self.right_arm._motors_ft.items()},
|
||||
}
|
||||
|
||||
@property
|
||||
def _cameras_ft(self) -> dict[str, tuple]:
|
||||
return {
|
||||
**{f"left_{k}": v for k, v in self.left_arm._cameras_ft.items()},
|
||||
**{f"right_{k}": v for k, v in self.right_arm._cameras_ft.items()},
|
||||
}
|
||||
|
||||
@cached_property
|
||||
def observation_features(self) -> dict[str, type | tuple]:
|
||||
return {**self._motors_ft, **self._cameras_ft}
|
||||
|
||||
@cached_property
|
||||
def action_features(self) -> dict[str, type]:
|
||||
return self._motors_ft
|
||||
|
||||
@property
|
||||
def is_connected(self) -> bool:
|
||||
return self.left_arm.is_connected and self.right_arm.is_connected
|
||||
|
||||
@check_if_already_connected
|
||||
def connect(self, calibrate: bool = True) -> None:
|
||||
self.left_arm.connect(calibrate)
|
||||
self.right_arm.connect(calibrate)
|
||||
|
||||
@property
|
||||
def is_calibrated(self) -> bool:
|
||||
return self.left_arm.is_calibrated and self.right_arm.is_calibrated
|
||||
|
||||
def calibrate(self) -> None:
|
||||
self.left_arm.calibrate()
|
||||
self.right_arm.calibrate()
|
||||
|
||||
def configure(self) -> None:
|
||||
self.left_arm.configure()
|
||||
self.right_arm.configure()
|
||||
|
||||
@check_if_not_connected
|
||||
def get_observation(self) -> RobotObservation:
|
||||
obs_dict = {}
|
||||
obs_dict.update({f"left_{k}": v for k, v in self.left_arm.get_observation().items()})
|
||||
obs_dict.update({f"right_{k}": v for k, v in self.right_arm.get_observation().items()})
|
||||
return obs_dict
|
||||
|
||||
@check_if_not_connected
|
||||
def send_action(self, action: RobotAction) -> RobotAction:
|
||||
left_action = {
|
||||
key.removeprefix("left_"): value for key, value in action.items() if key.startswith("left_")
|
||||
}
|
||||
right_action = {
|
||||
key.removeprefix("right_"): value for key, value in action.items() if key.startswith("right_")
|
||||
}
|
||||
|
||||
sent_action_left = self.left_arm.send_action(left_action)
|
||||
sent_action_right = self.right_arm.send_action(right_action)
|
||||
|
||||
return {
|
||||
**{f"left_{k}": v for k, v in sent_action_left.items()},
|
||||
**{f"right_{k}": v for k, v in sent_action_right.items()},
|
||||
}
|
||||
|
||||
@check_if_not_connected
|
||||
def disconnect(self) -> None:
|
||||
self.left_arm.disconnect()
|
||||
self.right_arm.disconnect()
|
||||
@@ -0,0 +1,29 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
from ..config import RobotConfig
|
||||
from ..rebot_b601_follower import RebotB601FollowerConfig
|
||||
|
||||
|
||||
@RobotConfig.register_subclass("bi_rebot_b601_follower")
|
||||
@dataclass
|
||||
class BiRebotB601FollowerConfig(RobotConfig):
|
||||
"""Configuration class for the bimanual reBot B601-DM follower robot."""
|
||||
|
||||
left_arm_config: RebotB601FollowerConfig
|
||||
right_arm_config: RebotB601FollowerConfig
|
||||
@@ -0,0 +1,20 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from .config_rebot_b601_follower import RebotB601FollowerConfig, RebotB601FollowerRobotConfig
|
||||
from .rebot_b601_follower import RebotB601Follower
|
||||
|
||||
__all__ = ["RebotB601Follower", "RebotB601FollowerConfig", "RebotB601FollowerRobotConfig"]
|
||||
@@ -0,0 +1,94 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
from lerobot.cameras import CameraConfig
|
||||
|
||||
from ..config import RobotConfig
|
||||
|
||||
|
||||
@dataclass
|
||||
class RebotB601FollowerConfig:
|
||||
"""Base configuration class for the Seeed Studio reBot B601-DM follower arm.
|
||||
|
||||
The B601-DM is a 6-DOF arm plus gripper driven by Damiao CAN motors. Motor
|
||||
communication goes through the ``motorbridge`` package.
|
||||
"""
|
||||
|
||||
# Communication port. For ``can_adapter="damiao"`` this is the Damiao serial
|
||||
# bridge device (e.g. "/dev/ttyACM0"); for ``can_adapter="socketcan"`` it is
|
||||
# the CAN channel name (e.g. "can0").
|
||||
port: str
|
||||
|
||||
# CAN adapter type:
|
||||
# "damiao" - Damiao dedicated serial bridge (default)
|
||||
# "socketcan" - SocketCAN based adapters (PCAN, slcan, embedded controllers, ...)
|
||||
can_adapter: str = "damiao"
|
||||
|
||||
# Baud rate for the Damiao serial bridge (only used when can_adapter="damiao").
|
||||
dm_serial_baud: int = 921600
|
||||
|
||||
disable_torque_on_disconnect: bool = True
|
||||
|
||||
# `max_relative_target` limits the magnitude of the relative positional target
|
||||
# vector for safety purposes (in degrees). Set to a positive scalar to apply the
|
||||
# same value to all motors, or to a dict mapping motor names to per-motor values.
|
||||
max_relative_target: float | dict[str, float] | None = None
|
||||
|
||||
# cameras
|
||||
cameras: dict[str, CameraConfig] = field(default_factory=dict)
|
||||
|
||||
# Maps motor names to their (send_can_id, recv_can_id) pair.
|
||||
motor_can_ids: dict[str, tuple[int, int]] = field(
|
||||
default_factory=lambda: {
|
||||
"shoulder_pan": (0x01, 0x11),
|
||||
"shoulder_lift": (0x02, 0x12),
|
||||
"elbow_flex": (0x03, 0x13),
|
||||
"wrist_flex": (0x04, 0x14),
|
||||
"wrist_yaw": (0x05, 0x15),
|
||||
"wrist_roll": (0x06, 0x16),
|
||||
"gripper": (0x07, 0x17),
|
||||
}
|
||||
)
|
||||
|
||||
# Target velocity for joints running in POS_VEL mode, in degrees/s. A scalar is
|
||||
# applied to every joint; a list provides one value per joint (in motor order).
|
||||
pos_vel_velocity: float | list[float] = field(default_factory=lambda: [150.0] * 7)
|
||||
|
||||
# Torque/current ratio for the gripper's FORCE_POS mode, in range [0, 1].
|
||||
gripper_torque_ratio: float = 0.1
|
||||
|
||||
# Soft joint limits (degrees). These are clipped against on every action.
|
||||
joint_limits: dict[str, tuple[float, float]] = field(
|
||||
default_factory=lambda: {
|
||||
"shoulder_pan": (-145.0, 145.0),
|
||||
"shoulder_lift": (-170.0, 1.0),
|
||||
"elbow_flex": (-200.0, 1.0),
|
||||
"wrist_flex": (-80.0, 90.0),
|
||||
"wrist_yaw": (-90.0, 90.0),
|
||||
"wrist_roll": (-90.0, 90.0),
|
||||
"gripper": (-270.0, 0.0),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@RobotConfig.register_subclass("rebot_b601_follower")
|
||||
@dataclass
|
||||
class RebotB601FollowerRobotConfig(RobotConfig, RebotB601FollowerConfig):
|
||||
"""Registered configuration for the reBot B601-DM follower robot."""
|
||||
|
||||
pass
|
||||
@@ -0,0 +1,289 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import logging
|
||||
import math
|
||||
import time
|
||||
from functools import cached_property
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from lerobot.cameras import make_cameras_from_configs
|
||||
from lerobot.motors import MotorCalibration
|
||||
from lerobot.types import RobotAction, RobotObservation
|
||||
from lerobot.utils.decorators import check_if_already_connected, check_if_not_connected
|
||||
from lerobot.utils.import_utils import _motorbridge_available, require_package
|
||||
|
||||
from ..robot import Robot
|
||||
from ..utils import ensure_safe_goal_position
|
||||
from .config_rebot_b601_follower import RebotB601FollowerRobotConfig
|
||||
|
||||
if TYPE_CHECKING or _motorbridge_available:
|
||||
from motorbridge import Controller as MotorBridgeController, Mode as MotorBridgeMode
|
||||
else:
|
||||
MotorBridgeController = None
|
||||
MotorBridgeMode = None
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Joint controlled in FORCE_POS mode; every other joint runs in POS_VEL mode.
|
||||
GRIPPER_MOTOR = "gripper"
|
||||
# Per-joint Damiao motor models for the B601-DM (passed to motorbridge).
|
||||
MOTOR_MODELS = {
|
||||
"shoulder_pan": "4340P",
|
||||
"shoulder_lift": "4340P",
|
||||
"elbow_flex": "4340P",
|
||||
"wrist_flex": "4310",
|
||||
"wrist_yaw": "4310",
|
||||
"wrist_roll": "4310",
|
||||
"gripper": "4310",
|
||||
}
|
||||
_ENSURE_MODE_RETRIES = 9
|
||||
_SETTLE_SEC = 0.01
|
||||
_ZERO_SETTLE_SEC = 0.1
|
||||
|
||||
|
||||
class RebotB601Follower(Robot):
|
||||
"""Seeed Studio reBot B601-DM follower arm (6-DOF + gripper, Damiao CAN motors).
|
||||
|
||||
Motor communication is handled by the ``motorbridge`` package over a CAN bus,
|
||||
reached either through a Damiao serial bridge or a SocketCAN adapter.
|
||||
"""
|
||||
|
||||
config_class = RebotB601FollowerRobotConfig
|
||||
name = "rebot_b601_follower"
|
||||
|
||||
def __init__(self, config: RebotB601FollowerRobotConfig):
|
||||
require_package("motorbridge", extra="rebot")
|
||||
super().__init__(config)
|
||||
self.config = config
|
||||
self.bus: MotorBridgeController | None = None
|
||||
self.motors: dict = {}
|
||||
self.motor_names = list(config.motor_can_ids.keys())
|
||||
self.cameras = make_cameras_from_configs(config.cameras)
|
||||
|
||||
@property
|
||||
def _motors_ft(self) -> dict[str, type]:
|
||||
return {f"{motor}.pos": float for motor in self.motor_names}
|
||||
|
||||
@property
|
||||
def _cameras_ft(self) -> dict[str, tuple]:
|
||||
return {
|
||||
cam: (self.config.cameras[cam].height, self.config.cameras[cam].width, 3) for cam in self.cameras
|
||||
}
|
||||
|
||||
@cached_property
|
||||
def observation_features(self) -> dict[str, type | tuple]:
|
||||
return {**self._motors_ft, **self._cameras_ft}
|
||||
|
||||
@cached_property
|
||||
def action_features(self) -> dict[str, type]:
|
||||
return self._motors_ft
|
||||
|
||||
@property
|
||||
def is_connected(self) -> bool:
|
||||
return self.bus is not None and all(cam.is_connected for cam in self.cameras.values())
|
||||
|
||||
@check_if_already_connected
|
||||
def connect(self, calibrate: bool = True) -> None:
|
||||
logger.info(f"Connecting {self} on {self.config.port} (adapter={self.config.can_adapter})...")
|
||||
if self.config.can_adapter == "damiao":
|
||||
self.bus = MotorBridgeController.from_dm_serial(
|
||||
serial_port=self.config.port,
|
||||
baud=self.config.dm_serial_baud,
|
||||
)
|
||||
elif self.config.can_adapter == "socketcan":
|
||||
self.bus = MotorBridgeController(channel=self.config.port)
|
||||
else:
|
||||
raise ValueError(
|
||||
f"Unsupported can_adapter '{self.config.can_adapter}'. Use 'damiao' or 'socketcan'."
|
||||
)
|
||||
|
||||
for motor_name, (send_id, recv_id) in self.config.motor_can_ids.items():
|
||||
self.motors[motor_name] = self.bus.add_damiao_motor(send_id, recv_id, MOTOR_MODELS[motor_name])
|
||||
|
||||
if not self.is_calibrated and calibrate:
|
||||
logger.info(
|
||||
"Mismatch between calibration values in the motor and the calibration file or no calibration file found"
|
||||
)
|
||||
self.calibrate()
|
||||
|
||||
for cam in self.cameras.values():
|
||||
cam.connect()
|
||||
|
||||
self.configure()
|
||||
logger.info(f"{self} connected.")
|
||||
|
||||
@property
|
||||
def is_calibrated(self) -> bool:
|
||||
return bool(self.calibration)
|
||||
|
||||
def calibrate(self) -> None:
|
||||
if self.calibration:
|
||||
user_input = input(
|
||||
f"Press ENTER to use provided calibration file associated with the id {self.id}, "
|
||||
"or type 'c' and press ENTER to run calibration: "
|
||||
)
|
||||
if user_input.strip().lower() != "c":
|
||||
logger.info(f"Using calibration file associated with the id {self.id}")
|
||||
return
|
||||
|
||||
logger.info(f"\nRunning calibration of {self}")
|
||||
self.bus.disable_all()
|
||||
print(
|
||||
"\nCalibration: set zero position.\n"
|
||||
"Manually move the reBot B601 to its ZERO POSITION and close the gripper.\n"
|
||||
"See the B601 manual for the zero pose (the default sit-down position).\n"
|
||||
)
|
||||
input("Press ENTER when ready...")
|
||||
|
||||
for motor in self.motors.values():
|
||||
motor.set_zero_position()
|
||||
time.sleep(_ZERO_SETTLE_SEC)
|
||||
logger.info("Arm zero position set.")
|
||||
|
||||
self.calibration = {}
|
||||
for motor_name, (send_id, _recv_id) in self.config.motor_can_ids.items():
|
||||
range_min, range_max = self.config.joint_limits[motor_name]
|
||||
self.calibration[motor_name] = MotorCalibration(
|
||||
id=send_id,
|
||||
drive_mode=0,
|
||||
homing_offset=0,
|
||||
range_min=int(range_min),
|
||||
range_max=int(range_max),
|
||||
)
|
||||
|
||||
self._save_calibration()
|
||||
print(f"Calibration saved to {self.calibration_fpath}")
|
||||
|
||||
def configure(self) -> None:
|
||||
self.bus.enable_all()
|
||||
for motor_name, motor in self.motors.items():
|
||||
target_mode = (
|
||||
MotorBridgeMode.FORCE_POS if motor_name == GRIPPER_MOTOR else MotorBridgeMode.POS_VEL
|
||||
)
|
||||
for attempt in range(_ENSURE_MODE_RETRIES + 1):
|
||||
try:
|
||||
motor.ensure_mode(target_mode)
|
||||
break
|
||||
except Exception:
|
||||
if attempt == _ENSURE_MODE_RETRIES:
|
||||
raise
|
||||
time.sleep(_SETTLE_SEC)
|
||||
logger.debug(f"{motor_name} mode set to {target_mode}")
|
||||
|
||||
@check_if_not_connected
|
||||
def disable_torque(self) -> None:
|
||||
"""Disable motor torque so the arm can be moved by hand (read-only debugging)."""
|
||||
self.bus.disable_all()
|
||||
logger.info(f"{self} torque disabled.")
|
||||
|
||||
def _present_pos(self) -> dict[str, float]:
|
||||
"""Read present joint positions in degrees."""
|
||||
for motor in self.motors.values():
|
||||
motor.request_feedback()
|
||||
try:
|
||||
self.bus.poll_feedback_once()
|
||||
except Exception:
|
||||
logger.warning("CAN bus poll feedback failed.")
|
||||
|
||||
present_pos = {}
|
||||
for motor_name, motor in self.motors.items():
|
||||
state = motor.get_state()
|
||||
present_pos[motor_name] = math.degrees(state.pos) if state is not None else 0.0
|
||||
return present_pos
|
||||
|
||||
@check_if_not_connected
|
||||
def get_observation(self) -> RobotObservation:
|
||||
start = time.perf_counter()
|
||||
obs_dict = {f"{motor}.pos": pos for motor, pos in self._present_pos().items()}
|
||||
dt_ms = (time.perf_counter() - start) * 1e3
|
||||
logger.debug(f"{self} read state: {dt_ms:.1f}ms")
|
||||
|
||||
for cam_key, cam in self.cameras.items():
|
||||
start = time.perf_counter()
|
||||
obs_dict[cam_key] = cam.read_latest()
|
||||
dt_ms = (time.perf_counter() - start) * 1e3
|
||||
logger.debug(f"{self} read {cam_key}: {dt_ms:.1f}ms")
|
||||
|
||||
return obs_dict
|
||||
|
||||
@check_if_not_connected
|
||||
def send_action(self, action: RobotAction) -> RobotAction:
|
||||
"""Command the arm to a target joint configuration.
|
||||
|
||||
Positions are expressed in degrees. The relative action magnitude may be
|
||||
clipped depending on `max_relative_target`, so the action actually sent is
|
||||
always returned.
|
||||
"""
|
||||
goal_pos = {key.removesuffix(".pos"): val for key, val in action.items() if key.endswith(".pos")}
|
||||
|
||||
# Clip against soft joint limits.
|
||||
for motor_name in list(goal_pos):
|
||||
if motor_name in self.config.joint_limits:
|
||||
min_limit, max_limit = self.config.joint_limits[motor_name]
|
||||
clipped = max(min_limit, min(max_limit, goal_pos[motor_name]))
|
||||
if clipped != goal_pos[motor_name]:
|
||||
logger.debug(f"Clipped {motor_name} from {goal_pos[motor_name]:.2f} to {clipped:.2f}")
|
||||
goal_pos[motor_name] = clipped
|
||||
|
||||
# Tolerate 6-DOF leaders that have no wrist_yaw joint by holding it at zero.
|
||||
# This is intentional: it lets a 6-DOF leader such as the SO-100 / SO-101
|
||||
# (so100_leader / so101_leader) teleoperate this 7-DOF follower — the missing
|
||||
# wrist_yaw command is simply treated as 0.0 instead of raising.
|
||||
if "wrist_yaw" not in goal_pos:
|
||||
goal_pos["wrist_yaw"] = 0.0
|
||||
|
||||
# Cap relative target when too far from the present position.
|
||||
if self.config.max_relative_target is not None:
|
||||
present_pos = self._present_pos()
|
||||
goal_present_pos = {key: (g, present_pos.get(key, g)) for key, g in goal_pos.items()}
|
||||
goal_pos = ensure_safe_goal_position(goal_present_pos, self.config.max_relative_target)
|
||||
|
||||
for motor_name, position_deg in goal_pos.items():
|
||||
motor = self.motors.get(motor_name)
|
||||
if motor is None:
|
||||
continue
|
||||
idx = self.motor_names.index(motor_name)
|
||||
vel_deg_s = (
|
||||
self.config.pos_vel_velocity[idx]
|
||||
if isinstance(self.config.pos_vel_velocity, list)
|
||||
else self.config.pos_vel_velocity
|
||||
)
|
||||
pos_rad = math.radians(position_deg)
|
||||
vel_rad = math.radians(vel_deg_s)
|
||||
if motor_name == GRIPPER_MOTOR:
|
||||
motor.send_force_pos(pos_rad, vel_rad, self.config.gripper_torque_ratio)
|
||||
else:
|
||||
motor.send_pos_vel(pos_rad, vel_rad)
|
||||
|
||||
return {f"{motor}.pos": val for motor, val in goal_pos.items()}
|
||||
|
||||
@check_if_not_connected
|
||||
def disconnect(self) -> None:
|
||||
for motor in self.motors.values():
|
||||
if self.config.disable_torque_on_disconnect:
|
||||
motor.disable()
|
||||
motor.clear_error()
|
||||
motor.close()
|
||||
|
||||
self.bus.close()
|
||||
self.bus = None
|
||||
self.motors = {}
|
||||
|
||||
for cam in self.cameras.values():
|
||||
cam.disconnect()
|
||||
|
||||
logger.info(f"{self} disconnected.")
|
||||
@@ -68,6 +68,14 @@ def make_robot_from_config(config: RobotConfig) -> Robot:
|
||||
from .bi_openarm_follower import BiOpenArmFollower
|
||||
|
||||
return BiOpenArmFollower(config)
|
||||
elif config.type == "rebot_b601_follower":
|
||||
from .rebot_b601_follower import RebotB601Follower
|
||||
|
||||
return RebotB601Follower(config)
|
||||
elif config.type == "bi_rebot_b601_follower":
|
||||
from .bi_rebot_b601_follower import BiRebotB601Follower
|
||||
|
||||
return BiRebotB601Follower(config)
|
||||
elif config.type == "mock_robot":
|
||||
from tests.mocks.mock_robot import MockRobot
|
||||
|
||||
|
||||
@@ -39,6 +39,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
Robot,
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
hope_jr,
|
||||
koch_follower,
|
||||
@@ -46,12 +47,14 @@ from lerobot.robots import ( # noqa: F401
|
||||
make_robot_from_config,
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
)
|
||||
from lerobot.teleoperators import ( # noqa: F401
|
||||
Teleoperator,
|
||||
TeleoperatorConfig,
|
||||
bi_openarm_leader,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
homunculus,
|
||||
koch_leader,
|
||||
@@ -59,6 +62,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
omx_leader,
|
||||
openarm_leader,
|
||||
openarm_mini,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
unitree_g1,
|
||||
)
|
||||
|
||||
@@ -45,16 +45,19 @@ from lerobot.model import RobotKinematics
|
||||
from lerobot.robots import ( # noqa: F401
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
koch_follower,
|
||||
make_robot_from_config,
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
)
|
||||
from lerobot.teleoperators import ( # noqa: F401
|
||||
TeleoperatorConfig,
|
||||
bi_openarm_leader,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
gamepad,
|
||||
koch_leader,
|
||||
@@ -62,6 +65,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
omx_leader,
|
||||
openarm_leader,
|
||||
openarm_mini,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
)
|
||||
from lerobot.utils.robot_utils import precise_sleep
|
||||
|
||||
@@ -120,6 +120,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
Robot,
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
earthrover_mini_plus,
|
||||
hope_jr,
|
||||
@@ -128,6 +129,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
reachy2,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
unitree_g1 as unitree_g1_robot,
|
||||
)
|
||||
@@ -135,6 +137,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
Teleoperator,
|
||||
TeleoperatorConfig,
|
||||
bi_openarm_leader,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
homunculus,
|
||||
koch_leader,
|
||||
@@ -143,6 +146,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
openarm_leader,
|
||||
openarm_mini,
|
||||
reachy2_teleoperator,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
unitree_g1,
|
||||
)
|
||||
|
||||
@@ -56,6 +56,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
Robot,
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
earthrover_mini_plus,
|
||||
hope_jr,
|
||||
@@ -64,6 +65,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
reachy2,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
unitree_g1,
|
||||
)
|
||||
|
||||
@@ -144,6 +144,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
Robot,
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
earthrover_mini_plus,
|
||||
hope_jr,
|
||||
@@ -151,6 +152,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
reachy2,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
unitree_g1 as unitree_g1_robot,
|
||||
)
|
||||
@@ -159,6 +161,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
Teleoperator,
|
||||
TeleoperatorConfig,
|
||||
bi_openarm_leader,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
homunculus,
|
||||
koch_leader,
|
||||
@@ -166,6 +169,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
openarm_leader,
|
||||
openarm_mini,
|
||||
reachy2_teleoperator,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
unitree_g1,
|
||||
)
|
||||
|
||||
@@ -30,20 +30,24 @@ import draccus
|
||||
|
||||
from lerobot.robots import ( # noqa: F401
|
||||
RobotConfig,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
koch_follower,
|
||||
lekiwi,
|
||||
make_robot_from_config,
|
||||
omx_follower,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
)
|
||||
from lerobot.teleoperators import ( # noqa: F401
|
||||
TeleoperatorConfig,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
koch_leader,
|
||||
make_teleoperator_from_config,
|
||||
omx_leader,
|
||||
openarm_mini,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
)
|
||||
|
||||
|
||||
@@ -72,6 +72,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
Robot,
|
||||
RobotConfig,
|
||||
bi_openarm_follower,
|
||||
bi_rebot_b601_follower,
|
||||
bi_so_follower,
|
||||
earthrover_mini_plus,
|
||||
hope_jr,
|
||||
@@ -80,6 +81,7 @@ from lerobot.robots import ( # noqa: F401
|
||||
omx_follower,
|
||||
openarm_follower,
|
||||
reachy2,
|
||||
rebot_b601_follower,
|
||||
so_follower,
|
||||
unitree_g1 as unitree_g1_robot,
|
||||
)
|
||||
@@ -87,6 +89,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
Teleoperator,
|
||||
TeleoperatorConfig,
|
||||
bi_openarm_leader,
|
||||
bi_rebot_102_leader,
|
||||
bi_so_leader,
|
||||
gamepad,
|
||||
homunculus,
|
||||
@@ -97,6 +100,7 @@ from lerobot.teleoperators import ( # noqa: F401
|
||||
openarm_leader,
|
||||
openarm_mini,
|
||||
reachy2_teleoperator,
|
||||
rebot_102_leader,
|
||||
so_leader,
|
||||
unitree_g1,
|
||||
)
|
||||
|
||||
@@ -0,0 +1,20 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from .bi_rebot_102_leader import BiRebotArm102Leader
|
||||
from .config_bi_rebot_102_leader import BiRebotArm102LeaderConfig
|
||||
|
||||
__all__ = ["BiRebotArm102Leader", "BiRebotArm102LeaderConfig"]
|
||||
@@ -0,0 +1,113 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import logging
|
||||
from functools import cached_property
|
||||
|
||||
from lerobot.types import RobotAction
|
||||
from lerobot.utils.decorators import check_if_already_connected, check_if_not_connected
|
||||
|
||||
from ..rebot_102_leader import RebotArm102Leader, RebotArm102LeaderTeleopConfig
|
||||
from ..teleoperator import Teleoperator
|
||||
from .config_bi_rebot_102_leader import BiRebotArm102LeaderConfig
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class BiRebotArm102Leader(Teleoperator):
|
||||
"""Bimanual Seeed Studio StarArm102 / reBot Arm 102 leader.
|
||||
|
||||
Composes two single-arm :class:`RebotArm102Leader` instances. Action keys of
|
||||
each arm are namespaced with a ``left_`` / ``right_`` prefix, so a bimanual
|
||||
leader can teleoperate a bimanual reBot B601 follower.
|
||||
"""
|
||||
|
||||
config_class = BiRebotArm102LeaderConfig
|
||||
name = "bi_rebot_102_leader"
|
||||
|
||||
def __init__(self, config: BiRebotArm102LeaderConfig):
|
||||
super().__init__(config)
|
||||
self.config = config
|
||||
|
||||
left_arm_config = RebotArm102LeaderTeleopConfig(
|
||||
id=f"{config.id}_left" if config.id else None,
|
||||
calibration_dir=config.calibration_dir,
|
||||
port=config.left_arm_config.port,
|
||||
baudrate=config.left_arm_config.baudrate,
|
||||
joint_ids=config.left_arm_config.joint_ids,
|
||||
joint_directions=config.left_arm_config.joint_directions,
|
||||
joint_ranges=config.left_arm_config.joint_ranges,
|
||||
)
|
||||
|
||||
right_arm_config = RebotArm102LeaderTeleopConfig(
|
||||
id=f"{config.id}_right" if config.id else None,
|
||||
calibration_dir=config.calibration_dir,
|
||||
port=config.right_arm_config.port,
|
||||
baudrate=config.right_arm_config.baudrate,
|
||||
joint_ids=config.right_arm_config.joint_ids,
|
||||
joint_directions=config.right_arm_config.joint_directions,
|
||||
joint_ranges=config.right_arm_config.joint_ranges,
|
||||
)
|
||||
|
||||
self.left_arm = RebotArm102Leader(left_arm_config)
|
||||
self.right_arm = RebotArm102Leader(right_arm_config)
|
||||
|
||||
@cached_property
|
||||
def action_features(self) -> dict[str, type]:
|
||||
return {
|
||||
**{f"left_{k}": v for k, v in self.left_arm.action_features.items()},
|
||||
**{f"right_{k}": v for k, v in self.right_arm.action_features.items()},
|
||||
}
|
||||
|
||||
@cached_property
|
||||
def feedback_features(self) -> dict[str, type]:
|
||||
return {}
|
||||
|
||||
@property
|
||||
def is_connected(self) -> bool:
|
||||
return self.left_arm.is_connected and self.right_arm.is_connected
|
||||
|
||||
@check_if_already_connected
|
||||
def connect(self, calibrate: bool = True) -> None:
|
||||
self.left_arm.connect(calibrate)
|
||||
self.right_arm.connect(calibrate)
|
||||
|
||||
@property
|
||||
def is_calibrated(self) -> bool:
|
||||
return self.left_arm.is_calibrated and self.right_arm.is_calibrated
|
||||
|
||||
def calibrate(self) -> None:
|
||||
self.left_arm.calibrate()
|
||||
self.right_arm.calibrate()
|
||||
|
||||
def configure(self) -> None:
|
||||
self.left_arm.configure()
|
||||
self.right_arm.configure()
|
||||
|
||||
@check_if_not_connected
|
||||
def get_action(self) -> RobotAction:
|
||||
action_dict = {}
|
||||
action_dict.update({f"left_{k}": v for k, v in self.left_arm.get_action().items()})
|
||||
action_dict.update({f"right_{k}": v for k, v in self.right_arm.get_action().items()})
|
||||
return action_dict
|
||||
|
||||
def send_feedback(self, feedback: dict[str, float]) -> None:
|
||||
raise NotImplementedError("Feedback is not implemented for the reBot Arm 102 leader.")
|
||||
|
||||
@check_if_not_connected
|
||||
def disconnect(self) -> None:
|
||||
self.left_arm.disconnect()
|
||||
self.right_arm.disconnect()
|
||||
@@ -0,0 +1,29 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
from ..config import TeleoperatorConfig
|
||||
from ..rebot_102_leader import RebotArm102LeaderConfig
|
||||
|
||||
|
||||
@TeleoperatorConfig.register_subclass("bi_rebot_102_leader")
|
||||
@dataclass
|
||||
class BiRebotArm102LeaderConfig(TeleoperatorConfig):
|
||||
"""Configuration class for the bimanual reBot Arm 102 leader teleoperator."""
|
||||
|
||||
left_arm_config: RebotArm102LeaderConfig
|
||||
right_arm_config: RebotArm102LeaderConfig
|
||||
@@ -0,0 +1,20 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from .config_rebot_102_leader import RebotArm102LeaderConfig, RebotArm102LeaderTeleopConfig
|
||||
from .rebot_102_leader import RebotArm102Leader
|
||||
|
||||
__all__ = ["RebotArm102Leader", "RebotArm102LeaderConfig", "RebotArm102LeaderTeleopConfig"]
|
||||
@@ -0,0 +1,83 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
from ..config import TeleoperatorConfig
|
||||
|
||||
|
||||
@dataclass
|
||||
class RebotArm102LeaderConfig:
|
||||
"""Base configuration class for the Seeed Studio StarArm102 / reBot Arm 102 leader.
|
||||
|
||||
The reBot Arm 102 is a 7-joint (incl. gripper) leader arm driven by FashionStar
|
||||
UART smart servos. Servo communication goes through ``motorbridge-smart-servo``.
|
||||
"""
|
||||
|
||||
# USB-to-UART device the leader arm is connected to (e.g. "/dev/ttyUSB0").
|
||||
port: str
|
||||
|
||||
baudrate: int = 1_000_000
|
||||
|
||||
# Servo id of each joint on the UART bus.
|
||||
joint_ids: dict[str, int] = field(
|
||||
default_factory=lambda: {
|
||||
"shoulder_pan": 0,
|
||||
"shoulder_lift": 1,
|
||||
"elbow_flex": 2,
|
||||
"wrist_flex": 3,
|
||||
"wrist_yaw": 4,
|
||||
"wrist_roll": 5,
|
||||
"gripper": 6,
|
||||
}
|
||||
)
|
||||
|
||||
# Per-joint sign applied to raw servo angles so the leader matches the follower
|
||||
# convention. The gripper additionally carries a scale (e.g. -6) to widen its
|
||||
# range to the reBot B601 follower's gripper travel.
|
||||
joint_directions: dict[str, int] = field(
|
||||
default_factory=lambda: {
|
||||
"shoulder_pan": -1,
|
||||
"shoulder_lift": -1,
|
||||
"elbow_flex": 1,
|
||||
"wrist_flex": 1,
|
||||
"wrist_yaw": 1,
|
||||
"wrist_roll": -1,
|
||||
"gripper": -6,
|
||||
}
|
||||
)
|
||||
|
||||
# Per-joint [min, max] output range in degrees. Matches the reBot B601 follower
|
||||
# joint limits so leader actions can drive the follower key-for-key.
|
||||
joint_ranges: dict[str, list[int]] = field(
|
||||
default_factory=lambda: {
|
||||
"shoulder_pan": [-150, 150],
|
||||
"shoulder_lift": [-170, 1],
|
||||
"elbow_flex": [-200, 1],
|
||||
"wrist_flex": [-80, 90],
|
||||
"wrist_yaw": [-90, 90],
|
||||
"wrist_roll": [-90, 90],
|
||||
"gripper": [-270, 0],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@TeleoperatorConfig.register_subclass("rebot_102_leader")
|
||||
@dataclass
|
||||
class RebotArm102LeaderTeleopConfig(TeleoperatorConfig, RebotArm102LeaderConfig):
|
||||
"""Registered configuration for the reBot Arm 102 leader teleoperator."""
|
||||
|
||||
pass
|
||||
@@ -0,0 +1,207 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Copyright 2026 The HuggingFace Inc. team. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import logging
|
||||
import time
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from lerobot.motors import MotorCalibration
|
||||
from lerobot.types import RobotAction
|
||||
from lerobot.utils.decorators import check_if_already_connected, check_if_not_connected
|
||||
from lerobot.utils.import_utils import _motorbridge_smart_servo_available, require_package
|
||||
|
||||
from ..teleoperator import Teleoperator
|
||||
from .config_rebot_102_leader import RebotArm102LeaderTeleopConfig
|
||||
|
||||
if TYPE_CHECKING or _motorbridge_smart_servo_available:
|
||||
from motorbridge_smart_servo import FashionStarServo, ServoMonitor
|
||||
else:
|
||||
FashionStarServo = None
|
||||
ServoMonitor = None
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
_SETTLE_SEC = 0.01
|
||||
|
||||
|
||||
class RebotArm102Leader(Teleoperator):
|
||||
"""Seeed Studio StarArm102 / reBot Arm 102 leader arm.
|
||||
|
||||
A 7-joint (incl. gripper) leader built on FashionStar UART smart servos. Servo
|
||||
communication is handled by the ``motorbridge-smart-servo`` package; this class
|
||||
only reads joint angles, so it produces actions but accepts no feedback.
|
||||
"""
|
||||
|
||||
config_class = RebotArm102LeaderTeleopConfig
|
||||
name = "rebot_102_leader"
|
||||
|
||||
def __init__(self, config: RebotArm102LeaderTeleopConfig):
|
||||
require_package("motorbridge-smart-servo", extra="rebot", import_name="motorbridge_smart_servo")
|
||||
super().__init__(config)
|
||||
self.config = config
|
||||
self.bus: FashionStarServo | None = None
|
||||
self.motor_names = list(config.joint_ids.keys())
|
||||
self._last_raw_positions: dict[str, float] = {}
|
||||
|
||||
@property
|
||||
def action_features(self) -> dict[str, type]:
|
||||
return {f"{motor}.pos": float for motor in self.motor_names}
|
||||
|
||||
@property
|
||||
def feedback_features(self) -> dict[str, type]:
|
||||
return {}
|
||||
|
||||
@property
|
||||
def is_connected(self) -> bool:
|
||||
return self.bus is not None
|
||||
|
||||
@check_if_already_connected
|
||||
def connect(self, calibrate: bool = True) -> None:
|
||||
logger.info(f"Connecting {self} on {self.config.port}...")
|
||||
bus = FashionStarServo(self.config.port, baudrate=self.config.baudrate)
|
||||
try:
|
||||
for motor_name, motor_id in self.config.joint_ids.items():
|
||||
if not bus.ping(motor_id):
|
||||
raise RuntimeError(f"Servo not found for {motor_name} (id={motor_id}).")
|
||||
self._last_raw_positions[motor_name] = 0.0
|
||||
self.bus = bus
|
||||
|
||||
if not self.is_calibrated and calibrate:
|
||||
logger.info(
|
||||
"Mismatch between calibration values in the motor and the calibration file or no calibration file found"
|
||||
)
|
||||
self.calibrate()
|
||||
|
||||
self.configure()
|
||||
except Exception:
|
||||
bus.close()
|
||||
self.bus = None
|
||||
raise
|
||||
|
||||
logger.info(f"{self} connected.")
|
||||
|
||||
@property
|
||||
def is_calibrated(self) -> bool:
|
||||
return bool(self.calibration) and set(self.calibration) == set(self.motor_names)
|
||||
|
||||
def calibrate(self) -> None:
|
||||
if self.calibration:
|
||||
user_input = input(
|
||||
f"Press ENTER to use provided calibration file associated with the id {self.id}, "
|
||||
"or type 'c' and press ENTER to run calibration: "
|
||||
)
|
||||
if user_input.strip().lower() != "c":
|
||||
logger.info(f"Using calibration file associated with the id {self.id}")
|
||||
return
|
||||
|
||||
logger.info(f"\nRunning calibration of {self}")
|
||||
input(
|
||||
"\nCalibration: set zero position.\n"
|
||||
"Manually move the reBot Arm 102 to its zero pose and close the gripper.\n"
|
||||
"Press ENTER when ready..."
|
||||
)
|
||||
|
||||
self.calibration = {}
|
||||
for motor_name, motor_id in self.config.joint_ids.items():
|
||||
self.bus.unlock(motor_id)
|
||||
time.sleep(_SETTLE_SEC)
|
||||
self.bus.set_origin_point(motor_id)
|
||||
range_min, range_max = self.config.joint_ranges[motor_name]
|
||||
self.calibration[motor_name] = MotorCalibration(
|
||||
id=motor_id,
|
||||
drive_mode=0,
|
||||
homing_offset=0,
|
||||
range_min=int(range_min),
|
||||
range_max=int(range_max),
|
||||
)
|
||||
|
||||
self._save_calibration()
|
||||
logger.info(f"Calibration saved to {self.calibration_fpath}")
|
||||
|
||||
def configure(self) -> None:
|
||||
for motor_id in self.config.joint_ids.values():
|
||||
self.bus.unlock(motor_id)
|
||||
time.sleep(_SETTLE_SEC)
|
||||
# Reset the multi-turn counter of each servo individually.
|
||||
for motor_id in self.config.joint_ids.values():
|
||||
self.bus.reset_multi_turn(motor_id)
|
||||
|
||||
def _read_raw_positions(self) -> dict[str, float]:
|
||||
result: dict[int, ServoMonitor | None] = self.bus.sync_monitor(list(self.config.joint_ids.values()))
|
||||
id_to_name = {v: k for k, v in self.config.joint_ids.items()}
|
||||
raw_positions: dict[str, float] = {}
|
||||
for motor_id, monitor in result.items():
|
||||
motor_name = id_to_name[motor_id]
|
||||
if monitor is None:
|
||||
raise RuntimeError(f"Servo {motor_name} (id={motor_id}) has never responded.")
|
||||
raw_positions[motor_name] = monitor.angle_deg
|
||||
return raw_positions
|
||||
|
||||
@staticmethod
|
||||
def _round_to_valid_range(value: float, min_value: float, max_value: float) -> tuple[float, int]:
|
||||
"""Unwrap a multi-turn angle into the ±180° window centred on (min+max)/2.
|
||||
|
||||
The servo may report an angle that has accumulated extra full rotations
|
||||
(value = true_angle + N*360). Subtract the nearest whole number of turns
|
||||
to bring it back into [center-180, center+180]. Returns the unwrapped
|
||||
angle and the number of turns removed.
|
||||
"""
|
||||
center = (min_value + max_value) / 2.0
|
||||
turns = round((value - center) / 360.0)
|
||||
return value - turns * 360.0, abs(turns)
|
||||
|
||||
@check_if_not_connected
|
||||
def get_action(self) -> RobotAction:
|
||||
start = time.perf_counter()
|
||||
try:
|
||||
raw_positions = self._read_raw_positions()
|
||||
self._last_raw_positions = raw_positions
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to read raw positions: {e}")
|
||||
logger.warning("[EMERGENCY STOP] Hold the follower arm and cut off the main power to the arms.")
|
||||
logger.warning(
|
||||
"[EMERGENCY STOP] Break the teleoperation session and check the leader USB connection or power."
|
||||
)
|
||||
raw_positions = self._last_raw_positions
|
||||
|
||||
action_dict: dict[str, float] = {}
|
||||
for motor_name in self.motor_names:
|
||||
range_min, range_max = self.config.joint_ranges[motor_name]
|
||||
direction = self.config.joint_directions[motor_name]
|
||||
sign = 1.0 if direction >= 0 else -1.0
|
||||
unwrapped, k = self._round_to_valid_range(
|
||||
raw_positions[motor_name], range_min * sign, range_max * sign
|
||||
)
|
||||
position = unwrapped * direction
|
||||
if k > 0:
|
||||
logger.debug(
|
||||
f"Servo {motor_name} (id={self.config.joint_ids[motor_name]}) wrapped {k} * 360°. "
|
||||
f"Unwrapped pos: {unwrapped:.1f}° (raw: {raw_positions[motor_name]:.1f}°)"
|
||||
)
|
||||
action_dict[f"{motor_name}.pos"] = max(float(range_min), min(float(range_max), position))
|
||||
|
||||
dt_ms = (time.perf_counter() - start) * 1e3
|
||||
logger.debug(f"{self} read action: {dt_ms:.1f}ms")
|
||||
return action_dict
|
||||
|
||||
def send_feedback(self, feedback: dict[str, float]) -> None:
|
||||
raise NotImplementedError("Feedback is not implemented for the reBot Arm 102 leader.")
|
||||
|
||||
@check_if_not_connected
|
||||
def disconnect(self) -> None:
|
||||
self.bus.close()
|
||||
self.bus = None
|
||||
logger.info(f"{self} disconnected.")
|
||||
@@ -99,6 +99,14 @@ def make_teleoperator_from_config(config: TeleoperatorConfig) -> "Teleoperator":
|
||||
from .openarm_mini import OpenArmMini
|
||||
|
||||
return OpenArmMini(config)
|
||||
elif config.type == "rebot_102_leader":
|
||||
from .rebot_102_leader import RebotArm102Leader
|
||||
|
||||
return RebotArm102Leader(config)
|
||||
elif config.type == "bi_rebot_102_leader":
|
||||
from .bi_rebot_102_leader import BiRebotArm102Leader
|
||||
|
||||
return BiRebotArm102Leader(config)
|
||||
else:
|
||||
try:
|
||||
return cast("Teleoperator", make_device_from_device_class(config))
|
||||
|
||||
@@ -114,6 +114,10 @@ _dynamixel_sdk_available = is_package_available("dynamixel-sdk", import_name="dy
|
||||
_feetech_sdk_available = is_package_available("feetech-servo-sdk", import_name="scservo_sdk")
|
||||
_reachy2_sdk_available = is_package_available("reachy2_sdk")
|
||||
_can_available = is_package_available("python-can", "can")
|
||||
_motorbridge_available = is_package_available("motorbridge")
|
||||
_motorbridge_smart_servo_available = is_package_available(
|
||||
"motorbridge-smart-servo", import_name="motorbridge_smart_servo"
|
||||
)
|
||||
_unitree_sdk_available = is_package_available("unitree-sdk2py", "unitree_sdk2py")
|
||||
_pyrealsense2_available = is_package_available("pyrealsense2") or is_package_available(
|
||||
"pyrealsense2-macosx", import_name="pyrealsense2"
|
||||
|
||||
Reference in New Issue
Block a user