遇见数据集

MrPsarik/rollout_SortToy_20260527_143413

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

资源简介:

该数据集是一个机器人学相关数据集,使用LeRobot创建。它包含42个总集数,24932个总帧数,帧率为30 fps。数据特征包括动作(action)和观测(observation),动作和观测状态均包含6个浮点型关节位置(如肩部平移、肩部提升、肘部弯曲、手腕弯曲、手腕旋转、夹爪位置)。观测还包括来自前置和顶部摄像头的图像数据,每个图像分辨率为480x640,3通道,视频编码为av1。数据集还包含时间戳、帧索引、集索引、索引和任务索引等元数据。数据以parquet文件格式存储,视频以mp4格式存储,总数据文件大小为100 MB,视频文件大小为200 MB。机器人类型为so_follower,并包含训练集划分。

This dataset is a robotics-related dataset created using LeRobot. It contains 42 total episodes, 24932 total frames, with a frame rate of 30 fps. The features include action and observation, where both action and observation state consist of 6 floating-point joint positions (e.g., shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, gripper). Observations also include image data from front and top cameras, each with a resolution of 480x640, 3 channels, and video codec av1. The dataset also includes metadata such as timestamp, frame index, episode index, index, and task index. Data is stored in parquet file format, videos in mp4 format, with total data file size of 100 MB and video file size of 200 MB. The robot type is so_follower, and it includes a training split.

提供机构:
MrPsarik
搜集汇总
数据集介绍
MrPsarik/rollout_SortToy_20260527_143413 数据集图片
构建方式
该数据集名为rollout_SortToy_20260527_143413,基于LeRobot框架构建,专为机器人操作任务设计。数据集包含42个任务片段(episodes),总计24,932帧,采样频率为30帧/秒。数据通过so_follower机器人采集,涵盖6维动作空间(包括肩部、肘部、腕部及夹爪位置)与对应的6维状态观测。视觉信息由前置与俯视两个摄像头同时记录,生成分辨率480×640的AV1编码彩色视频流。所有数据以Parquet格式存储压缩表格,视频文件则独立保存为MP4格式,两者通过索引文件关联,确保时间对齐的一致性。
使用方法
使用该数据集时,建议采用LeRobot框架的DataLoader进行高效加载。用户可通过指定数据集路径与配置名称(default)自动解析Parquet表格中的动作、状态及索引信息,并流式读取MP4视频帧到NumPy数组。由于数据结构遵循LeRobot标准格式(包含episode_index、frame_index等关键索引字段),开发者可直接集成到Python代码中,用于训练行为克隆(Behavior Cloning)、扩散策略(Diffusion Policy)等主流模仿学习算法。注意需安装OpenCV等视频解码库以支持AV1格式的视频处理。
背景与挑战
背景概述
该数据集名为rollout_SortToy_20260527_143413,由研究者在LeRobot框架下创建,旨在为机器人操作任务提供基准数据。数据集聚焦于玩具分拣场景,记录了机械臂执行分类任务的完整轨迹,涵盖关节位置、动作指令及多视角视觉信息。其30帧每秒的高频采样与超过24000帧的规模,为模仿学习与强化学习算法提供了高保真的训练素材。作为开源资源,该数据集以Apache-2.0协议发布,推动了机器人领域从仿真到真实环境迁移的研究进程,尤其为精细操作策略的泛化能力评估奠定了重要基础。
当前挑战
当前面临的核心挑战包括:领域问题层面,玩具分拣涉及非刚性物体识别与动态抓取规划,机械臂需应对不同形状、材质的玩具,这对视觉定位与力控制精度提出了严苛要求;构建过程中,数据集仅包含单任务(42个episodes)与单一机器人类型(so_follower),限制了模型对多场景的泛化能力。此外,视频压缩编码(av1)可能引入视觉信息损失,且缺乏标准评价指标与基线模型,使得跨数据集性能对比难以系统化开展。
常用场景
经典使用场景
在机器人学习领域,数据集扮演着驱动算法迭代的关键角色。rollout_SortToy_20260527_143413 数据集由LeRobot框架采集,聚焦于一种排序玩具的机械臂操作任务,囊括了42个完整片段、近25000帧的精细记录。该数据集的核心使用场景在于为模仿学习与行为克隆算法提供基准训练素材,研究者可借助其包含的六维关节动作指令、机械臂状态序列以及多视角视觉图像(前视与顶视),训练机器人从示范中复现精准的抓取与放置动作。这种多模态数据融合特性使其尤其适合用于开发端到端的操控策略,例如基于视觉语言模型的零样本泛化任务探究。
解决学术问题
该数据集解决了机器人操作领域中一个基础而棘手的学术难题:如何在高维度状态空间下实现从有限示范中高效泛化。传统方法往往依赖手动设计的特征提取器,难以适应环境的微小变化。此数据集通过提供30帧每秒的高频时序观测,使得研究人员可以深入探索时序建模在降维动作预测中的效用,同时其包含的关节位置反馈与图像空间信息为对比学习方法的验证提供了平台。基于此,学术界得以施展对因果推理在机器人策略迁移中的影响分析,从而推动了从简单排序任务向复杂装配场景的认知跃迁。
实际应用
在产业自动化的一线场景中,该数据集展现了不可忽视的实用价值。例如,在物流仓储领域,工人需频繁将各类异形工件按规格分拣入单元箱;利用此数据集训练的神经网络策略可引导轻量协作机器人掌握柔性排序技巧,无需每次重新编程。此外,在消费电子制造中,精细元件的排布与归类亦可借鉴此数据集的示范模式。由于数据采集频率与真实工厂作业速率高度吻合,使得算法能无缝迁移至实际产线,显著降低技术落地过程中所需的迭代调试成本。
数据集最近研究
最新研究方向
在具身智能与机器人操作领域,rollout_SortToy_20260527_143413数据集聚焦于双臂协作中的物体分拣任务,为模仿学习与强化学习算法的突破提供了关键训练素材。该数据集采集了42个闭环操作回合、近25000帧高保真视觉与关节状态数据,通过前端与顶部双视角AV1编码视频记录精细操作过程。当前研究热点正从单一抓取转向复杂环境下的顺序化、策略化操作,该数据的发布恰好回应了机器人导航、多模态感知与动作规划交叉融合的迫切需求,为探索端到端策略在非结构化场景中的泛化能力搭建了标准化评测平台。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务