遇见数据集

unitzero-cube-pick

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

资源简介:

该数据集用于机器人操控任务,具体为从桌子上拾取立方体(cube pick)。数据由LeRobot框架收集,包含一个双臂机器人(bi_yam_follower)的14维关节角度和夹爪位置动作,以及对应的14维关节角度状态观测。同时提供了三个视角的RGB视频:环境摄像头(480x640)、左腕摄像头和右腕摄像头,帧率为50fps。数据集共包含52个episode,总计104947帧,每个episode对应一个拾取立方体的演示。

This dataset is used for robot manipulation tasks, specifically for picking up a cube from a table. The data is collected by the LeRobot framework, containing 14-dimensional joint angles and gripper position actions of a dual-arm robot (bi_yam_follower), along with corresponding 14-dimensional joint angle state observations. It also provides RGB videos from three perspectives: an environment camera (480x640), a left wrist camera, and a right wrist camera, at a frame rate of 50fps. The dataset consists of a total of 52 episodes, with 104947 frames in total, each episode corresponding to a demonstration of picking up a cube.

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

数据集概述:unitzero-cube-pick

基本信息

  • 数据集名称: unitzero-cube-pick
  • 许可证: Apache-2.0
  • 任务类别: 机器人学(robotics)
  • 标签: LeRobot
  • 创建工具: 使用 LeRobot 创建

数据规模

指标 数值
总回合数(episodes) 52
总帧数 104,947
总任务数 1
采样频率(fps) 50
数据文件大小 100 MB
视频文件大小 200 MB
数据划分 train: 0:52(全部用于训练)

机器人类型

  • 机器人型号: bi_yam_follower

数据特征

数据集包含以下特征:

  1. action(动作): 14维float32数组,包含左右臂各6个关节位置和1个夹爪位置
  2. observation.state(观测状态): 14维float32数组,与action结构相同
  3. observation.images.env(环境图像): 视频数据,分辨率480x640x3,H.264编码,50fps
  4. observation.images.wrist_left(左腕部图像): 视频数据,分辨率480x640x3,H.264编码,50fps
  5. observation.images.wrist_right(右腕部图像): 视频数据,分辨率480x640x3,H.264编码,50fps
  6. timestamp(时间戳): float32类型
  7. frame_index(帧索引): int64类型
  8. episode_index(回合索引): int64类型
  9. index(索引): int64类型
  10. task_index(任务索引): int64类型

数据文件结构

  • 元数据文件: meta/info.json
  • 数据文件路径: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet(Parquet格式)
  • 视频文件路径: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4(MP4格式)
  • 配置: default,数据文件位于 data/*/*.parquet
  • 代码库版本: v3.0
二维码
社区交流群
二维码
科研交流群
商业服务