遇见数据集

hongj97/lift_nnk1_500_nomix

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

资源简介:

该数据集是一个机器人数据集,使用LeRobot创建,专门用于机器人任务。数据集包含500个episodes,总计109212帧,数据以parquet格式存储,视频以mp4格式存储。机器人类型为so101_follower,特征包括8维动作向量、6维状态观察(如关节位置:肩部平移、肩部提升、肘部弯曲、手腕弯曲、手腕旋转和夹爪位置),以及图像观察:手腕摄像头和前置摄像头的视频,分辨率均为480x640,3通道,帧率为30fps,使用AV1编解码器。此外,还包括时间戳、帧索引、episode索引、总索引和任务索引等元数据。数据集仅用于训练,无其他分割,总数据文件大小为100MB,视频文件大小为200MB。

This dataset is a robotics dataset created using LeRobot, specifically designed for robotics tasks. It contains 500 episodes with a total of 109212 frames, stored in parquet format for data and mp4 format for videos. The robot type is so101_follower, and features include an 8-dimensional action vector, 6-dimensional state observation (e.g., joint positions: shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper position), and image observations: videos from a wrist camera and a front camera, both with a resolution of 480x640, 3 channels, a frame rate of 30fps, using the AV1 codec. Additionally, metadata such as timestamp, frame index, episode index, overall index, and task index are included. The dataset is intended for training only, with no other splits, and has a total data file size of 100MB and video file size of 200MB.

提供机构:
hongj97
搜集汇总
数据集介绍
hongj97/lift_nnk1_500_nomix 数据集图片
构建方式
该数据集基于LeRobot框架构建,旨在为机器人操控学习提供高质量的训练数据。通过操控so101_follower机器人,采集了500个完整操作回合(episode),共计109,212帧数据,其中每个回合包含从起始状态到任务完成的完整动作序列。数据以Parquet格式存储,分为100个数据块(chunk),每个数据块体积约1 MB,便于分布式读取与训练。视频数据独立存储为MP4文件,采用AV1编码压缩,以30帧/秒的速率记录机器人腕部和前方两个视角的高清影像(480×640分辨率),有效兼顾了数据精度与存储效率。
使用方法
数据集可通过Hugging Face的LeRobot库直接加载,利用其标准API读取Parquet数据与视频文件。用户可基于`action`字段作为策略输出目标,以`observation.state`与`observation.images`作为输入,训练端到端的操控模型。建议采用固定窗口长度的序列采样方式(如chunk size=1000),将连续帧组织为子序列(SubSequenceDataset)进行批处理。由于未预设验证集,使用者需自行划分数据(如留出10%回合作为评估),并注意视频数据需解码为像素张量后与状态向量拼接,形成统一的多模态输入特征。
背景与挑战
背景概述
在机器人学习领域,模仿学习与行为克隆方法的发展高度依赖大规模、高质量的数据集,以训练出能够泛化至复杂真实环境的控制策略。lift_nnk1_500_nomix数据集由Hugging Face社区依托LeRobot框架创建,旨在为SO101系列机器人(尤指follower型号)提供标准化演示数据。该数据集通过单任务设定(total_tasks为1),收录了500个演示片段(total_episodes),总计超过十万帧(109212帧)的序列信息,涵盖8维动作空间与6维关节状态观测,并辅以腕部与前方双视角高清视频(480x640,30 FPS)。其发布体现了开源社区对机器人基础数据集构建的推动,为多模态机器人控制研究提供了可复现的基准素材,降低了入门门槛并促进了领域内的可对比性。
当前挑战
该数据集所面临的挑战首先体现在机器人学习领域共性问题上:单任务、有限演示数据(500个片段)虽利于算法基准测试,却难以支撑策略在面对环境扰动与新场景时的泛化需求;动作与状态维度较低(8维动作、6维状态),对复杂操作任务的表达能力不足。在构建层面,数据收集需要稳定的远程操作或运动规划系统以确保500次演示的一致性,同时多视角视频与同步状态信息的采集对硬件同步与存储提出了要求;训练/验证采用固定切分(0:500全为训练集),缺乏独立测试集评估过拟合与分布外表现,且视频采用AV1编解码格式,可能增加后续数据加载与实时处理的延迟。
常用场景
经典使用场景
在机器人学习与具身智能领域,lift_nnk1_500_nomix数据集为基于视觉的机械臂操作任务提供了丰富的训练素材。该数据集记录了SO101型从动机械臂在500个回合中执行单一提升任务的高频时序数据,包含腕部和前部双摄像头视角的640×480像素视频流,以及8维动作指令与6维关节状态信息。研究者常利用该数据集训练模仿学习或强化学习模型,通过观察视觉输入与动作序列的映射关系,使机械臂习得从初始状态到目标抓取姿态的精确控制策略。数据集以30帧/秒的采样率和1000帧的分块结构存储,便于部署在LeRobot等主流机器人学习框架中进行批量训练与评估。
解决学术问题
该数据集的核心价值在于填补了视觉-运动控制领域高质量标杆数据的空白。以往机械臂操作研究常受限于数据规模小、传感器配置单一或动作模态缺失等问题,而lift_nnk1_500_nomix通过标准化格式提供了视觉流与关节角度的同步记录,使研究者能够系统地探究多模态感知融合对操作精度的影响。它支持对比不同动作表征方法(如端到端控制与分层策略)在提升任务上的泛化能力,并为验证数据增强技术(如视角扰动或时序裁剪)对鲁棒性的贡献提供了基准。此数据集的公开也促进了可重复性研究,降低了机器人学实验的复现门槛。
实际应用
在工业与家庭服务场景中,基于该数据集训练的模型可迁移至实际机械臂的精细操作任务。例如,在自动化产线上,机械臂通过学习数据集中的提升动作模式,能够适应不同重量或材质的工件抓取,降低对预先编程的依赖;在辅助生活领域,搭载此类模型的机器人可协助完成物品取放、桌面整理等日常操作。双摄像头设计增强了模型对空间深度与遮挡情况的感知,使其在动态环境中仍能保持较高的任务成功率。数据集的开源特性还允许厂商根据自身机械臂构型进行微调,加速了从仿真实验到实体部署的转化进程。
数据集最近研究
最新研究方向
在机器人学习领域,lift_nnk1_500_nomix数据集以其标准化、多模态的采集协议,正成为模仿学习与深度强化学习研究的关键基石。该数据集基于so101_follower机器人平台,汇聚了500条完整轨迹,涵盖8维动作空间与6维关节状态,并同步提供高清腕部与前方视觉观测,为视觉-运动联合建模提供了高保真的异质数据源。当前前沿方向集中于利用此类规模化演示数据,结合扩散策略或基于Transformer的序列模型,探索从人类示教到机器人泛化技能的端到端映射,尤其关注训练数据分布外(OOD)场景下的鲁棒控制与行为克隆偏差缓解。这一数据集的出现,极大推动了低样本、跨任务迁移学习在精细操作场景中的实证研究,其影响深远,有望加速人机协作与柔性制造等实际应用的落地。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务