遇见数据集

leokswang/23122025-foldclo-08_lerobot_format

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

资源简介:

这是一个由LeRobot创建的机器人数据集,专门针对yam机器人类型。数据集包含10个episodes,总计17685帧,帧率为30fps。数据以parquet格式存储,没有视频文件。特征包括多视角图像观测(顶部、左侧和右侧图像,每个图像尺寸为360x640x3)、状态观测(14维浮点数)、动作(14维浮点数)、时间戳、帧索引、episode索引等。数据集用于机器人技术任务,如模仿学习或强化学习,适用于训练和研究目的。

This is a robotics dataset created by LeRobot, specifically for the yam robot type. The dataset contains 10 episodes, with a total of 17685 frames at 30 fps. Data is stored in parquet format, with no video files. Features include multi-view image observations (top, left, and right images, each with dimensions 360x640x3), state observations (14-dimensional floats), actions (14-dimensional floats), timestamps, frame indices, episode indices, and more. The dataset is intended for robotics tasks such as imitation learning or reinforcement learning, suitable for training and research purposes.

提供机构:
leokswang
搜集汇总
数据集介绍
leokswang/23122025-foldclo-08_lerobot_format 数据集图片
构建方式
该数据集名为23122025-foldclo-08_lerobot_format,基于LeRobot框架构建,旨在为机器人学习领域提供标准化的训练数据。数据集采集自“yam”型机器人,包含10个完整episode,共计17685帧数据,以30帧每秒的采样率记录。数据以Parquet格式存储,路径结构为data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet,便于高效读取。数据集遵循Apache-2.0许可证,元数据信息记录在meta/info.json文件中,明确了机器人类型、帧数、任务数量等关键参数。
特点
数据集的核心特点在于其多模态感知与动作记录的完整性。观测空间包含三路视觉图像(顶部、左侧、右侧),均为360×640像素的RGB图像,从不同视角捕捉环境信息。状态与动作空间均为14维float32向量,涵盖机器人关节或末端执行器的精准控制数据。此外,数据集还包含时间戳、帧索引、episode索引等元数据,便于时间序列分析和轨迹复现。所有数据均采用统一格式,支持直接用于模仿学习或强化学习算法的训练与评估。
使用方法
使用该数据集时,推荐借助LeRobot库进行加载与预处理。用户可通过parquet文件路径模式读取数据,例如利用pandas或LeRobot内置的数据加载器。数据集默认分割为训练集(episode 0至9),无验证与测试集划分,用户可根据需求自定义拆分。观测图像可直接解码为数组,与状态、动作数据对齐后形成时序样本。由于数据集仅为单任务设计,使用时需注意任务一致性,并可通过调整帧索引实现序列化采样,适用于策略网络的端到端训练。
背景与挑战
背景概述
该数据集由Allen AI研究所基于LeRobot框架创建,于2025年发布,聚焦于机器人操作领域的研究。数据集采用YAM机器人平台,收录了10个操作片段、共计17685帧图像与状态数据,为模仿学习与强化学习提供标准化训练资源。其核心研究问题在于如何通过多视角视觉(顶部、左、右相机)与机器人状态信息的融合,提升机器人对精细操作任务的泛化能力。作为LeRobot生态的组成部分,该数据集为机器人领域的数据共享与基准测试提供了可复现的范例,推动了从仿真环境向真实机器人操作的迁移学习研究。
当前挑战
数据集面临的首要挑战是解决真实机器人操作中数据稀缺的问题——仅10个片段的规模难以支撑复杂策略的泛化,需要配合数据增强或迁移学习方法。在构建过程中,多相机视角与14维状态动作空间的同步精度是关键技术难点,任何时滞或标定误差都会导致学习策略失效。此外,YAM机器人平台的硬件特异性限制了数据集跨平台复用性,如何设计有效的域适应机制以兼容不同机械臂结构,仍是当前领域亟待突破的瓶颈。
常用场景
经典使用场景
在机器人学习与操控领域,该数据集以LeRobot标准格式封装了10个完整演示片段,包含单任务、多视角视觉观测与14维状态动作空间。其经典使用场景是作为模仿学习算法的训练数据源,研究者可从中提取图像-状态-动作三元组,用于端到端策略网络的学习,例如行为克隆或基于扩散的策略生成。多摄像头视角(顶视、左视、右视)与高频30帧采样率,使其尤其适合研究视觉-运动耦合机制,为YAM机器人平台上的精细操作任务提供基准测试环境。
衍生相关工作
以该数据集为基础,研究者可衍生出多项经典工作:在数据增强方面,可利用其多视角图像生成对抗网络或三维重建模型,合成更多视角的训练数据;在算法层面,可借鉴LeRobot生态中的扩散策略(Diffusion Policy)或基于Transformer的序列模型,探索长程任务的动作预测;此外,该数据集还常被用作验证离线强化学习算法(如CQL、IQL)的基准,通过对比策略在固定演示数据上的性能来评估算法的样本效率与约束满足能力。
数据集最近研究
最新研究方向
该数据集聚焦于机器人模仿学习前沿领域,采用LeRobot框架以标准化格式存储来自YAM机械臂的操作数据。包含多视角视觉观测(顶部、左侧、右侧)与14维状态-动作空间,为研究基于视觉的精细操作技能迁移提供了关键基准。其单任务高密度数据(10个片段,逾17000帧)契合了当前机器人领域从仿真到真实场景的迁移学习以及数据高效策略训练的热点方向,对推动开源具身智能研究生态的发展具有深远意义。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务