遇见数据集

demo_src_threading_task_D2_robot_Sawyer_gripper_RethinkGripper

收藏
Hugging Face2026-06-25 更新2026-06-25 收录
官方服务:

资源简介:

该数据集是一个用于机器人学研究的跨体现数据集,由LeRobot创建。它包含机器人执行操作任务时的多模态数据,具体包括动作(7维浮点数组,表示dx、dy、dz、drx、dry、drz和夹爪控制)、观测状态(如末端执行器位置和四元数、关节位置、夹爪位置)、图像观测(来自agentview和wrist视角的256x256 RGB视频,帧率20fps),以及时间戳、帧索引、episode索引等元数据。数据集总共有100个episodes、23,362帧,数据以parquet格式存储,适用于机器人模仿学习或强化学习任务。

This dataset is a cross-embodied dataset for robotics research, created by LeRobot. It contains multimodal data from robots performing manipulation tasks, including actions (7-dimensional floating-point arrays representing dx, dy, dz, drx, dry, drz, and gripper control), observation states (such as end-effector positions and quaternions, joint positions, gripper positions), image observations (256x256 RGB videos from agentview and wrist perspectives at 20fps), and metadata like timestamps, frame indices, and episode indices. The dataset consists of 100 episodes and 23,362 frames, stored in parquet format, and is suitable for robot imitation learning or reinforcement learning tasks.

提供机构:
OliverHausdoerfer
创建时间:
2026-06-25
原始信息汇总

数据集概述

  • 数据集名称: demo_src_threading_task_D2_robot_Sawyer_gripper_RethinkGripper
  • 许可证: Apache-2.0
  • 任务类别: 机器人学(Robotics)
  • 标签: LeRobot、MimicGen、跨本体(Cross-embodiment)
  • 创建工具: 使用 LeRobot 创建

数据集规模

  • 总片段数: 100
  • 总帧数: 23362
  • 总任务数: 1
  • 数据集文件大小: 100 MB(Parquet 格式)
  • 视频文件大小: 200 MB(MP4 格式)
  • 帧率: 20 FPS
  • 数据块大小: 每块 1000 帧
  • 数据分割: 训练集(索引 0 到 99),无验证/测试集

数据结构

数据以 Parquet 文件存储,视频以 MP4 格式存储,路径结构固定:

  • 数据路径: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频路径: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

特征字段

特征名称 数据类型 形状 字段含义
action float32 (7,) 机器人动作指令,包括位移(dx, dy, dz)、旋转(drx, dry, drz)和夹爪(gripper)
observation.state float32 (2,) 机器人状态,包含两个零位参数(zero0, zero1)
observation.eef float32 (7,) 末端执行器位姿:位置(x, y, z)和四元数(qx, qy, qz, qw)
observation.gripper float32 (6,) 夹爪关节位置(gripper_qpos_0 至 gripper_qpos_5)
observation.joint float32 (7,) 机器人关节位置(joint_pos_0 至 joint_pos_6)
observation.eef_gripper float32 (13,) 组合特征:末端执行器位姿(7维)+ 夹爪关节位置(6维)
observation.joint_gripper float32 (13,) 组合特征:机器人关节位置(7维)+ 夹爪关节位置(6维)
observation.images.agentview video (256, 256, 3) 主视角 RGB 视频,分辨率 256×256,编码 AV1,FPS 20
observation.images.wrist video (256, 256, 3) 腕部视角 RGB 视频,分辨率 256×256,编码 AV1,FPS 20
timestamp float32 (1,) 时间戳(单一数值)
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 片段索引
index int64 (1,) 全局索引
task_index int64 (1,) 任务索引(当前为单一任务,值为0)

机器人类型

  • 机器人: 跨本体(cross_embodiment),具体为 Sawyer 机器人(Rethink 公司),配备 Rethink 夹爪

视频规格

视频特征 主视角 (agentview) 腕部视角 (wrist)
分辨率 256×256 256×256
色彩通道 3(RGB) 3(RGB)
编码格式 AV1 AV1
像素格式 yuv420p yuv420p
帧率 20 FPS 20 FPS
深度图

其他信息

  • 主页:更多信息待补充
  • 论文:更多信息待补充
  • 引用格式:更多信息待补充
  • 可视化:可通过 LeRobot 可视化工具 交互式浏览数据集内容
二维码
社区交流群
二维码
科研交流群
商业服务