droid_1.0.1_v30
收藏资源简介:
该数据集是一个使用LeRobot框架创建的机器人演示数据集,专门用于机器人学习任务。数据集采集自Franka机器人,包含大量机器人执行任务的交互数据,总计95,658个任务片段(episodes),27,630,375帧数据,涵盖49,630个不同的任务。数据以Parquet文件格式存储,并附带MP4格式视频文件(帧率15fps)。数据集仅包含训练集。内容涵盖多模态信息:状态观测(如夹爪位置、笛卡尔位置和关节位置)、视觉观测(三个视角的RGB视频流,分辨率180x320)、动作数据(包括夹爪和笛卡尔的位置/速度)、语言指令(最多三个自然语言指令)、元数据(任务类别、采集信息等)、强化学习信号(奖励和折扣因子)以及索引信息。适用于机器人模仿学习、强化学习、视觉语言动作模型训练,特别是多模态机器人策略学习研究。
This dataset is a robot demonstration dataset created using the LeRobot framework, specifically designed for robot learning tasks. It is collected from a Franka robot and contains extensive interaction data from robot task executions, totaling 95,658 episodes, 27,630,375 frames, and covering 49,630 distinct tasks. The data is stored in Parquet format along with corresponding MP4 video files at 15fps. The dataset includes only a training set. It features rich multimodal information: state observations (such as gripper position, Cartesian position, and joint positions), visual observations (three RGB video streams from different viewpoints with 180x320 resolution), action data (including gripper and Cartesian position/velocity), language instructions (up to three natural language descriptions), metadata (task categories, collection details, etc.), reinforcement learning signals (rewards and discount factors), and indexing information. It is suitable for research in robot imitation learning, reinforcement learning, vision-language-action model training, particularly for multimodal robot policy learning tasks.
数据集概述
- 数据集名称:
muacha/droid_1.0.1_v30 - 许可协议: Apache-2.0
- 任务类别: 机器人学 (robotics)
- 标签: LeRobot
- 创建工具: LeRobot
数据集规模
- 总片段数 (episodes): 95,658
- 总帧数 (frames): 27,630,375
- 总任务数 (tasks): 49,630
- 数据块大小 (chunks size): 1,000
- 数据文件总大小: 100 MB
- 视频文件总大小: 500 MB
- 帧率 (fps): 15
- 训练集划分:
train(0:95658)
机器人类型
- 机器人型号: Franka
数据集结构
- 数据路径:
data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet - 视频路径:
videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4
特征字段
状态观测 (observation.state)
| 字段 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
observation.state.gripper_position |
float32 | (1,) | 夹爪位置 |
observation.state.cartesian_position |
float32 | (6,) | 笛卡尔位置 (x, y, z, roll, pitch, yaw) |
observation.state.joint_position |
float32 | (7,) | 关节位置 (joint_0 到 joint_6) |
observation.state |
float32 | (8,) | 综合状态 (joint_0 到 joint_6 + gripper) |
图像观测 (observation.images)
| 字段 | 分辨率 | 编码 | 帧率 |
|---|---|---|---|
observation.images.wrist_left |
180×320×3 | AV1 (yuv420p) | 15 fps |
observation.images.exterior_1_left |
180×320×3 | AV1 (yuv420p) | 15 fps |
observation.images.exterior_2_left |
180×320×3 | AV1 (yuv420p) | 15 fps |
动作 (action)
| 字段 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
action.gripper_position |
float32 | (1,) | 夹爪位置动作 |
action.gripper_velocity |
float32 | (1,) | 夹爪速度动作 |
action.cartesian_position |
float32 | (6,) | 笛卡尔位置动作 (x, y, z, roll, pitch, yaw) |
action.cartesian_velocity |
float32 | (6,) | 笛卡尔速度动作 (x, y, z, roll, pitch, yaw) |
action.joint_position |
float32 | (7,) | 关节位置动作 (joint_0 到 joint_6) |
action.joint_velocity |
float32 | (7,) | 关节速度动作 (joint_0 到 joint_6) |
action.original |
float32 | (7,) | 原始动作 (x, y, z, roll, pitch, yaw, gripper) |
action |
float32 | (8,) | 综合动作 (joint_0 到 joint_6 + gripper) |
其他特征
| 字段 | 数据类型 | 说明 |
|---|---|---|
is_first |
bool | 是否为片段首帧 |
is_last |
bool | 是否为片段末帧 |
is_terminal |
bool | 是否为终止帧 |
language_instruction |
string | 语言指令 |
language_instruction_2 |
string | 第二条语言指令 |
language_instruction_3 |
string | 第三条语言指令 |
discount |
float32 | 折扣因子 |
reward |
float32 | 奖励值 |
task_category |
string | 任务类别 |
building |
string | 建筑或场景信息 |
collector_id |
string | 数据收集者ID |
date |
string | 数据采集日期 |
camera_extrinsics.wrist_left |
float32 | 腕部左相机外参 (6-DoF) |
camera_extrinsics.exterior_1_left |
float32 | 外部左相机1外参 (6-DoF) |
camera_extrinsics.exterior_2_left |
float32 | 外部左相机2外参 (6-DoF) |
is_episode_successful |
bool | 片段是否成功 |
timestamp |
float32 | 时间戳 |
frame_index |
int64 | 帧索引 |
episode_index |
int64 | 片段索引 |
index |
int64 | 全局索引 |
task_index |
int64 | 任务索引 |
引用
bibtex [More Information Needed]




