遇见数据集

kunhsiang/eval_exp9_sc1var_grasp_the_green_box_20260528-181805

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

资源简介:

该数据集是一个机器人操作数据集,使用LeRobot工具创建。它包含一个单一任务(抓取绿色盒子),总共有1个episode和2568帧数据,帧率为30fps。数据集记录了机器人的动作(包括肩部、肘部、腕部和夹爪的位置)和观察信息,观察包括机器人的状态(关节位置)以及来自三个摄像头的视频流:前置、顶部和夹爪摄像头,视频分辨率为480x640,编码为av1。数据以parquet格式存储,视频以mp4格式存储。机器人类型为so_follower,可能用于跟随或操作任务。数据集遵循Apache 2.0许可证。

This dataset is a robotics manipulation dataset created using the LeRobot tool. It contains a single task (grasping the green box) with 1 episode and 2568 frames at 30fps. The dataset records robot actions (including positions of shoulder, elbow, wrist, and gripper) and observations, which include the robots state (joint positions) and video streams from three cameras: front, top, and gripper cameras, with a resolution of 480x640 and av1 encoding. Data is stored in parquet format, and videos are in mp4 format. The robot type is so_follower, likely used for following or manipulation tasks. The dataset is licensed under Apache 2.0.

提供机构:
kunhsiang
搜集汇总
数据集介绍
kunhsiang/eval_exp9_sc1var_grasp_the_green_box_20260528-181805 数据集图片
构建方式
该数据集依托于LeRobot框架构建,旨在采集机器人执行抓取绿色盒子这一特定任务的示范数据。数据采集过程中使用了so_follower型机器人,以30帧每秒的速率记录单次完整操作,共包含2568帧画面与1个独立任务。数据集结构清晰,将原始数据拆分为1000帧大小的数据块(chunk),并以Parquet格式存储状态与动作数据,同时将多视角视频(包括前视、顶视和夹爪视角)编码为AV1格式的MP4文件,实现了高效存储与检索。
特点
该数据集的核心特点在于其多模态融合能力,同时提供了6维连续动作空间(包括肩部、肘部、腕部及夹爪的位姿)与对应观测状态,并辅以三个视角的高清视频(480×640分辨率),为机器人模仿学习提供了丰富的环境感知信息。尽管单次操作的数据量有限(约100MB的数值数据与200MB的视频),但较高的采样频率与完整的特征定义使其适合用于验证单任务下的行为克隆算法。此外,数据集遵循Apache-2.0许可协议,便于学术与工业界的开放使用。
使用方法
用户可通过LeRobot库直接加载该数据集,利用标准化的parquet文件读取动作和观测状态序列,并借助OpenCV或PyAV解码AV1格式的视频流。数据集中将前1帧作为训练集,适用于训练基于视觉的机器人操控策略。建议研究者将其作为评估工具,在相似场景下测试模型从示范中泛化抓取技巧的能力。注意需提前安装LeRobot依赖并配置好视频解码环境,方可复现完整的数据加载流程。
背景与挑战
背景概述
该数据集由研究者kunhsiang于2026年创建,基于LeRobot框架采集,聚焦于机器人操控领域的精细抓取任务,核心研究问题在于通过视觉引导使机械臂准确抓取绿色盒子。数据集包含单集2568帧的演示轨迹,记录了从肩部关节到夹爪的六维动作序列及多视角视觉信息,为模仿学习与强化学习提供了高质量示范数据。作为机器人学习社区的开源资源,该数据集有助于推动具身智能体在结构化环境中的操作能力研究,尤其在少样本泛化与任务复用方面具有潜在影响力。
当前挑战
数据集所解决的领域挑战在于机器人精细抓取任务中的视觉-运动耦合问题,即如何从多视角图像中提取有效特征以生成精确的关节控制指令,同时避免物体滑动或抓取失败。构建过程中面临的主要挑战包括:单集数据量有限(2568帧),难以覆盖绿色盒子在不同位姿与光照条件下的多样性;机械臂关节空间与图像像素空间之间的高维映射可能导致过拟合;视频压缩编码(AV1)虽节省存储,但可能引入帧间质量损失,影响视觉特征的保真度。
常用场景
经典使用场景
在机器人操作领域,该数据集专为‘抓取绿色盒子’这一精细化操作任务而设计。通过采集单次示教过程中六自由度机械臂的关节角度序列、夹爪状态以及多视角视觉观测(包括前视、顶视和夹爪摄像头),它为模仿学习与行为克隆算法提供了高质量的专家轨迹。研究者可利用该数据集训练深度神经网络,直接从视觉输入映射至关节空间动作,从而复现并泛化抓取策略,是验证端到端机器人操控模型性能的理想基准。
实际应用
在实际层面,该数据集可直接服务于工业分拣与仓储自动化场景。例如,在物流中心,装配了类似so_follower机器臂的系统可基于此类数据训练,实现对特定颜色与形状包裹的可靠抓取。此外,其标准化结构还便于迁移至家用服务机器人领域,如辅助老年人拾取指定物品,只需微调网络参数即可适应新场景,从而加速从仿真研究到物理机器人部署的转化过程。
衍生相关工作
基于LeRobot框架构建是该数据集的一大标志,使其天然兼容社区中涌现的一系列经典工作。例如,研究者可将其作为训练数据,验证扩散策略(Diffusion Policy)在精准抓取任务中的成效,或用于比较ACT(Action Chunking with Transformers)与iLQR等算法在少样本条件下的表现。这种标准化格式还启发了多任务扩展,例如将不同颜色的抓取数据合并,构建场景变量泛化基准,从而推动视觉运动策略领域取得更扎实的进步。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务