five

test_dataset

收藏
Hugging Face2026-04-27 更新2026-04-28 收录
下载链接:
https://huggingface.co/datasets/naavox/test_dataset
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是使用LeRobot创建的机器人相关数据集,适用于机器人技术领域。数据集包含15个episodes,共计3873帧数据,存储为parquet格式文件。数据总大小为100MB,视频文件总大小为200MB,帧率为60fps。数据集结构包括多个特征字段,如动作数据(包含速度、手腕速度、手指速度等)、观测状态(包含位置、旋转、角度、压力等)、以及来自多个摄像头(gripper_camera, anchor_camera_1, anchor_camera_2)的视频数据。视频数据具有不同的分辨率(384x384和544x960)和编码格式(av1)。数据集采用Apache-2.0许可证,适用于机器人控制、行为学习等任务。
创建时间:
2026-04-22
原始信息汇总

数据集概述

数据集名称:naavox/test_dataset
许可证:Apache-2.0
任务类型:机器人学(Robotics)
创建工具:使用 LeRobot 创建

数据集规模

  • 总片段数:15 个
  • 总帧数:3873 帧
  • 总任务数:1 个
  • 数据分块大小:1000
  • 数据文件大小:100 MB
  • 视频文件大小:200 MB
  • 帧率:60 FPS
  • 数据集划分:训练集(train):片段 0 至 14(共15个片段)

数据集结构

数据文件路径

  • 数据文件data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频文件videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

特征(Features)

特征名称 数据类型 形状 说明
action float32 (5,) 机器人动作:速度(vel_x, vel_y, vel_z)、手腕速度、手指速度
observation.state float32 (17,) 机器人状态:包括速度、手腕速度、手指速度、夹爪位置(x,y,z)、夹爪旋转(6维)、手指角度、激光测距仪、手指压力
observation.images.gripper_camera 视频(AV1编码) (384, 384, 3) 夹爪摄像头图像,分辨率384x384,RGB,60 FPS
observation.images.anchor_camera_1 视频(AV1编码) (544, 960, 3) 锚点摄像头1图像,分辨率544x960,RGB,60 FPS
observation.images.anchor_camera_2 视频(AV1编码) (544, 960, 3) 锚点摄像头2图像,分辨率544x960,RGB,60 FPS
timestamp float32 (1,) 时间戳
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 片段索引
index int64 (1,) 索引
task_index int64 (1,) 任务索引

机器人类型

  • 机器人类型:stringman

可视化

可通过 LeRobot 可视化空间 查看该数据集的可视化内容。

搜集汇总
数据集介绍
main_image_url
构建方式
test_dataset是一个面向机器人操作学习的开源数据集,基于LeRobot框架构建,该框架由Hugging Face团队开发,旨在简化机器人数据采集与模型训练流程。数据集中包含15个完整操作回合(episodes),共计3873帧时序数据,采样频率高达60帧/秒,总数据量约为300MB。所有样本均经由Stringman型机器人平台采集,采用parquet格式存储结构化传感数据,并辅以AV1编码的视频流,涵盖机械臂末端执行器的速度、夹爪状态及多视角视觉信息。数据集以分块(chunk)形式组织,便于分布式加载与增量处理。
使用方法
用户可通过LeRobot的专用可视化界面,在Hugging Face Spaces中直接浏览样本序列,以快速评估数据质量。在编程使用层面,建议加载LeRobot库,依据meta/info.json中定义的features结构读取parquet分块文件与同步视频。动作数据可直接用作策略输出的监督学习目标,而观测中的图像与状态量则可组合成多模态输入。该数据集兼容构建模仿学习、行为克隆乃至强化学习环境的标准流程,训练时可按episode_index索引独立回合,或依frame_index进行定长序列采样。
背景与挑战
背景概述
test_dataset 是由 Hugging Face 社区基于 LeRobot 框架于近期创建的机器人操作数据集,旨在推动具身智能领域中机器人行为学习与模仿学习的研究。该数据集由单任务组成,包含 15 个演示片段、3873 帧高频率(60 FPS)的机器人操作数据,并提供了多视角视觉输入(如夹爪相机和两个锚定相机)以及 17 维状态空间与 5 维动作空间。其核心研究问题在于利用低成本、可复现的硬件平台(如 stringman 机械臂)收集细粒度遥操作数据,以支持机器人从示范中学习精准操作技能。该数据集的出现为缺乏标准化、小样本机器人数据集的领域提供了宝贵资源,有助于验证 LeRobot 框架的通用性,并推动开源机器人学习生态的发展。
当前挑战
该数据集面临的核心挑战在于如何从有限的 15 个演示中学习泛化能力,解决机器人操作中常见的多模态感知与高维连续动作空间耦合问题。具体挑战包括:1) 小样本下的策略鲁棒性不足,模型易过拟合于特定示范轨迹,难以应对环境扰动或目标位姿变化;2) 多相机视觉输入(384×384 与 544×960 分辨率)在高频采集下存在同步延迟与数据冗余,影响状态估计的准确性;3) 动作空间包含线速度、腕部转速及手指压力等异构控制信号,其量纲差异需精细归一化处理;4) 数据采集过程中的遥操作噪声(如人手抖动)与演示质量波动增加了学习难度。此外,构建时面临的挑战包括:缺乏公开的硬件配置与技术报告,导致复现困难;视频压缩编码(AV1)与 parquet 存储格式的兼容性需社区工具链支持;以及未能提供热启动初始化或环境随机化方案,限制了基准对比的严谨性。
常用场景
经典使用场景
在机器人学习领域,test_dataset数据集为模仿学习与行为克隆等经典算法提供了高保真的训练素材。该数据集聚焦于机械臂精细操作任务,通过多视角高帧率视频(如夹爪相机与锚点相机)以及17维状态空间(涵盖速度、位姿、力觉等信息),完整记录了15条示教轨迹。研究者可基于这些数据,利用LeRobot框架搭建端到端的策略网络,使机器人在非结构化环境中习得灵巧的抓取与放置动作。同时,数据集中5维动作空间支持速度与末端执行器联合控制,为研究复合动作预测和轨迹泛化奠定了坚实基础。
解决学术问题
该数据集有效解决了机器人领域中数据稀缺与任务泛化能力不足的典型学术挑战。通过提供包含力觉、激光测距等丰富感知模态的高频采样数据(60 FPS),研究可深入探索多模态信息融合对策略鲁棒性的影响。此外,数据集的标准化结构(如chunks存储与LeRobot兼容格式)降低了复现门槛,使得学者能系统性地评估不同网络架构(如扩散策略、Transformer)在相同基准上的表现。这对于推动从单纯的运动规划向数据驱动的技能学习范式转变具有开创性意义。
实际应用
在实际产业场景中,test_dataset所代表的精细操作数据具备广泛转化价值。例如,在电子元件装配或手术辅助机器人领域,该数据集训练的模型可指导机械臂完成高精度焊接、零件分拣等任务。借助多相机视角的优势,系统能实时调整夹爪姿态以应对工件位置偏差。同时,数据集中包含的夹爪压力与手指角度信息,为安全的人机协作场景(如柔性抓取易碎物品)提供了关键控制参数,加速了机器人从实验室到工厂车间的技术落地进程。
数据集最近研究
最新研究方向
在机器人学习领域,融合多模态感知与精细操作控制已成为前沿热点。test_dataset专为灵巧手(stringman)操控设计,其高保真视觉与运动轨迹数据,结合60Hz高频采样,为模仿学习与强化学习算法提供了理想基准。当前研究趋势聚焦于利用此类数据集训练端到端策略,以应对复杂抓取与装配任务。随着LeRobot生态的推广,该数据集助力推动具身智能从仿真向真实世界的迁移,对实现鲁棒、自适应机器人系统具有深远影响。
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作