five

eval_pi0_record_test

收藏
Hugging Face2026-03-02 更新2026-03-03 收录
下载链接:
https://huggingface.co/datasets/Spartakfs/eval_pi0_record_test
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是使用LeRobot创建的,专注于机器人技术领域。数据集采用Apache-2.0许可证,包含机器人动作和观察数据。数据以parquet文件格式存储,总数据文件大小为100MB,视频文件大小为200MB,帧率为30fps。数据集结构包括机器人关节位置(如肩部、肘部、腕部等)、来自多个摄像头(follower、leader、realsense)的图像观察数据,以及时间戳和各种索引(如帧索引、任务索引等)。图像数据的分辨率为480x640,包含3个通道。该数据集适用于机器人控制、行为模仿等任务。
创建时间:
2026-02-22
原始信息汇总

数据集概述

基本信息

  • 数据集名称: eval_pi0_record_test
  • 发布者: Spartakfs
  • 许可证: Apache 2.0
  • 任务类别: 机器人学
  • 标签: LeRobot

来源与创建

  • 该数据集使用 LeRobot 创建。

数据集结构

  • 数据文件格式: Parquet
  • 数据文件路径模式: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频文件路径模式: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4
  • 分块大小: 1000
  • 数据文件总大小: 100 MB
  • 视频文件总大小: 200 MB
  • 帧率: 30 FPS
  • 代码库版本: v3.0
  • 机器人类型: so_follower

数据特征

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

动作

  • 数据类型: float32
  • 形状: [6]
  • 名称: shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos

状态观测

  • 数据类型: float32
  • 形状: [6]
  • 名称: shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos

图像观测

  • 跟随者图像
    • 数据类型: video
    • 形状: [480, 640, 3]
    • 名称: height, width, channels
  • 领导者图像
    • 数据类型: video
    • 形状: [480, 640, 3]
    • 名称: height, width, channels
  • Realsense图像
    • 数据类型: video
    • 形状: [480, 640, 3]
    • 名称: height, width, channels

索引与元数据

  • 时间戳
    • 数据类型: float32
    • 形状: [1]
  • 帧索引
    • 数据类型: int64
    • 形状: [1]
  • 回合索引
    • 数据类型: int64
    • 形状: [1]
  • 索引
    • 数据类型: int64
    • 形状: [1]
  • 任务索引
    • 数据类型: int64
    • 形状: [1]

数据统计

  • 总回合数: 0
  • 总帧数: 0
  • 总任务数: 0
  • 数据集分割: 未提供

引用信息

  • 主页: 未提供
  • 论文: 未提供
  • BibTeX 引用: 未提供
搜集汇总
数据集介绍
构建方式
在机器人学习领域,高质量的数据采集是模型训练与评估的基石。eval_pi0_record_test数据集依托LeRobot框架构建,通过记录单次完整任务执行过程,采集了涵盖851帧、时长为30帧每秒的时序数据。数据以分块Parquet文件形式存储,每块包含1000条记录,并同步保存了来自跟随者、领导者及RealSense相机的多视角视频流,确保了观测信息的丰富性与同步性。
特点
该数据集的核心特征在于其多维度的机器人状态与视觉观测融合。它精确记录了六自由度机械臂的关节位置动作与对应状态,同时提供了三路高分辨率彩色视频流,每路视频均以AV1编码、480x640分辨率呈现。数据结构化程度高,包含时间戳、帧索引、回合索引等元数据,支持对单任务、单回合的密集采样分析,为模仿学习与策略评估提供了精细的时空对齐基准。
使用方法
为有效利用该数据集进行算法开发或评估,研究者可通过加载指定的Parquet数据文件与配套MP4视频文件来访问数据。数据按特征键组织,例如‘action’、‘observation.state’及多路图像观测,便于直接提取关节空间指令与视觉上下文。鉴于数据集仅包含训练划分,它主要适用于模型在固定任务序列上的行为克隆训练或作为离线评估的基准,使用时需注意其单任务、小规模的特性,合理设计验证流程。
背景与挑战
背景概述
在机器人学领域,模仿学习作为实现智能体自主行为的关键范式,长期依赖于高质量、多模态的演示数据。eval_pi0_record_test数据集应运而生,它依托于LeRobot开源框架构建,专注于记录与评估机器人策略的交互轨迹。该数据集由HuggingFace社区贡献者于近期发布,旨在为机器人策略的离线评估提供标准化基准。其核心研究问题聚焦于如何有效采集并结构化机器人执行任务时的状态-动作对及多视角视觉观测,以支持策略泛化性与鲁棒性的量化分析。尽管数据集规模尚属初期,仅包含单一任务的一个完整片段,但其严谨的数据架构为后续大规模机器人行为数据集的构建奠定了技术基础,对推动数据驱动的机器人学习研究具有潜在影响力。
当前挑战
该数据集致力于解决机器人模仿学习与策略评估中的核心挑战,即如何在复杂、非结构化的真实环境中获取精确、同步的多模态演示数据。具体而言,领域问题的挑战体现在高维连续动作空间与部分可观测状态下的策略学习困难,以及从多相机视角(如跟随者、领导者及RealSense)融合异构感知信息以理解任务上下文的复杂性。在构建过程中,挑战主要源于数据采集的工程实现,包括确保多个传感器数据流的时间严格同步、高效压缩与存储大规模视频流(如采用AV1编解码),以及将关节位置、图像帧与时间戳等异构数据序列化为统一且可扩展的Parquet格式。这些技术障碍的克服,对于构建可靠、可复现的机器人评估基准至关重要。
常用场景
经典使用场景
在机器人学习领域,eval_pi0_record_test数据集为模仿学习与行为克隆提供了关键支持。该数据集通过记录机械臂在特定任务中的关节位置、图像观测及时间戳,构建了从感知到动作的完整映射。研究人员能够利用这些多模态数据训练端到端的控制策略,模拟人类操作者的行为模式,从而在仿真或真实环境中实现精细的抓取或操作任务。
解决学术问题
该数据集有效应对了机器人控制中样本效率低下与泛化能力不足的挑战。通过提供高维视觉输入与精确动作标签的对应关系,它促进了基于深度学习的策略优化方法的发展,如强化学习与监督学习的结合。这有助于解决动态环境下的状态估计误差问题,并为跨任务迁移学习提供了实证基础,推动了自主机器人系统在非结构化场景中的适应性研究。
衍生相关工作
围绕该数据集,衍生出多项经典研究工作,包括基于视觉的机械臂轨迹预测模型、多视角图像融合的态势感知算法,以及端到端模仿学习框架的优化。这些工作不仅扩展了数据集的利用维度,还催生了新的评估基准,促进了机器人学习社区在仿真到真实迁移、样本复用等方向的深入探索,为后续大规模机器人数据集的构建提供了范式参考。
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作