Robosuite中WHOLE_BODY_IK控制器的正确使用方法

问题背景

在使用Robosuite机器人仿真平台时，开发者可能会遇到机器人末端执行器（End-Effector）在保持静止状态下意外旋转90度的问题。这种情况通常发生在使用WHOLE_BODY_IK（全身逆运动学）控制器时，由于坐标系转换或观测值选择不当导致。

关键问题分析

在Robosuite中，WHOLE_BODY_IK控制器需要精确的末端执行器位姿信息作为输入。常见的问题根源包括：

1. 观测值选择错误：直接使用robot0_eef_quat而非robot0_eef_quat_site，导致坐标系不匹配
2. 旋转表示方式不当：使用欧拉角（Euler Angles）而非轴角（Axis-Angle）表示旋转
3. 控制器配置不完整：未正确设置IK控制器的旋转表示方式参数

解决方案

1. 正确的观测值选择

在Robosuite中，末端执行器的姿态观测有两个来源：

• robot0_eef_quat：表示机器人身体(body)的旋转
• robot0_eef_quat_site：表示末端执行器站点(site)的旋转

对于WHOLE_BODY_IK控制器，必须使用robot0_eef_quat_site观测值，因为控制器直接控制的是站点而非身体。

2. 旋转表示转换

WHOLE_BODY_IK控制器默认使用轴角表示法（Axis-Angle）作为旋转输入。正确的转换方法是使用Robosuite提供的工具函数：

from robosuite.utils.transform_utils import quat2axisangle

eef_quat = obs["robot0_eef_quat_site"]
eef_axis_angle = quat2axisangle(eef_quat)

3. 控制器配置

确保WHOLE_BODY_IK控制器的配置文件中包含正确的旋转表示设置：

{
    "ik_input_rotation_repr": "axis_angle",
    "ik_input_type": "absolute"
}

完整实现示例

# 加载控制器配置
controller_config = load_composite_controller_config(controller="whole_body_ik.json")

# 创建环境
env = suite.make(
    env_name="Stack",
    robots="UR5e",
    controller_configs=controller_config,
    has_renderer=False,
    has_offscreen_renderer=True,
)

# 重置环境
obs = env.reset()

# 获取正确的末端执行器位姿
eef_pos = obs["robot0_eef_pos"]
eef_quat = obs["robot0_eef_quat_site"]  # 使用站点而非身体的旋转
eef_axis_angle = quat2axisangle(eef_quat)  # 转换为轴角表示
gripper_state = [1.0]  # 夹持器状态

# 执行动作
for i in range(50):
    action = np.concatenate([eef_pos, eef_axis_angle, gripper_state])
    obs, reward, done, info = env.step(action)

技术要点总结

1. 坐标系一致性：确保控制器目标坐标系与观测坐标系一致，WHOLE_BODY_IK控制器操作的是站点坐标系。

2. 旋转表示选择：轴角表示法相比欧拉角能避免万向节锁问题，更适合机器人控制。

3. 控制器参数验证：检查控制器配置中的ik_input_rotation_repr和ik_input_type参数是否设置正确。

4. 观测值理解：深入理解Robosuite提供的各种观测值含义，特别是带有"_site"后缀的观测值。

通过以上方法，开发者可以确保WHOLE_BODY_IK控制器正确工作，避免机器人末端执行器意外旋转的问题。