遇见数据集

demo_src_three_piece_assembly_task_D1_robot_UR5e_gripper_Robotiq85Gripper_additionalCams

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

资源简介:

该数据集使用LeRobot创建,是一个机器人控制相关的跨本体数据集。包含500个训练片段,总计169,607帧数据,帧率为20fps。数据集记录了机器人执行任务时的多模态信息:包括7维动作空间(dx, dy, dz, drx, dry, drz, gripper),观测数据包含状态(2维)、末端执行器位姿(7维:位置和四元数)、夹爪状态(6维)、关节位置(7维)以及组合特征(末端执行器+夹爪13维,关节+夹爪13维)。此外,还提供两个视角的视觉观测:前视图和侧视图,均为256x256分辨率的RGB视频。数据集结构以parquet文件格式存储,视频使用AV1编码。

This dataset, developed using LeRobot, is a cross-embodiment dataset focused on robot control. It contains 500 training episodes, with a total of 169,607 frames at a frame rate of 20 fps. The dataset records multi-modal information generated during robot task execution: the action space is 7-dimensional (dx, dy, dz, drx, dry, drz, gripper). The observation data includes 2-dimensional states, 7-dimensional end-effector poses (comprising position and quaternion), 6-dimensional gripper states, 7-dimensional joint positions, as well as two combined feature sets: end-effector + gripper (13-dimensional) and joint + gripper (13-dimensional). Additionally, visual observations from two perspectives are provided: a front view and a side view, both are 256x256 resolution RGB videos. The dataset is stored in Parquet file format, and all videos are encoded with the AV1 codec.

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

数据集概述

该数据集是一个用于机器人操作任务的演示数据集,由 LeRobot 框架创建,采用 Apache-2.0 许可证。

数据集基本信息

  • 机器人类型:跨体态(cross_embodiment)
  • 机器人型号:UR5e 机械臂,配备 Robotiq85 夹爪及额外摄像头
  • 任务总数:1
  • 数据规模
    • 总片段数(episodes):500
    • 总帧数:169,607
    • 帧率(fps):20
    • 数据文件大小:100 MB
    • 视频文件大小:200 MB
  • 数据集划分:训练集包含全部 500 个片段(0:500)

数据格式与存储

  • 数据文件路径data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频文件路径videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4
  • 块大小(chunks_size):1000

数据特征

特征名称 数据类型 形状 说明
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,) 夹爪关节位置(0-5)
observation.joint float32 (7,) 机器人关节位置(0-6)
observation.eef_gripper float32 (13,) 末端执行器位姿 + 夹爪关节位置
observation.joint_gripper float32 (13,) 关节位置 + 夹爪关节位置
observation.images.frontview video (256,256,3) 前视摄像头彩色图像,AV1 编码,20 fps
observation.images.sideview video (256,256,3) 侧视摄像头彩色图像,AV1 编码,20 fps
timestamp float32 (1,) 时间戳
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 片段索引
index int64 (1,) 全局索引
task_index int64 (1,) 任务索引

可视化

可通过 可视化页面 在线查看数据集内容。

二维码
社区交流群
二维码
科研交流群
商业服务