handover_cube_to_human
收藏资源简介:
该数据集是一个用于机器人操作任务的模仿学习数据集,由 LeRobot 框架创建。数据集记录了机器人将立方体递给人类的过程。数据包含49个演示片段(episodes),共6685帧,涵盖8个不同任务。每个样本包含来自三个不同视角的摄像头图像(分辨率224x224,AV1编码视频)、机器人状态(7维:6个关节位置和夹爪状态)、动作指令(7维:平移增量、旋转增量和夹爪控制)、时间戳、帧索引、episode索引和任务索引。数据以Parquet文件和视频文件的形式存储,并已预划分为训练集。该数据集适用于机器人模仿学习、行为克隆、离线强化学习等任务。
This dataset is an imitation learning dataset for robot manipulation tasks, created using the LeRobot framework. It records the process of a robot handing a cube to a human. The dataset contains 49 demonstration episodes with a total of 6685 frames, covering 8 different tasks. Each sample includes camera images from three different viewpoints (resolution 224x224, AV1 encoded video), robot state (7-dimensional: 6 joint positions and gripper state), action commands (7-dimensional: translation deltas, rotation deltas, and gripper control), timestamps, frame indices, episode indices, and task indices. Data is stored as Parquet files and video files, and has been pre-split into training sets. The dataset is suitable for tasks such as robot imitation learning, behavior cloning, and offline reinforcement learning.
数据集概述:handover_cube_to_human
基本信息
- 数据集名称:handover_cube_to_human
- 许可证:Apache-2.0
- 任务类别:机器人学(robotics)
- 创建工具:由 LeRobot(https://github.com/huggingface/lerobot)创建
- 代码库版本:v3.0
数据集规模
- 总片段数(Episodes):49
- 总帧数(Frames):6,685
- 总任务数(Tasks):8
- 数据划分:训练集(train)包含全部49个片段("0:49")
- 数据文件大小:约100 MB(parquet格式)
- 视频文件大小:约200 MB(mp4格式)
数据特征(Features)
数据集中包含以下特征字段:
| 特征名 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
| observation.images.camera1 | 视频 | 224×224×3 | 摄像头1图像,AV1编码,20 FPS |
| observation.images.camera2 | 视频 | 224×224×3 | 摄像头2图像,AV1编码,20 FPS |
| observation.images.camera3 | 视频 | 224×224×3 | 摄像头3图像,AV1编码,20 FPS |
| observation.state | float32 | 7 | 机器人状态,包含6个关节(joint_0 至 joint_5)和1个夹爪(gripper) |
| action | float32 | 7 | 动作指令,包含6维位姿增量(dx, dy, dz, droll, dpitch, dyaw)和夹爪控制(gripper) |
| timestamp | float32 | 1 | 时间戳 |
| frame_index | int64 | 1 | 帧索引 |
| episode_index | int64 | 1 | 片段索引 |
| index | int64 | 1 | 全局索引 |
| task_index | int64 | 1 | 任务索引 |
数据存储结构
- 数据文件路径:
data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet - 视频文件路径:
videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4 - 分块大小:每块1000个样本(chunks_size: 1000)
视频编码参数
- 编码格式:AV1
- 视频帧率:20 FPS
- 像素格式:yuv420p
- 视频分辨率:224×224
- 颜色通道数:3
- CRF值:30
- 无音频轨道
可视化与使用
该数据集支持通过 Hugging Face 的 LeRobot 可视化工具进行查看(可视化地址:https://huggingface.co/spaces/lerobot/visualize_dataset?path=Hungdd88/handover_cube_to_human)。数据以 parquet 格式存储结构化数据,视频以 mp4 格式存储。
引用信息
当前数据集的引用信息(BibTeX)尚未提供(标注为"More Information Needed")。




