real_robot_data
收藏数据集概况
- 数据集名称:Real Robot Manipulation Dataset
- 许可证:Apache-2.0
- 任务类别:机器人学(robotics)
- 标签:真实机器人(real-robot)、操作(manipulation)、UR5e、模仿学习(imitation-learning)
- 数据规模:100K < n < 1M
数据采集平台
- 机器人:UR5e(通过RTDE控制)
- 夹爪:Robotiq 2F-140(自动校准)
- TCP朝向:垂直向下(轴角
[π, 0, 0]) - 相机:2x Intel RealSense,分辨率640×480,帧率20 Hz
- 一个固定基座相机(camera1)
- 一个腕部安装相机(camera2)
- 工作空间:梯形区域,
y ∈ [0.50, 0.80],x半宽0.10 → 0.18(随y线性变化)
任务与指令
| 文件夹 | 指令描述 |
|---|---|
put_blocks_in_plate |
从桌上拿起两个积木,放在盘子上。 |
remove_blocks_from_plate |
从盘子上拿起两个积木,放在桌上。 |
hide_blocks_with_bowl |
将两个积木配对,用碗盖住。 |
remove_blocks_under_bowl |
揭开隐藏的积木,将三个物体分散开。 |
hide_blocks_with_cup |
用倒扣的杯子盖住每个积木。 |
put_blocks_on_cup |
从积木上拿起杯子,将积木堆叠在杯子上。 |
put_blocks_down |
从杯子上取下积木,放在桌上。 |
每个任务文件夹包含task_instruction.json,记录完整的任务规范(物体、拾取/放置策略、夹爪模式、工作空间边界)。
数据规模
- 总集数:700集(每任务100集)
- 采样频率:20 Hz
- 每任务帧数与时长:
| 任务 | 集数 | 约计帧数 | 约计时长(墙钟) |
|---|---|---|---|
| put_blocks_in_plate | 100 | 45,576 | 38分钟 |
| remove_blocks_from_plate | 100 | 45,929 | 38分钟 |
| hide_blocks_with_bowl | 100 | 43,881 | 37分钟 |
| remove_blocks_under_bowl | 100 | 66,090 | 55分钟 |
| hide_blocks_with_cup | 100 | 约变化 | 约40分钟 |
| put_blocks_on_cup | 100 | 约变化 | 约50分钟 |
| put_blocks_down | 100 | 约变化 | 约30分钟 |
数据结构
目录布局
<task>/ 1.hdf5 ... 100.hdf5 # 每集轨迹(图像 + 状态) meta/ 1/meta.json ... 100/meta.json # 每集真值 task_instruction.json # 任务规范
HDF5格式(每集)
/timestamps (T,) float64 /observations/ images/ camera1 (T,H,W,3) uint8 # 基座相机,640x480 RGB camera2 (T,H,W,3) uint8 # 腕部相机,640x480 RGB qpos (T,7) float32 # 6个关节 + 夹爪(归一化) state (T,14) float32 # tcp_pos(3)+tcp_quat(4)+joints(6)+gripper(1) tcp_pos, tcp_quat, joints, gripper_pos
meta.json格式(每集)
所有任务包含每集真值物体位置:
json { "task": "<任务名称>", "episode": <整数>, "start": {<物体>: <(x,y)>, ...}, "end": {<物体>: <(x,y)>, ...}, "init_pose": [x, y, z] }
任务特定键值也会记录(例如杯类任务的perm和cup_order,碗隐藏任务的sign和active/passive,碗/杯移除任务的block_order)。




