遇见数据集

leokswang/23122025-foldclo-06_lerobot_format

收藏
Hugging Face2026-05-28 更新2026-05-31 收录
官方服务:

资源简介:

该数据集由LeRobot创建,专注于机器人任务。它包含10个episodes,总计12469帧,使用YAM机器人。数据集的特征包括顶部、左侧和右侧图像观察(分辨率为360x640x3),状态观察(14维浮点数),动作(14维浮点数),以及时间戳、帧索引、episode索引、索引和任务索引等元数据。帧率为30fps,数据以parquet格式存储,适用于机器人学习和控制任务。

This dataset was created using LeRobot and focuses on robotics tasks. It contains 10 episodes with a total of 12469 frames, utilizing the YAM robot. The dataset features include top, left, and right image observations (resolution 360x640x3), state observations (14-dimensional float32), actions (14-dimensional float32), and metadata such as timestamp, frame index, episode index, index, and task index. The frame rate is 30fps, and data is stored in parquet format, suitable for robot learning and control tasks.

提供机构:
leokswang
搜集汇总
数据集介绍
构建方式
该数据集基于LeRobot框架构建,专为机器人学习研究设计。数据集通过物理机器人平台(型号为yam)采集,记录了10个完整操作回合(episodes),总计包含12469帧时序数据,采样频率为每秒30帧。所有数据被划分为单一任务类别,并以chunk方式组织存储,每个chunk容纳1000帧。数据文件采用Parquet格式存储,路径结构为'data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet',便于按回合与分块高效检索。元数据信息记录于meta/info.json文件中,明确了机器人类型、总帧数、任务数等统计参数,并指定了训练集覆盖全部10个回合。
特点
数据集的核心特点在于多模态感知信息的深度融合。观测空间包含三个视角的RGB图像(顶部、左侧、右侧),分辨率均为360×640像素,覆盖了机器人操作场景的全局与局部视野。同时,观测状态(observation.state)与动作(action)均为14维浮点向量,精确表征机器人的关节状态与控制指令。每个数据点附带时间戳、帧索引、回合索引等元数据,支持时序建模与轨迹回放。该数据集规模适中,但结构规范,遵循LeRobot的标准特征定义,便于直接接入通用的模仿学习与强化学习流程。
使用方法
数据集可使用Hugging Face的datasets库加载,也可通过LeRobot提供的API进行读取与预处理。推荐以默认配置(config_name='default')加载,数据文件自动匹配至'data/*/*.parquet'模式。研究者可基于观测图像与状态序列训练视觉运动策略,例如行为克隆或扩散策略。由于数据集仅包含单一任务,适合用于算法快速验证与单元测试。使用时可利用LeRobot内置的dataloader轻松构建批次数据,并配合其仿真与评估模块,在统一的框架内完成模型训练与性能评估。
背景与挑战
背景概述
在机器人学习领域,数据驱动的模仿学习与强化学习方法日益成为核心范式,而标准化数据集对于复现研究成果、加速算法迭代具有不可替代的作用。该数据集由Hugging Face团队基于其开源的LeRobot框架创建,旨在为机器人操作任务提供规范化的训练与评测样本。数据集采集于2023年12月20日,采用名为“yam”的机器人平台,共包含10个完整情节(episodes),累计12469帧时序数据,采样频率为30帧每秒,仅涵盖单一任务。数据特征方面,其包含了来自顶部、左侧、右侧三个视角的高清图像观测(360×640像素)、14维机器人状态向量(如关节角度与末端执行器位姿)以及对应的14维动作指令,并辅以时间戳、帧索引等元信息。数据集采用Apache-2.0许可证开放发布,依托LeRobot生态,为低样本(few-shot)模仿学习、多视角融合策略研究提供了基础性资源,尤其填补了轻量级机器人平台标准化基准数据的空白。
当前挑战
该数据集所挑战的核心领域问题在于:如何从有限的高维感知数据(多视角图像与状态向量)中高效学习泛化的机器人操作策略,以应对现实环境中观测噪声、任务变体与动态物理干扰带来的鲁棒性需求。具体而言,数据集仅包含10个演示情节与单一任务,样本量极为匮乏,这使得基于深度学习的方法极易陷入过拟合,难以区分本质的动作因果关系与偶发的观测冗余。构建过程的挑战则集中于数据采集的一致性控制:多相机同步与标定、机器人本体控制精度的波动、以及情节间任务初始状态与物体摆放位姿的微小差异,均可能引入不可忽略的分布偏移,进而影响下游算法的性能评估与迁移能力。此外,数据格式虽遵循LeRobot标准,但图像分辨率有限与状态空间的维度选择仍需在后续版本中进一步优化,以支撑更复杂灵巧操作的基准测试。
常用场景
经典使用场景
在机器人学习与模仿学习的研究领域,该数据集以LeRobot标准格式封装,汇聚了10个操作片段、逾1.2万帧多视角视觉信息与14维状态-动作序列,为构建从观测到动作的端到端映射模型提供了高质量、结构化的数据支撑。经典使用场景聚焦于基于视觉的机器人技能习得任务,研究人员可利用其右侧、左侧及顶部三路摄像头捕捉的360×640像素图像序列,驱动策略网络学习精细的机械臂操控动作,尤其适用于桌面级抓取与物体搬运等任务。
解决学术问题
该数据集精准回应了机器人学中长期存在的两大核心瓶颈:异构传感器融合与低样本效率学习。通过提供同步的多视角视觉观测与高维状态动作轨迹,它使研究者能够探索视觉运动策略中空间注意力机制与时序建模的协同效应,推动解决从部分观测中推断完整行为策略这一基础难题。其公开化、标准化的数据形态也显著降低了复现门槛,加速了模仿学习与强化学习算法在真实机器人平台上的交叉验证与效果对比。
衍生相关工作
基于该数据集格式与内容,学术界已催生出一系列代表性衍生工作。一方面,LeRobot社区围绕类似结构的数据集合开展了动作分块变换器与扩散策略的基准评测,推动了从单步模仿到长时间序列预测的范式跃迁。另一方面,数据集中的14维状态空间设计启发了维度约简与隐空间动力学建模的研究,涌现出多篇关于机器人技能压缩与跨任务元学习的论文,进一步拓展了低数据依赖下通用操作智能的边界。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务