遇见数据集

KS325/open-upper-drawer-r2-emp-test-spare

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

资源简介:

该数据集使用LeRobot工具创建,专注于机器人技术领域,具体涉及so_follower机器人类型。数据集包含2个总集数、1499个总帧数,数据以parquet格式存储,视频以mp4格式存储。特征包括动作(6维浮点数组,对应机器人关节位置)、观测状态(6维浮点数组,与动作相同)、两个摄像头图像(分辨率480x640,3通道,30fps视频)、时间戳、帧索引、集索引、索引和任务索引等。数据集用于机器人控制或模仿学习任务,支持训练和测试。

This dataset was developed using the LeRobot toolkit, focusing on the field of robotics and specifically targeting the so_follower robot type. It contains 2 total episodes and 1499 total frames. The feature data is stored in Parquet format, while the video data is saved in MP4 format. Its features include 6-dimensional floating-point action arrays (corresponding to robot joint positions), 6-dimensional floating-point observation state arrays (identical to the action arrays), two camera images (480×640 resolution, 3 color channels, 30fps frame rate), timestamps, frame indices, episode indices, sample indices, and task indices. This dataset is designed for robot control or imitation learning tasks, and supports both training and testing.

提供机构:
KS325
搜集汇总
数据集介绍
KS325/open-upper-drawer-r2-emp-test-spare 数据集图片
构建方式
该数据集依托LeRobot框架构建,聚焦于机器人操作任务中的上肢抽屉开启场景,采用so_follower型机械臂进行数据采集。构建过程以模仿学习范式为核心,通过遥操作或程序化控制方式驱动机械臂执行预定动作,同步记录多模态感知信息。数据以Apache-2.0协议开放,原始记录经标准化处理后划分为2个训练回合,总计涵盖1499帧、30fps的时序样本,并按照分块策略存储为Parquet格式,视频流则编码为AV1以压缩体积,最终形成结构化的机器人操作数据集。
特点
该数据集呈现出多模态、高时序密度的特征。每条样本融合六维关节位置动作与状态向量,以及双路480×640分辨率、30fps的视频观测,充分捕捉机械臂运动与视觉环境动态。数据以30Hz频率采样,保证动作序列的平滑性与连续性,适用于精细操作策略学习。元信息中明确标注机器人类型、特征维度与文件组织方式,并借由嵌套的chunk结构提升大规模数据读取效率,为机器人学习研究提供了标准化、可复现的数据基础。
使用方法
使用者可通过HuggingFace平台直接加载该数据集,借助LeRobot库提供的接口解析Parquet数据文件与视频资源。数据集划分为训练集,包含两个完整回合,支持模仿学习、行为克隆或视觉-动作联合建模等任务。研究者可利用observation.state与action字段进行策略训练,调用observation.images.camera1与camera2进行视觉编码,并借助timestamp及frame_index实现时序对齐。可视化工具亦被集成于数据页,便于直观检验视频与动作同步质量,从而加速机器人操作算法的迭代与评估。
背景与挑战
背景概述
机器人学习领域长期受限于高质量操作数据的稀缺,尤其在精细操控任务中,真实环境下的多模态示范数据采集成本高昂且标准化程度不足。在此背景下,open-upper-drawer-r2-emp-test-spare数据集依托LeRobot开源生态构建,聚焦于机械臂开启上层抽屉这一典型操作任务。数据集采用so_follower机器人平台,以30帧/秒的采样频率记录了两段完整回合,累计1499帧,涵盖六自由度关节位置、夹爪状态以及双路视觉观测。该数据集为模仿学习与视觉-动作策略研究提供了标准化的真实操作样本,其Apache-2.0许可亦促进了社区内的可复现研究与算法基准比较。
当前挑战
该数据集所应对的核心挑战在于:如何在非结构化真实场景中实现高精度的视觉引导操作,其中机械臂需协调肩部、肘部与腕部关节,并配合夹爪完成抽屉把手的精准接近、抓取与拉拽。构建过程中的挑战同样显著:一是数据采集需保证视觉帧与关节状态在时间维度上的严格同步,以避免策略学习中的因果混淆;二是仅有两个回合的样本量对模型泛化能力构成严峻考验,要求算法在极少示范下仍能提取有效的动作表征;三是视频编码采用AV1格式,需在解码效率与图像质量之间取得平衡,以支撑后续的实时策略推理。
常用场景
经典使用场景
在机器人学习与具身智能领域,open-upper-drawer-r2-emp-test-spare数据集主要服务于模仿学习与视觉-动作策略的验证。其经典使用场景聚焦于机械臂开门任务,特别是打开上层抽屉的操作。通过LeRobot框架采集的1499帧数据,涵盖双视角视觉观测与六维关节位置,研究者可训练策略网络从图像和状态中预测动作序列,实现端到端的抓取与牵拉行为。该数据集为少样本条件下的精细操作提供了标准化的评估环境。
衍生相关工作
基于该数据集,后续研究可能衍生出针对少样本模仿学习的改进算法,如结合元学习或对比预训练的视觉编码器;亦可能催生多任务抽屉操作基准,扩展至不同高度与材质的抽屉。此外,LeRobot生态中的可视化工具与策略评估协议常以此类数据集为范例,推动开源机器人学习社区的标准制定与模型可比性。
数据集最近研究
最新研究方向
在具身智能与机器人操作学习领域,基于视觉-动作联合建模的精细操作技能习得已成为前沿探索热点。该数据集依托LeRobot框架构建,聚焦于上层抽屉开启任务,通过双路RGB视频流与六自由度关节位置及夹爪状态的同步记录,为视觉-运动策略学习提供了细粒度时序对齐数据。当前研究趋势倾向于利用此类含多视角感知与连续动作标签的数据集,推动模仿学习、视觉-语言-动作模型在家庭服务场景中的泛化能力。抽屉开启作为典型关节物体交互任务,其操作学习直接关联到机器人对铰链约束的理解与力觉-视觉协同控制,对提升非结构化环境下的操作鲁棒性具有显著意义,尤其为少样本条件下灵巧操作策略的迁移与适应研究提供了关键基准。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务