遇见数据集

nitred/so101-combined-25ep_1_20260530_181453

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

资源简介:

该数据集是一个机器人学数据集,使用LeRobot创建,专门用于机器人控制和学习任务。数据集包含25个episodes,总计9762帧数据,以30帧每秒(fps)的速率记录。数据特征包括动作(action)和观测状态(observation.state),其中动作和状态都包含6个浮点数值,分别对应机器人关节位置(如肩部平移、肩部提升、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置)。观测部分还包括两个视角的图像数据:顶部摄像头图像(observation.images.top)和腕部摄像头图像(observation.images.wrist),每个图像的分辨率为480x640像素,具有3个颜色通道(RGB),以视频格式存储,使用AV1编解码器。数据集还包含时间戳、帧索引、episode索引、任务索引等元数据。数据以parquet文件格式组织,视频文件以mp4格式存储,总数据大小约为100MB,视频文件大小约为200MB。机器人类型为so_follower,数据集仅包含训练分割(train split),涵盖所有25个episodes。该数据集适用于机器人视觉-动作映射、强化学习或模仿学习等研究。

This dataset is a robotics dataset created using LeRobot, designed for robot control and learning tasks. It contains 25 episodes with a total of 9762 frames, recorded at 30 frames per second (fps). The features include actions and observation states, where both actions and states consist of 6 floating-point values corresponding to robot joint positions (e.g., shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper positions). Observations also include image data from two perspectives: top camera images (observation.images.top) and wrist camera images (observation.images.wrist), each with a resolution of 480x640 pixels and 3 color channels (RGB), stored in video format using the AV1 codec. The dataset includes metadata such as timestamps, frame indices, episode indices, task indices, etc. Data is organized in parquet file format, with video files stored in mp4 format. The total data size is approximately 100MB, and video files size is approximately 200MB. The robot type is so_follower, and the dataset only includes a training split covering all 25 episodes. This dataset is suitable for research in robot vision-action mapping, reinforcement learning, or imitation learning.

提供机构:
nitred
搜集汇总
数据集介绍
nitred/so101-combined-25ep_1_20260530_181453 数据集图片
构建方式
该数据集的构建依托于LeRobot框架,以SO-100型机器人遥操作数据采集为基础,通过记录25个完整回合(episode)的操作序列,生成共计9762帧的连续运动数据。数据以Parquet格式存储结构化信息,包括六维关节动作指令(肩部、肘部、腕部及夹爪位置)与对应观测状态,同时同步采集了顶部和腕部两个视角的640×480分辨率视频流,视频采用AV1编码并以30帧/秒的帧率保存为MP4文件。数据集遵循Apache-2.0开源协议发布,其元信息文件清晰定义了特征类型与编码参数,为机器人模仿学习研究提供了标准化的多模态训练材料。
使用方法
用户可通过LeRobot库加载此数据集,利用其内置的数据加载器自动解析Parquet表格与关联视频文件。读取时需指定数据集标识符'nitred/so101-combined-25ep_1_20260530_181453',系统将根据meta/info.json中的配置按批次返回动作、观测状态及图像帧。为便于快速浏览,数据集主页提供了HuggingFace可视化接口,支持在线查看轨迹回放。研究人员可基于返回的'observation.images.top'和'observation.images.wrist'图像数据联合六维动作向量,构建从视觉输入到运动输出的端到端控制策略,或利用'timestamp'与'episode_index'字段进行时序一致性验证与跨回合分析。
背景与挑战
背景概述
在机器人学习领域,模仿学习(Imitation Learning)因其能够直接从人类示教中获取行为策略而备受关注,尤其在复杂操作任务的泛化与迁移中扮演关键角色。so101-combined-25ep_1_20260530_181453数据集由Hugging Face社区依托LeRobot框架创建,旨在为机器人灵巧操作任务提供标准化的训练与评估基准。该数据集聚焦于单任务场景,涵盖25个完整回合、总计9762帧的高频(30FPS)动作与状态序列,并配备来自顶置与腕部双视角的640×480视觉输入,为多模态策略学习提供了丰富且结构化的数据样本。作为开源机器人数据集家族的一员,其采用Apache-2.0许可,有助于促进学术界与工业界在6自由度机械臂操作子任务上的协同研究,推动从仿真到真实环境的知识迁移。
当前挑战
当前机器人操作领域面临的核心挑战在于如何从有限且高维的示教数据中提取稳健的策略,以应对动态环境下的泛化需求。so101-combined-25ep_1_20260530_181453数据集在此背景下需解决两大难题:一方面,领域问题要求模型能够从6维连续动作空间与多视角视觉流中学习精确的位姿映射,并克服末端执行器(如夹爪)在精细操作中的非线性控制与状态耦合;另一方面,数据构建过程中需应对示教轨迹的噪声扰动(如人为抖动)、有限回合数(25集)导致的数据稀疏性问题,以及多模态数据(视觉、关节状态、时间戳)的时空对齐挑战,同时确保30FPS的视频流在存储与压缩编码(AV1, CRF30)下不丢失关键操作细节,以维持模仿学习的保真度。
常用场景
经典使用场景
在机器人学习领域,模仿学习作为一项核心技术,其发展高度依赖于高质量的示范数据集。so101-combined-25ep_1_20260530_181453数据集正是为此而生,它聚焦于“so_follower”型机械臂的操作任务,通过LeRobot框架采集并整理了25个完整操作回合的高保真数据。该数据集的经典使用场景在于端到端的行为克隆训练,研究者可利用其包含的机械臂各关节位置状态与双视角视觉观测(顶部与腕部摄像头),构建从视觉输入到连续动作指令的映射模型,使机器人能够复现复杂的精细操作技能,如抓取、放置或装配等任务。
解决学术问题
该数据集精准回应了机器人学中一个核心挑战:如何在有限示范数据下实现高效且泛化的技能学习。它提供了结构化的多模态信息——6维关节动作、状态时间序列及480×640像素的同步视频流,这解决了以往研究中数据格式不统一、传感器信息缺失或采样频率不一致等痛点。通过apache-2.0许可开放共享,该数据集促进了模仿学习算法的可重复性验证,为探索少样本学习、领域随机化及跨任务迁移等前沿问题奠定了坚实的实验基础,显著推动了具身智能研究的规范化进程。
实际应用
在实际产业应用中,该数据集展现了巨大的转化潜力。其采集的so_follower机械臂操作数据可直接用于工业流水线上的视觉引导抓取、精密装配以及物料分拣等自动化场景。借助此数据集训练出的策略模型,能够使机器人通过视觉反馈自适应调整关节角度,从而在非结构化环境中稳定执行操作。此外,该数据集对腕部与顶部视角的联合利用,使得模型可同时获取全局环境信息与局部操作细节,这在医疗手术辅助、家庭服务机器人以及危险环境远程操控等对操作精度与安全要求极高的任务中尤具应用价值。
数据集最近研究
最新研究方向
该数据集聚焦于机器人操作技能的模仿学习与数据驱动控制,基于LeRobot框架构建,记录了SO-100机械臂在25个回合中的精确关节动作、状态信息及视觉观测(包含顶部与腕部相机),为细粒度机器人操控研究提供了多模态数据支撑。当前前沿方向集中于利用此类高质量演示数据集训练端到端的模仿学习模型,以提升机械臂在复杂场景下的泛化能力与鲁棒性;同时,结合大规模预训练视觉-语言模型与视觉运动对齐技术,推动机器人从简单任务复现向语义理解与适应性决策演进。该数据集的发布顺应了近年来机器人领域对数据规模化与标准化需求的浪潮,尤其为低成本科研平台(如降本增效的开放性机械臂系统)提供了可复现的基准数据,加速了具身智能的通用化进程,对促进机器人算法从实验室走向真实应用具有里程碑意义。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务