遇见数据集

jungjaeuk/so100_act_test_20260528_163849

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

资源简介:

该数据集是一个机器人控制数据集,使用LeRobot创建,专为机器人任务设计。数据集包含一个总任务,总帧数为2097,数据以30帧每秒(fps)的速率采集。特征包括动作(action)和观察状态(observation.state),均表示为6维浮点数组,对应机器人关节位置(如肩部平移、肩部提升、肘部弯曲、腕部弯曲、腕部滚动和夹持器位置)。其他特征包括时间戳、帧索引、剧集索引、索引和任务索引。数据以parquet格式存储,总数据文件大小为100MB,视频文件大小为200MB。机器人类型为so_follower,数据集结构支持训练分割(0:1)。

This dataset is a robotics control dataset created using LeRobot, designed for robotics tasks. It contains one total task with 2097 total frames, collected at a rate of 30 frames per second (fps). Features include action and observation.state, both represented as 6-dimensional float arrays corresponding to robot joint positions (e.g., shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper positions). Additional features include timestamp, frame index, episode index, index, and task index. Data is stored in parquet format, with total data file size of 100MB and video file size of 200MB. The robot type is so_follower, and the dataset structure supports a training split (0:1).

提供机构:
jungjaeuk
搜集汇总
数据集介绍
jungjaeuk/so100_act_test_20260528_163849 数据集图片
构建方式
该数据集基于LeRobot框架构建,专为机器人模仿学习任务设计。数据采集自so_follower型机器人,记录了单条完整操作轨迹,包含2097帧时序数据,按1000帧大小分块存储于Parquet格式文件中。视频数据则以MP4格式同步保存。数据集的元信息以JSON格式记录,明确标注了动作与观测状态的空间维度均为6维,对应机械臂各关节位置及夹爪状态。
特点
数据集的核心特色在于其精简而高效的结构设计。单条轨迹覆盖完整操作流程,便于快速验证算法。时序数据以30帧/秒的高采样率呈现,保证了动作的细腻度。特征空间包含动作、观测状态、时间戳等7类字段,其中动作与观测状态共享6维关节空间,简化了策略学习的映射难度。此外,数据集采用Apache-2.0开源协议,降低了使用门槛。
使用方法
使用LeRobot框架可直接加载该数据集,预设的splits配置将全部1个episode归入训练集。用户可通过HuggingFace提供的可视化工具在线预览数据,或利用Parquet与MP4文件进行本地开发。推荐用于行为克隆、逆强化学习等算法验证,由于数据量约为100MB(数据文件)+200MB(视频文件),适合在消费级GPU上开展实验。
背景与挑战
背景概述
在机器人学习领域,数据驱动的模仿学习已成为突破传统规划控制瓶颈的关键路径。so100_act_test_20260528_163849数据集由研究者jungjaeuk基于LeRobot框架创建,聚焦于六自由度机械臂的精细动作控制,采集时间暗示其可能服务于最新一代机器人感知-行动闭环的研究。该数据集以30Hz频率记录肩部、肘部、腕部及夹爪的连续位姿序列,共包含1个完整任务、1个回合的2097帧轨迹数据,为研究低样本条件下的机械臂行为克隆与动作泛化提供了标准化测试基准。其核心研究问题在于探索如何通过极少量演示数据,结合ACT(Action Chunking with Transformers)等算法实现高精度任务复现。作为LeRobot生态中的典型实例,该数据集推动了开源机器人数据采集规范(如固定分块大小、统一状态-动作特征命名)的建立,为后续跨机器人平台的动作表征迁移研究奠定了重要基础。
当前挑战
该数据集所面对的领域核心挑战在于:如何在仅包含单个任务回合的有限示教下,解决模仿学习中的动作分布外推与长期依赖问题。单次演示无法覆盖机械臂关节极限、动态碰撞规避等多模态环境状态,导致模型易陷入过拟合或产生灾难性遗忘。构建过程中面临的具体挑战包括:首先,30Hz高频采集下需要保证传感器时序同步精度,避免200MB视频流与100MB关节状态数据间的帧对齐误差;其次,单一回合中2097帧轨迹的片段切分策略(chunks_size=1000)需平衡动作块长度与模型注意力窗口的匹配;最后,开源框架下的数据标注缺失使得任务语义自动提取成为瓶颈,现行特征命名体系(如shoulder_pan.pos)虽具可读性,但难以直接映射至复杂空间操作要求,增加了下游模型设计时的特征工程负担。
常用场景
经典使用场景
so100_act_test_20260528_163849数据集是面向机器人模仿学习研究的典型数据资源,由LeRobot框架构建,专注于记录SO-100型机械臂的操作轨迹。该数据集包含单个任务、2097帧连续时间序列,每帧记录了六维关节动作指令与对应的观测状态,涵盖了肩部回转、肩部升降、肘部屈伸、腕部俯仰、腕部回转及夹爪开合等核心自由度。经典使用场景在于提供高保真的机器人操控时间序列数据,用于训练基于Transformer架构的动作生成模型(如ACT算法),实现从人类示教到机器人自主执行的端到端技能迁移。
解决学术问题
该数据集精准回应了机器人学习领域一个基础性难题——如何以简洁高效的数据形式表征和传递精细操控技能。它缓解了高维度连续动作空间中的小样本学习挑战,使研究者能够在单一任务种子下验证行为克隆、隐式策略(如策略熵正则化)等方法的收敛性与泛化能力。通过提供标准化、带有精确时间戳的关节级动作-状态对,该数据集为推动可复现的机器人操作学习基准奠定了数据基础,显著提升了学术实验在相同硬件条件下的可比性。
衍生相关工作
该数据集为多项机器人策略学习经典方法的验证提供了数据依托。基于其记录的动作与状态序列,派生工作包括行为克隆基准测试、基于扩散策略的动作生成模型对比评估,以及多任务元学习框架下的快速适应实验。LeRobot社区围绕此类数据集构建了标准化的训练流水线,衍生出如ACT、IBC(隐式行为克隆)等方法的开源实现,并促进了与RT-2、Octo等大型预训练模型在微调实验中的融合探索。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务