遇见数据集

KS325/open-lower-drawer-r2-emp-val

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

资源简介:

该数据集是一个机器人学数据集,使用LeRobot工具创建,专门用于机器人控制任务。数据集包含4个总集、2999个总帧和1个总任务,数据以parquet格式存储。机器人类型为so_follower,特征包括动作(如肩部平移、肩部提升、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置)、观察状态(关节位置)、两个摄像头(camera1和camera2)的视频图像(分辨率为480x640,30fps,编码为av1),以及时间戳、帧索引、集索引、索引和任务索引等元数据。数据集适用于机器人学习和控制研究,例如模仿学习或强化学习。

This is a robotics dataset developed with the LeRobot toolkit, tailored specifically for robotic control tasks. It contains 4 total datasets, 2999 total frames, and 1 overall task, and is stored in Parquet format. The robotic platform type is so_follower. The dataset's features include actions (such as shoulder translation, shoulder elevation, elbow flexion, wrist flexion, wrist rotation, and gripper position), observation states (joint positions), video images from two cameras (camera1 and camera2) with a resolution of 480×640, 30 frames per second, encoded using AV1, as well as metadata including timestamps, frame indices, dataset indices, individual sample indices, and task indices. This dataset is applicable to robotics learning and control research, such as imitation learning or reinforcement learning.

提供机构:
KS325
搜集汇总
数据集介绍
KS325/open-lower-drawer-r2-emp-val 数据集图片
构建方式
该数据集的构建依托于LeRobot框架,专注于机器人操作技能的学习与验证。构建过程采集了SO-100 follower机械臂在执行开启下层抽屉任务时的多模态数据,通过遥操作或预设策略生成动作序列,并同步记录关节位置与视觉观测。数据以30帧每秒的频率采样,涵盖4个独立回合,总计2999帧,单任务设定确保了数据的一致性。原始数据经标准化处理后,以Parquet格式存储动作与状态信息,视频流则采用AV1编码压缩为MP4文件,最终按照LeRobot v3.0规范组织为分块结构,便于高效加载与训练。
特点
该数据集聚焦于精细操作任务,其核心特征在于多维度的状态与动作空间。动作和观测状态均包含6个自由度,涵盖肩部旋转、肩部升降、肘部弯曲、腕部弯曲、腕部旋转及夹爪位置,精准刻画了机械臂的运动学特性。同时,数据集集成了双路视觉观测,两路摄像头均以480×640分辨率、30帧每秒的速率采集RGB图像,为策略学习提供了丰富的环境感知信息。此外,数据以分块形式组织,每块最大100MB,视频文件限制在200MB以内,辅以元数据索引,确保了数据管理的轻量化与可扩展性。
使用方法
使用该数据集时,研究者可借助LeRobot库直接加载Parquet格式的数据文件与对应的视频流,通过解析meta/info.json获取完整的特征定义与分块路径。数据以训练集形式提供,覆盖全部4个回合,用户可按照data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet的路径模板读取动作、状态及时间戳等字段,并利用video_path模板提取视觉观测。为便于直观验证,数据集提供了在线可视化工具,用户可通过Hugging Face Spaces链接查看任务执行过程。在模型训练中,该数据集适用于模仿学习或强化学习算法的验证,尤其适合评估机械臂在抽屉开启任务上的策略泛化能力。
背景与挑战
背景概述
在具身智能与机器人学习领域,高质量的真实世界操作数据集是推动策略泛化与算法迭代的核心基石。open-lower-drawer-r2-emp-val数据集依托LeRobot框架构建,聚焦于SO系列机械臂执行开启下层抽屉这一典型家庭服务任务,记录了涵盖关节位置与多视角视觉观察的完整示范轨迹。该数据集以30帧每秒的采样频率提供2999帧、4个回合的演示数据,其发布顺应了机器人学习社区对标准化、可复现操作数据日益增长的需求。尽管目前尚未标注具体的研究机构与发表文献,其采用的Apache-2.0许可与LeRobot生态的紧密集成,为后续研究者提供了便捷的基准测试与算法对比平台,对推动精细操作技能的学习与迁移具有积极的实证参考价值。
当前挑战
该数据集所应对的领域问题在于,如何在有限示范样本条件下实现机械臂对下层抽屉开启这一接触密集型任务的稳定学习与泛化。抽屉开启过程涉及复杂的接触动力学、遮挡条件下的视觉感知以及精确的力觉与位置协调,对策略的鲁棒性构成显著挑战。在构建层面,仅四个回合的演示数据虽然降低了采集成本,却对数据增强与策略样本效率提出了更高要求;同时,多相机视角的同步采集、动作空间与观察状态的一致性对齐,以及真实机器人操作中不可避免的噪声干扰,均增加了数据集构建的技术难度。如何在保持数据真实性的同时,为社区提供足以支撑可靠评估的基线性能,仍是该数据集后续发展需要持续回应的关键问题。
常用场景
经典使用场景
在机器人学习与具身智能领域,基于视觉的精细操作技能习得始终是核心议题。open-lower-drawer-r2-emp-val数据集依托LeRobot框架构建,以so_follower机械臂为平台,采集了四段总计2999帧、30fps的演示数据,涵盖双路视觉观测与六维关节位置及夹爪状态,辅以对应动作序列。其最经典的使用场景在于模仿学习范式下的视觉-动作映射训练,即通过同步的视频流与本体感觉信号,驱动策略网络习得开启下方抽屉这一长程操作技能,并借助验证集评估策略在未见初始位形下的泛化表现。
解决学术问题
该数据集直面机器人模仿学习中演示数据稀缺与策略泛化评估困难两大痼疾。在学术研究中,低维状态与高维视觉的异构融合常导致策略过拟合于训练轨迹,而该数据集通过提供独立的验证划分(train: 0:4),为检验策略在时序与空间扰动下的鲁棒性创造了条件。其意义在于,以轻量级演示样本推动了小样本模仿学习算法的发展,并为人-机器人技能迁移中的视觉表征学习与动作分块建模提供了可复现的基准,对厘清示教数据规模与策略性能之间的标度关系具有实证价值。
衍生相关工作
以该数据集为基石,衍生工作主要围绕LeRobot生态展开。一方面,研究者利用其验证集构建了模仿学习策略的泛化性评测协议,推动了动作分块Transformer与扩散策略在精细操作中的对比研究;另一方面,该数据集启发了基于多视角视觉融合的示教增强方法,例如通过视角随机遮蔽提升策略对相机位姿变化的鲁棒性。此外,其数据格式与HuggingFace数据集库的无缝集成,促进了开放机器人数据标准的形成,为后续跨本体技能迁移与少样本元学习研究提供了可复用的数据基座。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务