遇见数据集

2026-04-21_direction_2-lerobot-without-rinse

收藏
Hugging Face2026-05-12 更新2026-05-15 收录
官方服务:

资源简介:

该数据集是一个机器人操作数据集,使用LeRobot工具创建。数据集包含61个完整episodes,总计31777个数据帧,帧率为50fps。数据以Parquet文件格式存储,总数据量约为200MB,并包含等量的视频数据(200MB)。数据集专门用于训练任务。数据内容涵盖一个双机械臂(ALOHA类型)系统的多模态观测和动作记录。观测数据包括:14维的关节状态(位置)、关节速度、关节力矩;以及来自四个摄像头的视觉观测:高位摄像头、低位摄像头、左手腕摄像头和右手腕摄像头,所有视频均为640x480分辨率、3通道的RGB格式。动作数据为14维的关节控制指令。数据集还包含丰富的元数据,如任务索引、子任务描述、时间戳、帧索引、episode索引等。该数据集适用于机器人模仿学习、强化学习、行为克隆等任务,特别是涉及双机械臂视觉运动控制的研究。

This dataset is a robotic manipulation dataset created using the LeRobot tool. It contains 61 full episodes, totaling 31,777 data frames at a frame rate of 50 FPS. Stored in Parquet file format, the raw dataset is approximately 200MB, with an accompanying video dataset of the same size (200MB). It is explicitly designed for robotic training tasks. The dataset covers multimodal observations and action recordings of a dual-manipulator (ALOHA-type) robotic system. Observation data includes: 14-dimensional joint states (positions), joint velocities, and joint torques; as well as visual observations from four cameras: a top camera, a bottom camera, a left wrist camera, and a right wrist camera. All videos are in 3-channel RGB format with a resolution of 640×480. The action data comprises 14-dimensional joint control commands. The dataset also contains comprehensive metadata including task indices, subtask descriptions, timestamps, frame indices, episode indices, and more. This dataset is applicable to research on robotic imitation learning, reinforcement learning, behavior cloning, and other related tasks, especially studies involving dual-manipulator visual motion control.

创建时间:
2026-04-29
搜集汇总
数据集介绍
2026-04-21_direction_2-lerobot-without-rinse 数据集图片
构建方式
该数据集基于LeRobot框架构建,采用ALOHA机器人平台采集,共包含61个有效回合(episode),总计31777帧数据。数据以分块形式存储,按每1000帧切分为多个chunk,对应parquet文件保存结构化状态与动作信息,同时以MP4视频文件记录四路摄像头(cam_high、cam_low、cam_left_wrist、cam_right_wrist)的视觉观测,所有数据均遵循Apache-2.0开源许可协议。
特点
数据集特点在于其多模态与高频率特性:采样频率达50帧/秒,每帧包含14维机器人关节状态(如腰部、肩部、腕部及夹爪位置)、14维动作指令、14维速度与力矩信息,以及四路640×480分辨率彩色视频流。所有回合对应单一任务,且均被标记为训练数据,结构紧凑、维度丰富,适合用于模仿学习与机器人控制策略的端到端训练。
使用方法
使用过程中,可通过LeRobot库直接加载该数据集。用户需指定配置名'default',自动读取data/目录下所有parquet文件作为结构化数据,videos/目录下视频文件则作为图像观测。数据集已按'0:61'划分为训练集,无验证与测试分割,为简化流程,可直接调用LeRobot的DataLoader接口进行批次迭代,适用于强化学习与行为克隆等算法的模型输入准备。
背景与挑战
背景概述
该数据集由Hugging Face的LeRobot社区创建,于2026年4月21日发布,专注于双机械臂操控任务的模仿学习研究。核心研究问题在于如何利用高频率(50帧/秒)的多视角视觉与状态动作数据,训练机器人完成精细的物体操作任务。数据集采用A LOHA机器人平台,收集了61个示范片段,共计31777帧,涵盖14维关节状态、动作、速度及力矩信息,并包含四个视角的高清视频(480×640像素)。作为LeRobot生态系统的一部分,该数据集旨在推动机器人学习领域的标准化与可复现性,为基于视觉-运动耦合的模仿学习算法提供高质量基准。其贡献在于弥合仿真与真实场景之间的鸿沟,促进机器人从人类演示中自主习得复杂技能的能力。
当前挑战
领域问题方面,该数据集主要应对机器人模仿学习中的‘数据饥渴’与‘领域泛化’挑战。尽管提供了高密度的状态与动作序列(如14维关节控制信号),但仅涵盖单一任务(direction_2)和61个片段,难以支撑策略模型对未见过场景的鲁棒泛化。构建过程中,多视角视频与传感器数据流(50帧/秒)的同步采集是显著难题,需确保机械臂运动学与视觉图像在时间戳上的精确对齐;同时,200MB的视频与数据文件量虽适中,但高频采集产生的冗余帧和噪声可能引入动作偏差,影响学习效果。此外,数据仅包含训练集而无验证/测试划分,增加了模型过拟合评估的难度,亟需更精细的数据增强与跨任务迁移策略来突破性能天花板。
常用场景
经典使用场景
在机器人学习领域,模仿学习是使机器人从人类示范中习得复杂操作技能的关键范式。该数据集依托LeRobot框架,采集自双臂协作机器人平台Aloha,包含61个完整操作片段与逾三万个时空稠密的高频帧序列,完美记录了机器人执行单一任务时关节状态、速度、力矩以及多视角视觉观测的完整演化历程。其最经典的使用场景是作为行为克隆与逆强化学习的训练样本库,研究者可从中提取状态-动作映射关系,训练端到端的策略网络,实现对精细化双手协同操作动作的精准复现。
实际应用
从真实环境部署的视角审视,该数据集为机器人灵巧操作技能的实际落地提供了关键支撑。基于其训练的策略可直接应用于工业精密装配、柔性物件处理以及医疗辅助操作等场景中,例如双机械臂协同完成零件贴合或线缆插拔。由于数据采集方案贴近真实部署条件,模型无需繁琐的仿真迁移步骤即可在实体机器人上运行,大大缩短了从实验室原型到生产线应用的技术转化周期。
衍生相关工作
该数据集的出现催生了一系列围绕双臂协同模仿学习的前沿探索。基于其结构化的高频序列数据,衍生出了结合隐式运动规划与扩散策略的高精度操作模型,以及融合光学流与立体视觉的空间感知架构。此外,研究人员还据此开发了面向非静态环境下的在线微调方法,在不遗忘先前技能的前提下主动适应动态扰动。这些工作不仅验证了数据集本身的通用性,更推动了LeRobot生态下标准化数据驱动范式的成熟与普及。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务