遇见数据集

zhang311/testlerobot3

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

资源简介:

该数据集由LeRobot创建,专注于机器人领域,包含一个双臂机器人(bi_so101_follower)的演示数据。数据集总计120个episodes,49486帧,以parquet格式存储数据,mp4格式存储视频。特征包括:12维动作(左右臂各6个关节的位置)、12维观测状态(与动作相同的关节位置)、三个视角的图像观测(左摄像头480x640分辨率,上摄像头360x640分辨率,右摄像头480x640分辨率,均为30fps RGB视频),以及时间戳、帧索引、episode索引等元数据。数据全部用于训练,无其他分割。

This dataset was created by LeRobot, focusing on the robotics domain, and contains demonstration data for a dual-arm robot (bi_so101_follower). It has a total of 120 episodes and 49486 frames. The data is stored in Parquet format, while the videos are stored in MP4 format. The features include: 12-dimensional actions (positions of 6 joints for each of the left and right arms), 12-dimensional observation states (same joint positions as the actions), image observations from three camera perspectives (left camera: 480×640 resolution, top camera: 360×640 resolution, right camera: 480×640 resolution, all 30fps RGB videos), as well as metadata such as timestamps, frame indices, and episode indices. All data is used for training with no additional data splits.

提供机构:
zhang311
搜集汇总
数据集介绍
zhang311/testlerobot3 数据集图片
构建方式
该数据集依托LeRobot框架构建,旨在服务于机器人模仿学习与操控任务。数据采集自一款名为bi_so101_follower的双臂机器人,共包含120个演示片段,总计49,486帧画面,帧率稳定维持在30fps。每条数据记录均以Parquet格式存储于分块文件中,而视觉观测数据则采用H.264编码的MP4视频形式保存,总数据量约300MB。数据集仅涵盖单一任务类别,并按照完整片段比例划分为训练集,未设置验证或测试子集,体现了针对特定机器人操控场景的精细化构建思路。
特点
数据集在结构设计上展现出高度的系统性与完备性。其动作空间与观测状态空间均包含12维连续值,精确反映双臂各关节的位置信息,尤其涵盖肩关节、肘关节及腕关节的旋转与抓取动作。视觉观测部分配置了三路摄像头——左侧、右侧及顶部视角,分别以640×480和640×360的分辨率同步记录场景信息,为多视角联合建模提供了先天优势。此外,数据中还附带时间戳、帧索引、任务索引等元信息,便于研究者进行时序建模与任务关联分析。
使用方法
该数据集可通过HuggingFace的datasets库直接加载并解析为标准化格式。用户需指定配置名称为'default',并指向包含Parquet文件与视频文件的目录结构。结合LeRobot工具链,研究者能够便捷地提取每帧的动作向量与观测状态,同时以视频流方式访问多视角图像序列。建议以片段索引为单位进行迭代,利用内置的帧统计信息构建批处理流水线。对于希望开展双臂机器人模仿学习或多模态策略训练的团队而言,该数据集提供了完备的输入输出接口与高效的存储方案。
背景与挑战
背景概述
随着机器人学习领域的飞速发展,对高质量、标准化数据集的需求日益迫切,以推动从模仿学习到强化学习等算法的训练与评估。testlerobot3数据集由Hugging Face社区基于LeRobot框架创建,旨在为双臂协作机器人操控任务提供结构化的演示数据。该数据集生成于2025年之前,汇集了来自bi_so101_follower型机器人的120个演示片段,总计近五万帧带时间戳的观察与动作序列,涵盖了左右双臂各六自由度的关节位置及多视角视觉信息。作为LeRobot生态的一部分,它遵循Apache-2.0许可开放使用,为研究者在真实机器人数据与仿真环境之间搭建桥梁,推动机器人技能学习领域的可重复性研究。
当前挑战
该数据集所解决的领域核心挑战在于双臂协作机器人精细操控任务的模仿学习,其动作空间维度高达12维,且需融合来自左、右、顶部三个摄像头的视觉流(分辨率高达640×480),对模型的时空对齐与多模态融合能力提出严峻考验。构建过程中,数据采集面临机器人硬件精度校准、多传感器同步采样(30帧/秒)、以及确保演示动作质量与任务一致性等难题;同时,将高维时序数据以Parquet格式高效存储并组织为1000帧/块的数据分块,还需兼顾视频编解码(H.264)带来的压缩失真与数据完整性,以及基于episode索引划分训练集时如何平衡样本多样性。
常用场景
经典使用场景
在机器人学习领域,testlerobot3数据集为模仿学习与行为克隆提供了高质量的演示数据。该数据集包含120个回合、近5万帧的机器人双臂操作记录,涵盖了左右臂各六个自由度的关节位置与姿态信息,以及来自左、右、顶部三个视角的同步视频流。研究者可据此训练机器人从人类示范中理解精细的操作策略,例如抓取、组装或搬运等任务,从而在仿真或真实环境中复现复杂的运动技能。
实际应用
在实际应用层面,testlerobot3数据集助力工业与服务机器人实现柔性化作业。例如,其双臂协同数据可用于开发自动生产线上的精密装配系统,或家政机器人中餐具整理、物品分拣等日常任务。此外,结合视觉与关节反馈的同步数据,还能为远程操控、人机协作提供训练样本,推动机器人在医疗辅助、仓储物流等领域的落地部署,提升作业效率与安全性。
衍生相关工作
依托此数据集,衍生出了多项经典工作,涵盖基于扩散策略的动作生成、Transformer架构的决策模型以及对比学习在状态表示中的应用。其中,以LeRobot框架为核心的后续研究探索了数据增广与域随机化技术,增强了策略在不同环境间的鲁棒性。这些工作不仅验证了数据集的高效性,还催生了开箱即用的基线模型,为机器人学习社区提供了可复现的评估标准,加速了算法迭代与知识共享。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务