遇见数据集

nitred/so101-combined-25ep_1_20260530_182958

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

资源简介:

该数据集是一个机器人学数据集,使用LeRobot创建。它包含来自25个episodes的总计11252帧数据,涉及一个任务(task_index为1)。数据集结构包括动作数据(action),表示机器人6个关节的位置(如肩部平移、肩部抬升、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置);观测数据(observation),包括机器人状态(与动作相同的关节位置)以及来自顶部摄像头(observation.images.top)和腕部摄像头(observation.images.wrist)的视频观测,每个视频帧分辨率为480x640,3通道,帧率为30fps,使用AV1编解码器。此外,数据集还包含时间戳、帧索引、episode索引、索引和任务索引等元数据。机器人类型为so_follower。数据以parquet文件格式存储,视频以mp4格式存储。数据集用于训练或评估机器人控制模型。

This dataset is a robotics dataset created using LeRobot. It contains a total of 11252 frames from 25 episodes, involving one task (task_index of 1). The dataset structure includes action data representing the positions of 6 robot joints (such as shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper positions); observation data, which includes robot state (same joint positions as actions) and video observations from a top camera (observation.images.top) and a wrist camera (observation.images.wrist), each video frame with a resolution of 480x640, 3 channels, a frame rate of 30fps, using the AV1 codec. Additionally, the dataset contains metadata such as timestamp, frame index, episode index, index, and task index. The robot type is so_follower. Data is stored in parquet file format, and videos are stored in mp4 format. The dataset is intended for training or evaluating robot control models.

提供机构:
nitred
搜集汇总
数据集介绍
nitred/so101-combined-25ep_1_20260530_182958 数据集图片
构建方式
so101-combined-25ep_1_20260530_182958数据集基于LeRobot框架构建,专注于机器人操作任务的仿真数据采集。该数据集通过25次完整交互轨迹(episodes)收集而成,共计11252帧图像和动作序列。数据以Parquet格式存储结构化信息,包括机器人关节状态(如肩关节、肘关节、腕关节及夹爪的位置)与高分辨率视觉观测(640×480像素的顶部和腕部摄像头图像)。视频采用AV1编码压缩,帧率为30FPS,确保在保留细节的同时优化存储效率。数据集被划分为单个训练集(前25个episodes),便于直接用于模仿学习或强化学习模型的训练。
使用方法
使用该数据集需依赖LeRobot库及其生态工具。用户可通过Hugging Face Datasets库加载数据,或利用LeRobot内置的Visualize Dataset空间在线预览轨迹回放。在代码中,数据集支持按episode索引抽取完整的动作-观测对,或按帧索引提取单步数据。推荐的典型用法是构建PyTorch DataLoader,将图像序列与动作张量对齐后输入策略网络(如扩散策略或Transformer-based模型)。需注意视觉数据以压缩视频格式存储,训练前应解码为原始RGB张量(H×W×C,值域[0,255]),并通过预定义的归一化参数进行标准化。数据集的Parquet表格还包含时间戳、任务索引等辅助字段,可用于过滤特定子集或实现时序数据增强。
背景与挑战
背景概述
该数据集由Hugging Face社区的研究者基于LeRobot框架创建,旨在为机器人操作学习提供标准化数据资源。数据采集于2026年5月30日,聚焦于单一操作任务,包含25个示范片段,总帧数达11252帧。数据集记录的机器人类型为so_follower,动作空间涵盖六自由度关节位置与夹爪控制,同时提供来自顶部和腕部摄像头的640×480分辨率视频流。作为开放机器人学习生态系统的一部分,该数据集降低了机器人模仿学习的研究门槛,为评估算法在真实机械臂操控中的泛化能力提供了基准。
当前挑战
当前机器人操作学习面临的核心挑战是数据异构性与泛化困难。该数据集虽提供了精细化的电机状态和视觉信息,但单一任务、单一机器人平台的配置限制了模型在多变场景下的适应性。构建过程中,数据采集的复现成本高昂,高精度运动轨迹与同步视频的标注需要大量人力与时间投入。此外,30Hz的采样率与有限的25个示范片段可能无法覆盖复杂操作中的动态变化,导致模型在面对未见过物体或环境扰动时出现性能退化,成为从示范中高效学习的瓶颈。
常用场景
经典使用场景
so101-combined-25ep_1_20260530_182958数据集是面向机器人操作领域精心构建的仿真与真实混合数据资源,其核心用途在于训练和评估基于视觉与状态信息的机器人操控策略。该数据集以25个完整回合(episode)的高频采样序列为基础,记录了SO系列机械臂在执行单一任务时的关节动作与双视角视觉观测,包括自上而下的全局视角与腕部近距视角。研究者借助此数据集可模拟机器人在精密操作场景下的感知-决策闭环,从而开展模仿学习、行为克隆以及基于强化学习的控制策略训练。其结构化的动作空间(6自由度关节位置与夹爪控制)与规整的观察模态,为验证运动规划算法在真实噪声环境中的鲁棒性提供了标准化测试平台。
解决学术问题
该数据集在学术层面主要回应了机器人学习领域面临的样本效率与泛化性两大核心挑战。通过提供高频率、多模态的真实操控轨迹,它解决了传统仿真环境难以捕捉的物理动力学细节(如摩擦、柔性变形)所导致的Sim-to-Real迁移困难。研究者得以利用该数据探索如何在有限示范数据下(仅25回合)通过数据增强或表征学习提升策略的样本效率。同时,数据集内一致的观测结构使学术界能够系统性地量化视觉特征对策略泛化能力的影响,为理解端到端学习在非结构化环境中的失效边界提供了关键实证基础,推动了机器人小样本学习与鲁棒控制理论的演进。
实际应用
在实际产业应用层面,该数据集为工业机器人的自适应编程与快速部署提供了数据基石。例如,在电子元件组装或精密仪器调试场景中,机械臂需根据实时视觉反馈调整夹爪位姿与力度。数据集中的腕部与顶部摄像机记录的高清视频流,可直接用于训练机器人在非结构化工件摆放条件下的视觉伺服控制器。此外,其Apache-2.0许可协议降低了商业应用门槛,使得智能仓储中包裹分拣、柔性制造中的零件抓取等场景可借助该预训练模型大幅减少手动编程工作量,加速从实验室原型到工厂产线的转化周期。
数据集最近研究
最新研究方向
该数据集聚焦于机器人操作技能的模仿学习领域,通过LeRobot框架采集了25个完整操作回合的演示数据,包含6自由度机械臂的关节角度动作与双视角视觉观测(顶部与腕部摄像头)。当前前沿研究多以此类高保真、多模态的实操数据为基础,结合扩散策略或基于Transformer的神经架构,推动灵巧操作技能从简单抓取向复杂序列任务迁移。数据集以每30帧的精细时间分辨率记录状态-动作对,使得研究者能够端到端地训练出可复现精细动作的智能体,这对于实现精准装配、柔性操控等工业场景具有里程碑式的意义。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务