遇见数据集

ZuoShun-AI/demo_clean_repo

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

资源简介:

这是一个机器人学数据集,使用LeRobot创建,专门用于aloha类型机器人的研究。数据集包含50个episodes,总计5660帧,覆盖47个不同任务,数据以parquet格式存储,帧率为50fps。数据集结构丰富,包括状态观测(14维浮点数,对应左右机械臂的关节状态,如腰部、肩部、肘部、前臂滚动、手腕角度、手腕旋转和夹爪)、动作(同样14维,控制机械臂的相同关节)、以及三个摄像头图像:高角度摄像头、左腕摄像头和右腕摄像头(均为3通道、480x640分辨率图像)。此外,还包含时间戳、帧索引、episode索引、索引和任务索引等元数据。所有数据按chunk组织,用于训练任务。

This dataset is a robotics dataset created using LeRobot, specifically for research on aloha type robots. It contains 50 episodes, totaling 5660 frames, covering 47 different tasks. The data is stored in parquet format with a frame rate of 50fps. The dataset structure is comprehensive, including state observations (14-dimensional float32 values corresponding to joint states of the left and right arms, such as waist, shoulder, elbow, forearm roll, wrist angle, wrist rotation, and gripper), actions (also 14-dimensional, controlling the same joints), and images from three cameras: high-angle camera, left wrist camera, and right wrist camera (all with 3 channels, 480x640 resolution). Additionally, metadata such as timestamp, frame index, episode index, index, and task index are included. The data is organized into chunks and is intended for training purposes.

提供机构:
ZuoShun-AI
搜集汇总
数据集介绍
ZuoShun-AI/demo_clean_repo 数据集图片
构建方式
在机器人学习领域,高质量示范数据集的构建通常依托标准化框架以保障可复现性与互操作性,demo_clean_repo 即借助 LeRobot 工具链完成创建。该数据集以 aloha 双臂机器人为采集平台,围绕 47 项任务收集了 50 个回合的示范轨迹,累计涵盖 5660 帧观测与动作记录。数据按 chunk 分块存储为 parquet 文件,并同步生成包含机器人类型、帧率、特征维度等元信息的 info.json 文件,从而在数据组织层面实现了统一规范,为后续的模型训练与评估奠定了结构化基础。
使用方法
使用该数据集时,可依托 LeRobot 提供的加载接口,依据 info.json 中声明的数据路径模板读取 parquet 文件,并借助 splits 字段划分训练集。模型输入可组合 observation.state 与三路 observation.images,输出则对齐 action 维度,以进行模仿学习或策略回归。timestamp、frame_index、episode_index 与 task_index 等字段可用于时序对齐、回合边界识别及任务条件建模。若需加载视频模态,可依据 video_path 模板访问对应 mp4 文件,从而支持端到端视觉-动作策略的训练与评测。
背景与挑战
背景概述
机器人学习领域长期受限于高质量、多模态演示数据的匮乏,尤其在双臂精细操作场景中,真实世界轨迹的采集与标准化始终是制约算法泛化的瓶颈。demo_clean_repo数据集依托LeRobot框架构建,由Hugging Face社区于近年发布,聚焦ALOHA双臂机器人平台,收录50条演示轨迹、5660帧数据,覆盖47项操作任务,每帧同步记录14维关节状态与动作向量,以及三路RGB视觉观测。该数据集的核心研究问题在于为模仿学习与视觉-运动策略提供结构统一、语义清晰的开源基准,其标准化特征命名与分块存储设计对推动可复现的机器人学习研究具有基础性影响。
当前挑战
该数据集所应对的领域问题在于双臂协同的精细操作策略学习,要求模型从高维视觉与本体感知中解耦左右臂协调模式,并泛化至未见任务配置。构建过程中的挑战则体现为多源异构数据的时空对齐:三路摄像头与关节编码器需在50Hz频率下实现帧级同步,任务语义标注须在47类操作间保持一致性,而ALOHA平台固有的机械间隙与传感噪声亦对轨迹平滑性构成干扰。此外,仅50条演示轨迹的有限规模难以充分覆盖任务空间分布,对策略的鲁棒性评估构成显著约束。
常用场景
经典使用场景
在机器人学习与具身智能研究领域,基于示教数据的模仿学习是赋予机械臂复杂操作能力的核心范式。demo_clean_repo数据集依托ALOHA双臂机器人平台,通过遥操作方式采集了50个任务片段、共计5660帧的同步多模态数据,涵盖14维关节状态与动作序列、三路视觉观测(顶部及左右腕部相机),以及任务索引与时间戳等元信息。其最经典的使用场景在于为行为克隆(Behavior Cloning)与扩散策略(Diffusion Policy)等模仿学习算法提供高质量的训练语料。研究者可依托该数据集构建从视觉观测到双臂协同动作的端到端映射模型,在精细操作任务中验证算法对高维连续动作空间的建模能力。
解决学术问题
该数据集有效缓解了机器人操作研究中真实世界示教数据稀缺、多模态同步困难与任务多样性不足等常见问题。在学术层面,其解决了双臂协调控制中高维动作空间探索效率低下的难题,为策略泛化性评估提供了标准化基准。同时,数据集涵盖47项不同任务,使得研究者能够系统探究任务条件对策略学习的影响,推动少样本泛化与多任务迁移学习的发展。其意义在于构建了从数据采集到算法验证的闭环研究基础设施,为具身智能领域的可复现研究奠定了重要基础,对推动机器人操作从单臂向双臂协同的范式跃迁具有深远影响。
实际应用
在实际应用层面,该数据集可直接服务于家庭服务、仓储物流与精密装配等场景中的双臂机器人技能开发。通过对数据集中抓取、放置、开合等操作模式的学习,机器人能够获得在非结构化环境中执行复杂任务的能力,例如协助人类完成餐具整理、物品分拣或工具传递等日常作业。其多视角视觉观测与关节状态同步记录的特性,使得训练所得策略具备较强的环境适应性与鲁棒性,可迁移至真实机器人平台部署。此外,该数据集亦可用于工业场景中重复性操作任务的自动化,降低人工示教成本,加速智能机器人在柔性制造中的落地进程。
数据集最近研究
最新研究方向
在具身智能与机器人学习领域,基于LeRobot框架构建的demo_clean_repo数据集正推动着双臂协作操作策略的前沿探索。该数据集以aloha机器人平台为依托,涵盖50个演示回合、5660帧、47项任务,提供14维关节状态与动作空间以及三路视觉观测(顶部与双腕摄像头),其结构化设计契合当前模仿学习与视觉-语言-动作模型对多模态时序数据的需求。近期研究聚焦于利用此类高保真双臂演示数据训练端到端策略,以提升复杂环境下精细操作任务的泛化能力,尤其在少样本学习与跨任务迁移方面展现出显著潜力,为通用机器人操作系统的构建提供了关键数据支撑。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务