transfer_cube_20260704_230228
收藏资源简介:
该数据集是使用LeRobot框架创建的机器人操作数据集,专门用于机器人技术领域的研究与应用。数据集的核心内容围绕一个机械臂操作任务(推测为物体转移或操作,根据数据集名称Transfer Cube推断),记录了机器人在执行任务过程中的状态、动作及环境观测。数据集包含50个完整的任务执行回合(episodes),共计22,385个时间步(帧),覆盖25个不同的任务实例。数据以高效的分块格式(Parquet)存储,每块包含1000帧。每个数据样本包含以下关键字段:1. 动作(action):一个6维浮点向量,表示机器人在该时间步执行的关节位置命令,具体控制肩部平移、肩部抬升、肘部弯曲、腕部弯曲、腕部旋转以及夹爪开合。2. 观测状态(observation.state):一个6维浮点向量,表示机器人在该时间步的实际关节位置传感器读数,与动作字段的维度与含义一一对应。3. 观测图像(observation.images.overhead):一个来自顶部固定视角的RGB视频帧,分辨率为640x480,帧率为30 FPS,以AV1编码格式存储,提供了任务执行环境的视觉信息。4. 元数据:包括时间戳(timestamp)、当前帧在回合内的索引(frame_index)、所属回合的索引(episode_index)、全局唯一索引(index)以及任务类型索引(task_index)。数据集使用的机器人平台为so_follower类型。整个数据集被指定为训练用途(splits: train)。数据总大小约为300MB(其中结构化数据约100MB,视频数据约200MB)。该数据集适用于机器人模仿学习、行为克隆、视觉运动策略学习、离线强化学习等任务,为研究者提供了动作-状态-视觉观测对齐的多模态演示数据。
数据集概述
- 名称: transfer_cube_20260704_230228
- 许可证: Apache-2.0
- 任务类别: 机器人学 (robotics)
- 标签: LeRobot
数据集结构
- 总集数 (total_episodes): 50
- 总帧数 (total_frames): 22,385
- 总任务数 (total_tasks): 25
- 帧率 (fps): 30
- 机器人类型 (robot_type): so_follower
- 数据分割 (splits): 训练集 (train) 包含全部 50 集 (0:50)
特征 (Features)
| 特征名称 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
action |
float32 | [6] | 机器人动作,包含 shoulder_pan, shoulder_lift, elbow_flex, wrist_flex, wrist_roll, gripper 的位置 |
observation.state |
float32 | [6] | 机器人状态,与动作相同的6维位置信息 |
observation.images.overhead |
video | [480, 640, 3] | 俯视摄像头视频,分辨率 480x640,3通道彩色,AV1编码,30fps |
timestamp |
float32 | [1] | 时间戳 |
frame_index |
int64 | [1] | 帧索引 |
episode_index |
int64 | [1] | 集索引 |
index |
int64 | [1] | 索引 |
task_index |
int64 | [1] | 任务索引 |
文件大小
- 数据文件 (data_files): 约 100 MB (Parquet 格式,路径:
data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet) - 视频文件 (video_files): 约 200 MB (MP4 格式,路径:
videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4)
其他信息
- 来源: 使用 LeRobot 工具创建
- 可视化: 可通过 Hugging Face Space 进行可视化
- 论文: 暂无提供




