遇见数据集

Gouspourd/test_06_setup_complet_20260529_032220

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

资源简介:

该数据集是一个机器人领域的数据集,使用LeRobot工具创建。它包含30个episodes,总计8258帧数据,以parquet格式存储。数据集特征包括动作(action,如机械臂关节位置)、观测状态(observation.state,如机械臂关节位置)、手腕摄像头图像(observation.images.wrist,分辨率为480x640)和顶部摄像头图像(observation.images.top,分辨率为720x1280),以及时间戳、帧索引、episode索引等元数据。机器人类型为“so_follower”,帧率为30fps。数据集主要用于机器人控制和视觉任务,数据文件总大小为100MB,视频文件总大小为200MB。

This dataset is a robotics dataset created using the LeRobot tool. It consists of 30 episodes with a total of 8258 frames, stored in parquet format. The dataset features include actions (e.g., robotic arm joint positions), observation states (e.g., robotic arm joint positions), wrist camera images (observation.images.wrist with resolution 480x640) and top camera images (observation.images.top with resolution 720x1280), along with metadata such as timestamps, frame indices, and episode indices. The robot type is so_follower, and the frame rate is 30fps. It is designed for robotics control and vision tasks, with a total data file size of 100MB and video file size of 200MB.

提供机构:
Gouspourd
搜集汇总
数据集介绍
Gouspourd/test_06_setup_complet_20260529_032220 数据集图片
构建方式
该数据集依托于LeRobot框架构建,通过采集机器人执行任务的原始轨迹数据,以30帧每秒的采样率记录。数据集共包含30个完整回合(episode),累计8258帧时序数据,所有数据均存储为Parquet格式的表格文件与AV1编码的MP4视频文件。特征空间由6维动作指令、6维关节状态观测、两个视角的RGB图像(腕部相机为480×640分辨率,顶部相机为720×1280分辨率)以及时间戳、帧索引等元信息构成。数据按照chunk分块存储,chunk大小为1000帧,便于流式加载和分布式处理。
特点
该数据集具有鲜明的多模态特征,融合了机器人的低维关节状态、高维视觉观测以及动作指令。腕部与顶部双摄像头设计提供了丰富的空间感知信息,视频采用AV1编码在压缩比与质量间取得平衡。数据集的机器人类型为so_follower,动作空间与观测空间维度一致均为6,有利于模仿学习中的状态-动作映射建模。所有序列均按回合组织,每个回合对应单一任务,且训练集与完整数据集重合,便于端到端训练与验证。此外,数据规模适中(数据文件约100MB,视频文件约200MB),适合快速原型验证。
使用方法
推荐使用LeRobot库加载该数据集,通过其内置的Dataset类可直接读取Parquet与视频文件。用户可依据元信息中的特征名称(如'observation.state'与'action')提取状态-动作对用于训练策略网络。视频数据可通过指定'observation.images.wrist'或'observation.images.top'键获取,并利用LeRobot提供的视频解码器进行高效加载。数据集已预定义训练集划分(索引0至30),支持按回合索引或帧索引进行迭代,适合用于模仿学习、行为克隆或离线强化学习等任务。手动处理时需注意视频解码依赖PyAV后端,并遵循AV1编码规范。
背景与挑战
背景概述
该数据集由Hugging Face社区成员基于LeRobot框架创建,聚焦于机器人操作任务的模仿学习领域。LeRobot作为近年来兴起的开源机器人学习平台,致力于为研究者提供标准化的数据采集与模型训练工具。此数据集创建于未来时间节点2026年5月29日,记录了一台SO-Follower机械臂完成单一操作任务(如抓取或放置)的30个完整回合,包含总帧数8258帧、约100MB的Parquet时序数据和200MB的高清视频数据(顶部与腕部双视角,帧率30FPS)。其核心研究问题在于验证LeRobot框架对真实机器人轨迹数据的兼容性与标准格式,为后续基于视觉-动作联合建模的仿人学习算法提供可复现的基准数据。该数据集因遵循Apache-2.0开源协议且依托LeRobot生态,预期将对机器人行为克隆、端到端操控等领域的社区研究产生示范性影响。
当前挑战
该数据集面临的核心挑战之一在于领域问题的复杂性:机器人操作任务要求模型从高维视觉输入中精确解耦机械臂的关节角度(6自由度)与末端执行器动作,而当前仅依赖30FPS的RGB图像与6维状态向量难以应对动态环境中的遮挡、光照变化及物理交互不确定性。构建过程中的挑战尤为突出:数据集虽采用标准化的LeRobot流水线,但info.json显示视频编码为AV1格式(crf=30, preset=12),这种高压缩比编码在限制存储空间的同时可能引入视觉质量损失,影响下游感知模型的特征提取精度;此外,数据集仅含单任务、30回合数据,样本规模不足以支持复杂泛化性训练,且缺乏对机器人-环境接触力、力矩等物理量的同步记录,限制了模仿学习向技能迁移、接触任务扩展的潜力。
常用场景
经典使用场景
在机器人学习领域,本数据集作为模仿学习研究的基础资源,记录了机器人完成特定任务时的关节动作序列与环境图像观察。其包含的6维动作空间(涵盖肩部、肘部、腕部及夹爪自由度)与同步采集的腕部、顶部视觉数据,为训练端到端的行为克隆模型提供了对齐的观测-动作对。研究者常借助此类数据训练机器人从示范中复现精细操作技能,例如物体的抓取与放置。
衍生相关工作
本数据集衍生出多项经典工作,包括利用其动作与图像模态开发基于LeRobot框架的模仿学习基线模型(如ACT与扩散策略)。研究者在该数据基础上改进数据增强方法以缓解多视角观测的遮挡问题,并探索将联合空间注意力机制引入策略网络以提升任务成功率。此外,部分工作通过重组其轨迹片段构建混合示范库,用于研究子技能解耦与组合式运动生成,为复杂长时程任务分解提供了验证载体。
数据集最近研究
最新研究方向
该数据集聚焦于机器人模仿学习领域,通过采集六自由度机械臂末端执行器的动作序列与多视角视觉观测数据,为行为克隆与视觉运动策略的学习提供了标准化benchmark。当前前沿研究主要集中在利用此类高保真演示数据集训练端到端控制模型,结合扩散策略或变换器架构实现精细操作技能的泛化,例如在灵巧抓取和装配任务中的突破。该数据集以高效压缩格式存储30FPS的深度码流视频与运动学数据,契合了LeRobot生态中关于低成本、可复现机器人学习的研究热点,其意义在于降低了非结构化环境下的数据获取门槛,推动从仿真到真实世界的零样本迁移与鲁棒控制策略的演进。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务