遇见数据集

kunhsiang/eval_exp9_sc1var_grasp_the_small_box_20260529-114038

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

资源简介:

该数据集是一个用于机器人技术的数据集,专门针对so_follower类型的机器人。它包含1个完整的任务episode,总计2525帧数据,以30fps的帧率采集。数据以parquet格式存储,视频文件以mp4格式存储。数据集的特征包括机器人的动作(action)和观测状态(observation.state),动作和状态都包含6个关节位置信息:肩部平移、肩部升降、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置。此外,观测数据还包括来自三个不同视角的图像:前视(front)、顶部(top)和夹爪(gripper)摄像头,每个图像的分辨率为480x640像素,3通道彩色视频。数据集适用于机器人控制、模仿学习或强化学习等任务。

This dataset is a robotics dataset specifically designed for so_follower type robots. It contains 1 complete task episode with a total of 2525 frames, collected at a frame rate of 30 fps. The data is stored in parquet format, and video files are stored in mp4 format. The dataset features include robot actions and observation states, both containing 6 joint position information: shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper position. Additionally, the observation data includes images from three different perspectives: front, top, and gripper cameras, each with a resolution of 480x640 pixels and 3-channel color video. The dataset is suitable for tasks such as robot control, imitation learning, or reinforcement learning.

提供机构:
kunhsiang
搜集汇总
数据集介绍
kunhsiang/eval_exp9_sc1var_grasp_the_small_box_20260529-114038 数据集图片
构建方式
本数据集基于LeRobot框架构建,专注于机器人操作任务的记录与学习。数据集包含单一场景(eval_exp9_sc1var)下的一个完整操作序列,共2525帧,时长约84秒,帧率为30fps。操作任务为抓取一个小盒子(grasp_the_small_box),由so_follower机器人执行。数据以parquet格式存储动作与状态信息,动作空间涵盖6个自由度(肩部、肘部、腕部及夹爪位置),观测空间包含机器人关节状态和多视角视觉输入(前视、顶视、夹爪三路摄像头),视频采用AV1编码,分辨率为480×640。数据集分为单一训练集,便于直接用于模仿学习或强化学习方法的验证。
特点
该数据集以高保真度和多模态信息为特色。单条完整轨迹包含连续的关节角度序列与同步的多视角视频流,为机器人运动学与视觉的联合建模提供了丰富素材。视觉观测覆盖前视、顶视及夹爪视角,分别捕捉操作环境的全局与局部细节,有助于提升策略对空间关系的理解。动作数据以每帧30Hz的频率记录,精度为float32,确保动力学分析的准确性。数据集规模精简(约300MB),适合作为算法快速迭代的基准,尤其适用于需要精确定向抓取任务的场景。
使用方法
数据集可通过LeRobot库直接加载与可视化。用户可使用LeRobot的dataset接口读取parquet文件,获取结构化字典形式的观测、动作与元数据。建议采用训练-验证拆分,或直接使用全部帧进行模型训练。为利用多视角视觉信息,可对三个摄像头图像进行预处理(如归一化、裁剪),并同步时间戳对齐关节状态数据。动作预测任务可基于当前观测输出6维控制指令,适用于行为克隆或离线强化学习框架。可视化工具(如Hugging Face Spaces)支持播放视频轨迹,便于定性评估策略表现。
背景与挑战
背景概述
该数据集由研究者kunhsiang采用LeRobot框架构建,创建于2026年5月29日,聚焦于机器人精细操作领域中的小目标抓取任务。核心研究问题在于探索单一变量场景下机械臂对小型物体(如小盒子)的精准抓取能力,通过记录机械臂6自由度关节状态与多视角视觉观测数据,为模仿学习或强化学习提供训练样本。尽管数据集规模较小(仅含1个episode、2525帧),但其高分辨率多视角视频(640×480,30fps)与同步动作状态记录,为验证机器人从感知到控制的闭环行为提供了基准,对推动家庭服务、工业分拣等场景中的精密操作研究具有潜在价值。
当前挑战
该数据集应对的领域挑战包括:1) 机器人精细操作中的小目标识别与抓取稳定性问题,小物体常因几何尺寸小、纹理稀缺导致视觉定位困难,且夹爪力控制需适配物体脆弱性;2) 单一变量场景(如固定目标位置与姿态)虽简化了任务,但限制了模型在真实环境中的泛化能力,需平衡仿真与真实数据分布差异。构建过程中的挑战集中于数据采集效率与质量:单episode仅提供有限训练样本,难以覆盖抓取失败的边缘情况;高帧率多摄像头同步摄录导致存储压力(视频达200MB),且机械臂的重复运动轨迹可能引入噪声,需精细标定传感器对齐与时间戳一致性。
常用场景
经典使用场景
在机器人学习领域,精细操作任务始终是衡量算法鲁棒性与泛化能力的关键试金石。该数据集聚焦于单变量场景下的抓取小型盒子任务,通过配备多视角视觉传感器(前视、顶视及夹爪视角)的第六自由度机械臂,采集了密集的动作序列与状态信息。其经典使用方式在于为模仿学习算法提供高质量的专家示范轨迹,尤其适用于行为克隆(Behavior Cloning)和离线强化学习方法,研究者可基于30Hz的平滑动作流与高分辨率视频数据,训练机器人精确复现抓取动作。
解决学术问题
该数据集为机器人操作中样本效率低下与多模态感知融合难题提供了规范化的研究基准。学术研究中,稀疏奖励下的策略学习长期受到探索困境的制约,而本数据集通过完整记录2525帧连贯的力位混合控制过程,使得离线预训练与在线微调的结合成为可能。它不仅推动了解耦表征学习在六维动作空间中的应用,还催生了关于视触觉融合机制、轨迹平滑性约束以及小样本泛化能力的理论探讨,显著降低了精细操作研究中数据采集和标准化的门槛。
衍生相关工作
该数据集衍生了一系列重要的机器人学习工作。在算法层面,研究者基于其多视角观测特性,开发了融合隐式动作分区的结构化Transformer模型,以及在动作空间施加流形约束的扩散策略(Diffusion Policy)。在框架层面,LeRobot社区围绕此数据格式建立了标准化示范数据加载与可视化工具链,催生了如数据集蒸馏、跨具身迁移等扩展方法。此外,相关工作还探索了利用该数据集进行数据增强自动生成新轨迹,并通过逆强化学习挖掘潜在奖励函数,从而推动从单一任务向多任务泛化的演进。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务