遇见数据集

Physical_AI_SO101_Cup_Nesting_Task

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

资源简介:

DecisionFacts Physical AI Dataset 是一个专为机器人学习设计的高质量遥操作数据集,基于 SO-101 机械臂(so_follower)采集,旨在支持从模仿学习策略到大规模视觉-语言-动作(VLA)模型及世界模型的训练与评估。每个 episode 都是由训练有素的操作员通过遥操作完成的真实演示,以 30 fps 的帧率同步记录两个摄像头视角的 RGB 视频、机械臂的完整本体感知状态(6 自由度关节状态:shoulder_pan, shoulder_lift, elbow_flex, wrist_flex, wrist_roll, gripper)以及动作信号。数据采集遵循严格的标准化协议,包括专家遥操作、同步多模态捕获、一致性检查以及成功/失败/恢复的 episode 结果标注。数据集采用 Hugging Face LeRobotDataset v3.0 格式,每个任务拥有独立的目录,包含 data/(Parquet 格式的帧级状态、动作、索引等)、videos/(两个视角的 AV1 编码 MP4 视频,480×640 分辨率)和 meta/(自描述元数据,如配置、统计信息、任务描述、episode 索引)。每个任务的目标 episode 数量为 30 个以上,并通过域随机化(物体位置、姿态、场景条件变化)增强多样性。数据集适用于 VLA 端到端训练、机器人后训练、VLM 预训练/后训练以及世界模型学习。该仓库为公共评估样本,遵循 CC BY-NC 4.0 许可,仅限研究和评估用途;完整目录可通过商业许可获取。

The DecisionFacts Physical AI Dataset is a high-quality teleoperation dataset designed for robot learning, collected using the SO-101 robotic arm (so_follower), aimed at supporting training and evaluation from imitation learning policies to large-scale Vision-Language-Action (VLA) models and world models. Each episode is a real demonstration performed by a trained operator via teleoperation, synchronously recording RGB videos from two camera perspectives at 30 fps, the complete proprioceptive state of the robotic arm (6-DOF joint states: shoulder_pan, shoulder_lift, elbow_flex, wrist_flex, wrist_roll, gripper), and action signals. Data collection follows a strict standardized protocol, including expert teleoperation, synchronous multi-modal capture, consistency checks, and episode result annotations (success/failure/recovery). The dataset adopts the Hugging Face LeRobotDataset v3.0 format, where each task has an independent directory containing data/ (frame-level states, actions, indices, etc. in Parquet format), videos/ (AV1-encoded MP4 videos from two perspectives, 480×640 resolution), and meta/ (self-descriptive metadata such as configuration, statistics, task descriptions, episode indices). The target number of episodes per task is more than 30, and diversity is enhanced through domain randomization (object position, pose, scene condition variations). The dataset is suitable for VLA end-to-end training, robot post-training, VLM pre-training/post-training, and world model learning. This repository is a public evaluation sample, licensed under CC BY-NC 4.0, for research and evaluation purposes only; the full catalog can be obtained via a commercial license.

创建时间:
2026-08-21
原始信息汇总

DecisionFacts Physical AI 数据集 — SO-101 机械臂遥操作数据

该数据集是物理AI机器人学习数据集,专门针对SO-101机械臂(so_follower)的遥操作数据进行整理,用于支持从模仿学习到VLA(视觉-语言-动作)模型及世界模型等各类机器人学习模型的训练与评估。

数据概览

  • 数据形式:人类专家通过遥操作完成的真实操作演示数据,非脚本或模拟生成
  • 数据规模:每个任务包含30条以上演示片段
  • 采集频率:30 fps 同步多模态采集
  • 多模态内容:双视角RGB视频、6自由度关节状态、控制信号
  • 数据格式:遵循 LeRobotDataset v3.0 标准格式

数据集结构

采用 Hugging Face LeRobotDataset v3.0 格式,相比于v2.1(每个片段一个文件),v3.0将多个片段打包到更少的大块文件中,通过关系型元数据而非文件名解析片段边界。

data/ — Parquet分片,包含帧级关节状态、动作、时间戳、索引 videos/ — 双视角同步视频流 observation.images.cam_front/ — 工作区正面视角 observation.images.cam_top/ — 工作区俯视视角(均为 480×640,30fps,AV1编码MP4,无音频) meta/ — 自描述元数据 info.json — 架构、fps、机器人类型、分块配置 stats.json — 特征归一化统计 tasks.parquet — 任务索引 → 自然语言任务描述 episodes/ — 每个片段的长度、任务引用、文件/字节偏移

关键说明:片段不是以独立文件夹或文件存储,而是合并到共享的、容量上限的分块文件中(chunk-000, chunk-001…),需要通过 meta/episodes/ 查询具体位置和帧偏移。

方法论

  • 领域随机化:在片段间变化物体位置、姿态和场景条件,降低下游模型对单一演示模式的过拟合
  • 标准化采集流程:统一采用30fps双摄同步、6自由度状态/动作记录、一致的任务语言标注
  • 人工质量审核:每位操作者演示并复核,确保片段反映一致、成功的任务执行
  • 结果多样性:除成功演示外,同时保留部分失败和恢复片段(带success/failure/recovery结果标签),供世界模型和鲁棒性训练使用

适用人群与场景

  • VLA模型训练 — 端到端训练从视觉和语言指令到机器人动作的映射策略
  • 机器人后训练 — 基于高质量演示数据微调预训练机器人策略/基础模型
  • VLM预训练/后训练 — 利用成对的视觉、语言与交互数据增强具身理解
  • 世界模型训练 — 从多视角同步视频和动作/状态序列学习环境与物体动力学预测

许可与访问

  • 公开评估样本采用 CC BY-NC 4.0 许可,仅限研究与评估用途
  • 商业使用、再分发或用于商业部署模型训练,需另行获取许可
  • 完整目录包含更多任务、更大的每任务片段数,并支持自定义采集
  • 可联系 info@decisionfacts.io 探讨试点批次(200-500个片段)、生产批次(1000+片段)或定制采集方案

引用信息

若使用该数据集,建议引用:

@misc{decisionfacts_teleops_dataset, credits = {Prabhu Raghav, Sreeram B Unni, Balamurugan Pandi, Sriram Gopalan}, year = {2026}, howpublished = {url{https://huggingface.co/datasets/DecisionFacts/Physical_AI_SO101_Cup_Nesting_Task}} }

搜集汇总
数据集介绍
Physical_AI_SO101_Cup_Nesting_Task 数据集图片
构建方式
该数据集通过专家遥操作的方式,在SO-101机械臂上采集了真实世界的操作演示数据。采集过程遵循严格的标准化协议,包括同步的多模态捕获,即以30帧/秒的频率记录关节状态、动作指令以及双摄像头RGB视频。每个episode均经过完整性、时间对齐及任务成功性的审查,确保数据质量。数据采用LeRobotDataset v3.0格式存储,将多个episode打包进分块文件,通过元数据索引解析episode边界,从而实现高效加载与流式访问。此外,数据采集引入了域随机化策略,变化物体位姿和场景条件,以增强策略的泛化能力,并包含成功、失败和恢复三类结果标签,为世界模型和鲁棒性训练提供丰富信号。
使用方法
该数据集适用于多种机器人学习场景。研究者可将其用于端到端VLA策略训练,利用视觉观测和语言指令直接映射到动作;也可用于微调预训练策略或基础模型,以适应新任务。数据中的多视角视频和动作序列可支持世界模型学习,预测环境动态。此外,任务描述与视觉数据的配对可用于视觉语言模型的预训练或后训练,提升具身智能的语言接地能力。数据集按任务独立存放,便于组合多个任务进行联合训练。用户需遵守CC BY-NC 4.0许可,仅限研究用途,商业使用需另行授权。
背景与挑战
背景概述
该数据集由DecisionFacts团队于2026年创建,核心成员包括Prabhu Raghav、Sreeram B Unni、Balamurugan Pandi和Sriram Gopalan。其核心研究问题在于为现代机器人学习模型——从模仿学习到视觉-语言-动作(VLA)模型及世界模型——提供高质量、多模态的真实遥操作数据。该数据集基于SO-101机械臂,采用LeRobotDataset v3.0格式,包含同步双视角RGB视频与完整的本体感知及控制信号。通过专家遥操作、域随机化及严格的质量控制流程,数据集在开放机器人学习生态系统中具有基准意义,为物理AI领域的研究提供了可复现、可扩展的数据基础,推动了机器人操作技能的泛化研究。
当前挑战
该数据集面临的挑战包括:其一,在构建过程中,需同时保证多模态数据的精确同步(如30fps下的视频与关节状态对齐),并应对域随机化带来的数据多样性需求,确保策略模型能泛化至未见场景;其二,领域方面,机器人操控任务本身面临复杂物体交互、动态环境及长时程决策的困难,数据需覆盖成功、失败及恢复等多样化结果,以支持鲁棒的世界模型与策略学习。此外,数据采集的成本与硬件兼容性,以及跨具身数据的缺乏,亦制约了数据集的规模扩展与通用性提升。
常用场景
经典使用场景
该数据集聚焦于SO-101机械臂的杯嵌套操作任务,通过人类遥操作采集了多模态演示数据,涵盖双视角同步视频、关节状态与动作指令。其标准化的LeRobotDataset v3.0格式确保了与开源机器人学习生态的无缝兼容,适用于模仿学习策略的端到端训练、视觉-语言-动作(VLA)模型的微调,以及世界模型的预测学习,为机器人操作研究提供了高保真、可复现的训练基准。
解决学术问题
该数据集解决了机器人操作学习中高质量演示数据稀缺与跨平台兼容性差的难题。通过领域随机化与失败/恢复轨迹的标注,它支持研究者探索策略的泛化性、鲁棒性及错误恢复机制,突破了传统固定场景演示的局限。其结构化元数据与标准化协议促进了研究结果的可比性与可重复性,为量化评估不同算法在物理世界中的表现提供了坚实的数据基础,推动了具身智能领域从仿真到现实迁移的研究范式演进。
实际应用
在实际应用中,该数据集可直接用于训练机械臂执行精密装配、物体分拣等工业自动化任务,降低编程成本并提升操作灵活性。同时,其VLA与VLM训练能力可赋能智能仓储、家庭服务机器人,使其根据自然语言指令自主完成物品整理等复杂操作。此外,双视角视频与状态数据的同步记录为远程监控与故障诊断提供了参考,有助于优化人机协作效率,加速机器人技术在柔性制造与日常服务场景中的落地部署。
数据集最近研究
最新研究方向
该数据集聚焦于机器人学习领域的前沿方向,特别是通过遥操作数据驱动视觉-语言-动作(VLA)模型与世界模型的训练与评估。数据采集严格遵循专家演示协议,涵盖多视角同步视频与完整的本体感知状态,为模仿学习提供高保真示范。其强调域随机化与失败/恢复情节的纳入,拓宽了策略学习的数据分布,增强模型的泛化能力。采用LeRobotDataset v3.0标准格式,实现高效存储与流式加载,与开放机器人学习生态无缝衔接。该数据集以低成本SO-101机械臂为载体,旨在推动具身智能研究,促进物理AI从仿真到现实的有效迁移,为构建可泛化的机器人操作策略奠定数据基础。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务