bimanual-lerobot
收藏DexBench Bimanual 数据集详情
数据集概述
DexBench Bimanual 是一个双臂灵巧操作演示数据集,通过 Isaac Lab 回放并使用 LeRobot v2.1 格式打包。每个帧包含第三人称、左手腕和右手腕的 RGB 视频(256×256 @ 30 fps)、本体感知数据以及对应的关节空间动作。
基本信息
| 项目 | 数值 |
|---|---|
| 片段数 | 746 |
| 总帧数 | 298,482 |
| 任务数 | 9 |
| 帧率 | 30 FPS |
| 摄像头分辨率 | 256 × 256 |
| 机器人 | 浮动基座双臂 Shadow Dexterous Hands |
| LeRobot 版本 | v2.1 |
| 每片段视觉随机化 | 禁用(所有片段使用固定 HDRI 背景和桌面纹理) |
特征结构
| 特征 | 数据类型 | 形状 |
|---|---|---|
observation.state |
float32 | (56,) |
observation.images.third_person |
video (h264) | (256, 256, 3) |
observation.images.left_wrist |
video (h264) | (256, 256, 3) |
observation.images.right_wrist |
video (h264) | (256, 256, 3) |
action |
float32 | (56,) |
任务列表
| 任务索引 | 任务名称 |
|---|---|
| 0 | Dexbench-FixateThenManipulate-LiftBasketHandle-v0 |
| 1 | Dexbench-FixateThenManipulate-OpenFlatFolder-v0 |
| 2 | Dexbench-FixateThenManipulate-OpenHuaweiPhone-v0 |
| 3 | Dexbench-FixateThenManipulate-OpenLaptop-v0 |
| 4 | Dexbench-FixateThenManipulate-OpenStapler-v0 |
| 5 | Dexbench-FixateThenManipulate-SlideUtilityKnife-v0 |
| 6 | Dexbench-FixateThenManipulate-SqueezeScissors-v0 |
| 7 | Dexbench-BimanualLiftBasket-v0 |
| 8 | Dexbench-BimanualLiftTray-v0 |
部分任务具有额外的任务条件变体(如 _lose_startup、_onthetable),这些变体的片段已合并到上述基础 task_index 下。
使用示例
python from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
ds = LeRobotDataset("dexbench/bimanual-lerobot") print(ds.num_episodes, ds.num_frames)
frame = ds[0]
frame["observation.state"], frame["action"]: torch.Tensor(56,)
frame["observation.images.third_person"],
frame["observation.images.left_wrist"],
frame["observation.images.right_wrist"]: torch.Tensor(3, 256, 256)
frame["task"]: str (one of the 9 task ids above)
配套数据集
- dexbench/single-lerobot — 单手子集(735 个片段,15 个任务,28 维动作,一个腕部摄像头)。
数据来源
来源于 dexbench/DexBench_dataset 仓库中的遥操作轨迹 pickle 文件(目录 teleop_dataset/{bimanual_articulation, grasping}/...),使用 DexBench 的 scripts/create_demo_files_sequential.py 脚本在 Isaac Sim 环境中回放生成(每个任务一个环境,片段在环境内串行回放,禁用每片段 HDRI/桌面纹理随机化以获得视觉稳定的演示)。然后使用 scripts/convert_to_lerobot.py 脚本转换(h264 视频,CRF 18)。
许可证
Apache-2.0。对于底层资产和遥操作数据,遵循上游 DexBench 的许可条款。




