遇见数据集

pick_two_cube_dataset

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

资源简介:

该数据集是一个机器人操作数据集,用于双臂机器人抓取两个立方体的任务。数据集包含45个演示片段,总计15504帧,涉及2个不同的任务。每个片段包含机器人的动作序列(6个关节位置)、状态观测(6个关节位置)以及两个摄像头视角的图像:侧方摄像头(480x640)和手腕摄像头(480x640)。数据以30fps录制,采用Apache-2.0许可证。

This dataset is a robot manipulation dataset for the task of dual-arm robot grasping two cubes. It contains 45 demonstration episodes with a total of 15504 frames, involving 2 different tasks. Each episode includes the robots action sequence (6 joint positions), state observation (6 joint positions), and images from two camera perspectives: a side camera (480x640) and a wrist camera (480x640). Data is recorded at 30fps and licensed under Apache-2.0.

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

数据集概述

数据集名称:pick_two_cube_dataset

任务类型:机器人学(Robotics)

许可证:Apache-2.0

数据集创建工具:由 LeRobot 创建,LeRobot 是一个用于机器人学习的数据集工具库。

数据集可视化工:可通过 LeRobot 提供的可视化平台在线查看数据集内容(可视化链接)。


数据集结构

  • 数据格式:Parquet 文件,存储在 data/*/*.parquet 路径下。
  • 视频内容:包含摄像头画面,存储在 videos/ 目录中,视频使用 AV1 编码,分辨率为 480x640,帧率 30 FPS。
  • 版本信息:代码库版本为 v3.0,采样频率为 30 FPS。

特征字段

特征名 数据类型 形状 说明
action float32 [6] 机器人动作指令,包括肩部、肘部、腕部位置及夹爪位置
observation.state float32 [6] 机器人观测状态,与动作字段对应
observation.images.side 视频 480x640x3 侧面摄像头画面
observation.images.wrist 视频 480x640x3 腕部摄像头画面
timestamp float32 [1] 时间戳
frame_index int64 [1] 帧索引
episode_index int64 [1] 回合索引
index int64 [1] 全局索引
task_index int64 [1] 任务索引

数据统计

  • 总回合数:115 个回合
  • 总帧数:41,249 帧
  • 任务数量:2 个不同任务
  • 数据文件大小:约 100 MB
  • 视频文件大小:约 200 MB
  • 数据划分:仅包含训练集(train),共 115 个回合,无验证集或测试集划分

机器人类型

  • 机器人型号so_follower,一种机械臂,包含肩部、肘部、腕部及夹爪共 6 个自由度。

引用信息

暂无提供 BibTeX 引用格式(标注为“More Information Needed”)。

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