B1k_Recovery
收藏资源简介:
B1k_Recovery是一个用于机器人失败恢复和演示学习的数据集,基于BEHAVIOR-1K模拟环境,采用LeRobot v3.0格式组织。当前版本包含task5(设置捕鼠器),由官方人类演示和自动生成的导航、放置/恢复、成功示范片段混合而成。数据集总计5,795个episode,3,537,240帧,时长约32.752小时(30帧/秒)。其中约58%的帧来自官方人类演示,其余42%为生成数据。每个episode包含61维的观测状态(包括基座、双臂、末端执行器、夹爪和躯干的位姿与速度)和23维的R1Pro VLA动作指令。观测数据来自三个RGB摄像头(头部、左/右RealSense),分辨率224×224,H.264编码。数据集适用于训练机器人操作策略,特别是需要失败恢复能力的任务。使用时需注意:本地task_index(0)需映射到BEHAVIOR-1K全局任务索引5;基座线速度是基座坐标系下的,不应再旋转;推荐使用校正后的归一化统计量;基座通道存在强烈的静止先验偏差。
B1k_Recovery is a dataset for robot failure recovery and demonstration learning, based on the BEHAVIOR-1K simulation environment, organized in LeRobot v3.0 format. The current version includes task5 (setting a mousetrap), composed of official human demonstrations and automatically generated navigation, placement/recovery, and successful demonstration fragments. The dataset totals 5,795 episodes, 3,537,240 frames, with a duration of approximately 32.752 hours (30 fps). About 58% of the frames are from official human demonstrations, and the remaining 42% are generated data. Each episode contains 61-dimensional observation states (including base, dual arms, end-effectors, grippers, and torso poses and velocities) and 23-dimensional R1Pro VLA action commands. Observations come from three RGB cameras (head, left/right RealSense) with a resolution of 224×224, H.264 encoded. The dataset is suitable for training robot manipulation policies, especially for tasks requiring failure recovery capabilities.
B1k_Recovery 数据集详情
数据集概述
B1k_Recovery 是一个面向 BEHAVIOR-1K 机器人任务失败恢复与演示的数据集,采用 LeRobot v3.0 格式,按任务组织为独立子文件夹。当前包含 task5(setting_mousetraps),后续将添加更多任务作为同级文件夹。
数据构成与规模
| 来源类型 | 片段数 | 帧数 | 时长(小时,@30fps) |
|---|---|---|---|
| 官方人类演示 | 200 | 2,039,222 | 18.882 |
| 生成导航数据 | 1,170 | 562,590 | 5.209 |
| 生成成功演示 | 37 | 327,684 | 3.034 |
| 生成放置/恢复数据 | 4,388 | 607,744 | 5.627 |
| 总计 | 5,795 | 3,537,240 | 32.752 |
- 约 58% 的帧来自官方人类演示,42% 为生成数据
- 片段长度呈强双峰分布:5,102 个片段低于 500 帧(短恢复片段,约 25% 帧),237 个片段超过 3,000 帧(长演示,约 67% 帧)
- 采样按帧进行,均匀洗牌批次反映帧分布而非片段分布
技术规格
| 属性 | 值 |
|---|---|
| 格式 | LeRobot v3.0 |
| 机器人 | R1Pro |
| 任务 | setting_mousetraps,本地 task_index=0 |
| 观测状态 | float32[61] |
| 动作 | float32[23] |
| 频率 | 30 Hz |
| RGB 相机 | 3 路 224×224 H.264(yuv420p)流 |
| 磁盘大小 | 7.3 GB,共 23,192 个文件 |
| 截断后帧采样数(H=30) | 3,363,390 |
观测状态(61维,BEHAVIOR 2026布局)
| 切片 | 含义 |
|---|---|
0:3 |
base-frame 观测的 base_qvel = [vx, vy, wz] |
3:10, 10:17 |
左臂 qpos, qvel |
17:20, 20:24 |
左端执行器位置, 四元数 |
24:26, 26:28 |
左夹爪 qpos, qvel |
28:35, 35:42 |
右臂 qpos, qvel |
42:45, 45:49 |
右端执行器位置, 四元数 |
49:51, 51:53 |
右夹爪 qpos, qvel |
53:57, 57:61 |
躯干 qpos, qvel |
动作(23维 R1Pro VLA 动作)
text base[0:3], trunk[3:7], left_arm[7:14], left_gripper[14], right_arm[15:22], right_gripper[22]
相机
observation.rgb.zed_link_camera_0(头部)observation.rgb.left_realsense_link_camera_0observation.rgb.right_realsense_link_camera_0
训练注意事项
1. 任务索引重映射
meta/tasks.parquet 将 setting_mousetraps 映射为本地索引 0,但 BEHAVIOR-1K 全局索引为 5。若直接使用存储的 task_index,会导致阶段数为 5 而非正确的 12。应通过活动名称进行重映射。
2. base_qvel 为 base-frame 坐标系
经验验证:std = (0.0694, 0.0292, 0.0762),std_x/std_y = 2.37,77.7% 帧中 |vx| > |vy|。world-frame 数据应接近 1.0 且无前进偏差。评估器如有帧切换选项(如 B1K_BASE_QVEL_FRAME),应保持未设置/base,强制 world 会向 base-frame 策略输入旋转信号。
3. 使用 qvel 校正的归一化统计
若归一化统计中 state[0:3] 的 std 比自身 action[0:3] std 小约 10 倍,说明基于损坏的 qvel 通道计算。正确的缩放尺度为 std ≈ (0.109, 0.061, 0.133)。
4. 已知偏差:base 通道存在强烈静止先验
- 81.6% 帧
|state_vx| < 0.005,其中仅约 2.0% 命令|action_vx| > 0.05 corr(state[0:3], action[0:3]) = 0.88 / 0.79 / 0.68- 策略可能通过回声当前速度而非学习启动时机来获得低 base 速度损失,评估时表现为机器人原地转动但不前进
- 生成的恢复片段可减少此泄漏;人类专属子集更严重(corr ≈ 0.997,静止帧仅 0.05% 命令运动)
验证状态
发布前已全面验证:
- 全部 5,795 个 Parquet 文件和 3,537,240 行状态/动作数据已扫描
- 所有状态 61 维、动作 23 维、数值有限
- 片段/帧/全局索引、时间戳、任务映射及元数据总数一致
- 全部 17,385 个视频已解码并与片段帧数核对
- 各来源族精确通过源到输出检查
- 所有来源族样本均通过 LeRobot 0.5.2 加载,包含全部三个 RGB 张量
关联模型
基于 BEHAVIOR-1K checkpoint_1 在 task5 上微调的检查点发布在 fastwalker1118/b1kcheckpoints(task5ckptmerged15000 至 39999)。30,000 步、batch 128 约等于 1.14 个 epoch。
兼容性警告
本数据集为 2026 格式。若使用传统 2025 流水线训练,需显式格式适配器,否则无法处理 256 维状态、world/canonical-frame base qvel 或 observation.images.rgb.* 相机键。




