five

leokswang/22122025-foldclo-07_lerobot_format

收藏
Hugging Face2026-05-28 更新2026-05-31 收录
下载链接:
https://hf-mirror.com/datasets/leokswang/22122025-foldclo-07_lerobot_format
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是一个机器人数据集,专门用于YAM机器人,使用LeRobot工具创建。它包含10个episodes,总计15028帧数据,帧率为30fps。数据以parquet格式存储,并包括多个特征:顶部、左侧和右侧的图像观察(分辨率为360x640,3通道),状态观察(14维浮点数),动作(14维浮点数),以及时间戳、帧索引、episode索引等元数据。数据集仅包含训练分割(episodes 0到10),许可证为Apache-2.0。

This dataset is a robotics dataset specifically for the YAM robot, created using the LeRobot tool. It contains 10 episodes with a total of 15028 frames at a frame rate of 30fps. The data is stored in parquet format and includes multiple features: image observations from top, left, and right views (resolution 360x640, 3 channels), state observations (14-dimensional float32), actions (14-dimensional float32), and metadata such as timestamp, frame index, and episode index. The dataset only includes a training split (episodes 0 to 10) and is licensed under Apache-2.0.
提供机构:
leokswang
搜集汇总
数据集介绍
main_image_url
构建方式
本数据集基于LeRobot框架构建,面向机器人操纵任务,收录了由YAM机器人采集的10个完整演示片段,共计15,028帧数据,帧率为30 FPS。数据存储采用Parquet格式,按分块索引存放于data/chunk-{episode_chunk:03d}/目录下,每个片段独立编号为episode_{episode_index:06d}.parquet,便于高效加载与索引。元信息通过meta/info.json文件定义,明确标注了机器人类型、任务数量及数据拆分方式,其中全部10个片段均归入训练集。
特点
该数据集的核心特点在于多模态感知与动作空间的统一编码。观测空间包含三路视觉图像(顶部、左、右),分辨率均为360×640像素的RGB图像,可提供丰富的环境立体视角。状态与动作空间均为14维浮点向量,涵盖机器人关节角度或末端执行器位姿等信息。此外,数据集记录了时间戳、帧索引、片段索引等结构化元字段,支持时序建模与轨迹回放。数据集采用Apache 2.0许可证发布,适用于学术研究与工业应用。
使用方法
使用者可通过LeRobot库直接加载该数据集,利用其提供的Dataset API访问parquet文件中的各特征字段。典型用法包括:调用load_dataset函数指定数据集名称与配置,获取迭代器遍历每一帧的观测图像、状态及动作数据。由于数据已按片段组织并分块存储,可方便地用于模仿学习、行为克隆或离线强化学习算法的训练与评估。所有图像数据为原始像素值,无需额外预处理即可输入视觉编码器。
背景与挑战
背景概述
该数据集由Allen Institute for AI(AI2)于2022年12月20日创建,采用LeRobot框架进行构建,专注于机器人操作领域的研究。数据集围绕YAM机器人平台,记录了10个完整回合、共计15028帧的精细操作数据,涵盖来自顶部、左侧和右侧三个视角的视觉图像及14维状态与动作信息。其核心研究问题在于为机器人模仿学习与行为克隆提供高质量的示范数据,推动从感知到动作的端到端策略学习。作为开源可复现的数据集,它采用Apache-2.0许可,为机器人学习社区提供了标准化的训练与评估基准,尤其在多视角视觉-运动联合建模方面具有重要参考价值。
当前挑战
该数据集首先需解决机器人领域中示范数据稀缺与多样性不足的根本问题,仅包含单一任务与10个回合限制了策略的泛化能力。构建过程中,精确同步三个摄像头的高帧率图像与14维动作状态流是一大技术挑战,需确保时间戳对齐与数据完整性。此外,从原始传感器数据中提取并标准化为LeRobot格式,涉及多模态数据的高效存储与索引设计,例如Parquet格式的分块处理与视频路径的映射关系,以支持大规模分布式训练。最终,数据集缺乏公开论文与评估指标,使得其相较于IMAGENET等基准在影响力与标准化方面仍面临诸多挑战。
常用场景
经典使用场景
在机器人学习与操作领域,该数据集依托LeRobot框架构建,专注于YAM型机器人平台。其经典使用场景在于为模仿学习与行为克隆算法提供高质量的示范数据。数据集中包含了多视角视觉观测(顶部、左、右相机)以及14维的机器人状态与动作序列,采样频率为30帧每秒,涵盖10个完整回合,共计约1.5万帧。这使得研究者能够在真实机器人操作任务中,利用该数据集训练端到端的视觉运动策略,如从像素到关节指令的直接映射,进而完成抓取、放置等精细操控。
实际应用
在实际产业应用中,该数据集可为装配流水线上的YAM型机器人提供快速任务适配的范例。例如,结合LeRobot生态,工程师能利用该数据集离线训练初始策略,再通过少量在线微调完成新零件的分拣或组装任务。多视角图像的引入使机器人对环境光照和遮挡具有更强的鲁棒性,适用于仓储物流中的柔性抓取场景。此外,数据集的结构化存储格式(Parquet)与无视频文件的轻量化设计,便于在边缘计算设备上进行快速部署与策略更新。
衍生相关工作
围绕该格式的数据集,社区已衍生出一系列经典工作。在LeRobot框架下,研究者开发了基于扩散策略与Transformer的动作序列预测模型,利用此类多回合示范数据实现复杂操作的联合建模。同时,数据增强技术与域随机化方法被引入以提升策略对未见场景的适应能力。此外,有工作探讨了跨不同YAM型机器人的策略迁移,通过对齐状态与动作空间的表征,实现了从本数据集到其他同构平台的知识共享,为大规模机器人技能库的构建提供了范本。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务