遇见数据集

merge_task1_UM_epi_200

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

资源简介:

该数据集基于LeRobot创建,是一个机器人数据集。使用Panda机器人,包含200个episodes,共30145帧,涉及2个不同任务。数据以15fps录制,包含主摄像头(256x342)和腕部摄像头(256x342)的图像,以及机器人状态(包括笛卡尔坐标、夹爪位置、7个关节角度、目标状态)和7维动作(包括位置、姿态和夹爪控制)。

This dataset is created based on LeRobot and is a robot dataset. It uses the Panda robot, contains 200 episodes with a total of 30,145 frames, and involves 2 different tasks. The data is recorded at 15fps, including images from the main camera (256x342) and wrist camera (256x342), robot states (including Cartesian coordinates, gripper position, 7 joint angles, target state), and 7-dimensional actions (including position, orientation, and gripper control).

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

数据集概述

  • 名称cbrian/merge_task1_UM_epi_200
  • 许可证:Apache-2.0
  • 任务类别:机器人学(Robotics)
  • 标签:LeRobot
  • 创建工具:LeRobot(https://github.com/huggingface/lerobot)

数据集结构

  • 数据版本:v3.0
  • 机器人类型:Panda(Franka Emika Panda)
  • 总集数(Episodes):200
  • 总帧数:30,145
  • 任务数量:2
  • 分块大小(Chunks):1000
  • 数据文件大小:约100 MB(Parquet格式)
  • 视频文件大小:约200 MB(MP4格式)
  • 帧率(FPS):15
  • 数据路径data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频路径videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

数据划分

  • 训练集train(包含所有200个片段,即 0:200
  • 验证/测试集:无(未提供)

特征说明(Features)

该数据集包含以下特征,每个特征的详细信息如下:

1. 图像观测(Observation Images)

  • observation.images.primary:主摄像头图像
    • 类型:视频(Video)
    • 形状:256 × 342 × 3(高×宽×通道)
    • 视频编码:AV1,像素格式:yuv420p,帧率:15 FPS(存储时30 FPS,数据集内15 FPS)
    • 无音频,非深度图
  • observation.images.wrist:腕部摄像头图像
    • 类型:视频(Video)
    • 形状:256 × 342 × 3(高×宽×通道)
    • 视频编码:AV1,像素格式:yuv420p,帧率:15 FPS(存储时30 FPS,数据集内15 FPS)
    • 无音频,非深度图

2. 状态观测(Observation State)

  • observation.state.cartesian:笛卡尔坐标(6维)
    • 包含:x, y, z, roll, pitch, yaw
  • observation.state.gripper:夹爪状态(1维)
    • 包含:gripper
  • observation.state.joints:关节角度(7维)
    • 包含:joint_0 至 joint_6
  • observation.state.target:目标状态(6维)
    • 包含:target_x, target_y, target_z, target_roll, target_pitch, target_yaw
  • observation.state:综合状态(20维)
    • 包含:笛卡尔坐标、夹爪、关节角度以及目标状态(顺序:x, y, z, roll, pitch, yaw, gripper, joint_0~joint_6, target_x~target_yaw)

3. 动作(Action)

  • action:机器人动作(7维)
    • 包含:x, y, z, roll, pitch, yaw, gripper

4. 元数据(Metadata)

  • timestamp:时间戳(float32,1维)
  • frame_index:帧索引(int64,1维)
  • episode_index:片段索引(int64,1维)
  • index:全局索引(int64,1维)
  • task_index:任务索引(int64,1维)

其他信息

  • 可视化:可通过LeRobot的可视化工具直接查看数据集内容(访问路径:https://huggingface.co/spaces/lerobot/visualize_dataset?path=cbrian/merge_task1_UM_epi_200)
  • 引用信息:暂无BibTeX引用(页面标注为[More Information Needed]

总结

该数据集是一个用于机器人操作任务的LeRobot格式数据集,包含200个演示片段,涉及2个任务,总帧数约3万帧。数据由Panda机器人采集,包含主摄像头和腕部摄像头的视频流,以及完整的机器人状态(笛卡尔坐标、关节角度、夹爪状态、目标状态)和动作数据。数据文件较小(约100 MB),适合用于机器人模仿学习或强化学习的训练。

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