遇见数据集

aibot2_grab_bag_combined_over_under

收藏
Hugging Face2026-07-03 更新2026-07-03 收录
官方服务:

资源简介:

该数据集是使用LeRobot创建的机器人数据集,专门用于机器人技术研究。它包含351个训练片段,总计21050帧,覆盖4个不同任务。数据以每秒10帧的速率采集,包括机器人动作和状态观测信息:动作特征涵盖左右机械臂的3D位置和四元数姿态,以及左右夹爪的位置;状态观测特征包括左右机械臂的7个关节位置和左右夹爪的关节位置。观测图像数据来自四个摄像头(前、左、右、顶),每个摄像头的视频分辨率为240x320像素,3通道彩色,编码为AV1格式。数据集还包含时间戳、帧索引、片段索引和任务索引等元数据。数据以Parquet格式存储,总数据文件大小为100MB,视频文件大小为200MB。

This dataset is a robotics dataset created using LeRobot, specifically designed for robotics research. It contains 351 training episodes, totaling 21,050 frames, covering 4 different tasks. Data is collected at a rate of 10 frames per second and includes robot actions and state observations: action features consist of 3D positions and quaternion poses for the left and right robotic arms, as well as positions for the left and right grippers; state observation features include 7 joint positions for the left and right robotic arms and joint positions for the left and right grippers. Observation image data comes from four cameras (front, left, right, top), each with video resolution of 240x320 pixels, 3-channel color, encoded in AV1 format. The dataset also includes metadata such as timestamps, frame indices, episode indices, and task indices. Data is stored in Parquet format, with a total data file size of 100MB and video file size of 200MB.

提供机构:
alphabot2
创建时间:
2026-07-03
原始信息汇总

数据集概述

  • 数据集名称: aibot2_grab_bag_combined_over_under
  • 许可证: Apache-2.0
  • 任务类别: 机器人学 (Robotics)
  • 创建工具: 使用 LeRobot 创建

数据集规模

  • 总片段数 (Episodes): 351
  • 总帧数 (Frames): 21,050
  • 总任务数 (Tasks): 4
  • 帧率 (FPS): 10
  • 数据文件大小: 100 MB
  • 视频文件大小: 200 MB

数据集结构

  • 数据集拆分: 仅有训练集 (train),包含全部 351 个片段 (索引 0:351)
  • 数据格式: Parquet 文件,路径为 data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频格式: MP4 文件,路径为 videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

特征 (Features)

数据集包含以下特征字段:

特征名称 数据类型 形状 说明
action float32 [16] 机械臂动作,包含左右臂位置/姿态 (x, y, z, qx, qy, qz, qw) 及夹爪位置 (pos)
observation.state float32 [16] 机械臂关节状态,包含左右臂 7 个关节位置及夹爪关节位置
observation.images.camera_front video [240, 320, 3] 前置摄像头图像 (AV1 编码, 10 FPS)
observation.images.camera_left video [240, 320, 3] 左侧摄像头图像 (AV1 编码, 10 FPS)
observation.images.camera_right video [240, 320, 3] 右侧摄像头图像 (AV1 编码, 10 FPS)
observation.images.camera_top video [240, 320, 3] 顶部摄像头图像 (AV1 编码, 10 FPS)
timestamp float32 [1] 时间戳
frame_index int64 [1] 帧索引
episode_index int64 [1] 片段索引
index int64 [1] 全局索引
task_index int64 [1] 任务索引

动作定义

action 字段包含 16 个数值,分别对应:

  • left_arm.xleft_arm.qw: 左臂末端姿态 (位置 + 四元数)
  • right_arm.xright_arm.qw: 右臂末端姿态 (位置 + 四元数)
  • left_gripper.pos: 左夹爪位置
  • right_gripper.pos: 右夹爪位置

状态定义

observation.state 字段包含 16 个数值,分别对应:

  • ZPF_left_Joint1.posZPF_left_Joint7.pos: 左臂 7 个关节位置
  • ZPF_right_Joint1.posZPF_right_Joint7.pos: 右臂 7 个关节位置
  • 2FEG_l_Joint1.pos: 左夹爪关节位置
  • 2FEG_r_Joint1.pos: 右夹爪关节位置

图像信息

  • 分辨率: 240×320 像素
  • 颜色通道: 3 (RGB)
  • 编码格式: AV1
  • 像素格式: yuv420p
  • 帧率: 10 FPS

引用信息

BibTeX 引用信息暂时缺失 ([More Information Needed])。

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