遇见数据集

pick_two_cube_dataset_20260806_153929

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

资源简介:

该数据集是一个机器人操作数据集,使用LeRobot框架创建。数据包含6维动作(肩部、肘部、腕部、夹爪位置)和状态,以及侧视和腕部摄像头图像(480x640,30fps,AV1编码)。总共有1个episode,365帧,1个任务。机器人类型为so_follower。数据集以parquet和mp4格式存储,分为训练集。

This dataset is a robot manipulation dataset created using the LeRobot framework. The data includes 6-dimensional actions (shoulder, elbow, wrist, gripper position) and states, as well as side-view and wrist camera images (480x640, 30fps, AV1 encoded). It contains 1 episode, 365 frames, and 1 task. The robot type is so_follower. The dataset is stored in parquet and mp4 formats and is split into a training set.

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

数据集概述

该数据集名为 mwhnh10/pick_two_cube_dataset_20260806_153929,是一个用于机器人学习(Robotics)任务的数据集,基于 LeRobot 框架创建。

基本信息

  • 许可证:Apache-2.0
  • 任务类别:机器人(Robotics)
  • 标签:LeRobot
  • 创建来源:使用 LeRobot 框架生成

数据集结构

  • 数据格式:parquet 文件,路径模式为 data/*/*.parquet
  • 版本:codebase_version v3.0
  • 帧率:30 FPS

数据特征(Features)

数据集中包含以下特征:

特征名 数据类型 形状 说明
action float32 [6] 机器人动作,包含6个关节位置
observation.state float32 [6] 机器人状态观测,同样包含6个关节位置
observation.images.side video [480, 640, 3] 侧方视角视频图像,AV1编码,30 FPS
observation.images.wrist video [480, 640, 3] 腕部视角视频图像,AV1编码,30 FPS
timestamp float32 [1] 时间戳
frame_index int64 [1] 帧索引
episode_index int64 [1] 回合索引
index int64 [1] 全局索引
task_index int64 [1] 任务索引

动作和状态中的6个关节名称分别为:shoulder_pan.posshoulder_lift.poselbow_flex.poswrist_flex.poswrist_roll.posgripper.pos

规模统计

  • 总回合数:1
  • 总帧数:365
  • 总任务数:1
  • 数据文件大小:约100 MB
  • 视频文件大小:约200 MB
  • 训练/测试划分:train 部分为 0:1(全部数据用于训练)

机器人类型

  • 机器人型号:so_follower(SO-100 跟随式机器人)

数据内容说明

该数据集包含一个机器人执行“拿起两个方块”(pick two cube)任务的演示数据,记录了6自由度关节控制信号、侧方与腕部两个视角的视频数据,以及时间戳和索引信息。

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