遇见数据集

P_P_Green_Cube_Structured_Placement_Generalisation

收藏
Hugging Face2026-08-02 更新2026-08-03 收录
官方服务:

资源简介:

这是一个使用LeRobot框架创建的机器人操作数据集,专注于绿色立方体的结构化放置泛化任务。数据集包含34个episodes,共17993帧,所有帧均属于单个任务。数据特征包括:6维动作(肩部旋转、肩部抬升、肘部弯曲、腕部弯曲、腕部旋转、夹爪位置)、6维观测状态(相同关节位置)、两个视角的视频图像(俯视摄像头和腕部摄像头,分辨率为480x640,帧率30fps,AV1编码)、时间戳、帧索引、episode索引、任务索引等。机器人类型为so_follower。数据以Parquet和MP4格式存储,总数据大小约100MB,视频大小约200MB。该数据集适用于机器人模仿学习、行为克隆等任务。

This is a robotic manipulation dataset created using the LeRobot framework, focusing on the generalized structured placement task for green cubes. The dataset contains 34 episodes with a total of 17993 frames, all belonging to a single task. The data features include: 6-dimensional actions (shoulder rotation, shoulder elevation, elbow flexion, wrist flexion, wrist rotation, gripper position), 6-dimensional observation states (same joint positions), video images from two viewpoints (top-down camera and wrist camera, with a resolution of 480x640, frame rate 30fps, AV1 encoding), timestamps, frame indices, episode indices, task indices, etc. The robot type is so_follower. The data is stored in Parquet and MP4 formats, with a total data size of approximately 100 MB and a video size of approximately 200 MB. This dataset is applicable to tasks such as robotic imitation learning and behavioral cloning.

创建时间:
2026-07-21
原始信息汇总

数据集总结

P_P_Green_Cube_Structured_Placement_Generalisation 是一个用于机器人操作任务的数据集,旨在支持结构化放置泛化场景下的机器人学习与研究。

基本信息

  • 许可证:Apache-2.0
  • 任务类型:机器人学(robotics)
  • 标签:LeRobot
  • 创建工具:该数据集使用 LeRobot 框架创建。

数据集概述

  • 总回合数:34 个回合(episodes)
  • 总帧数:17,993 帧
  • 总任务数:1 个任务
  • 帧率(fps):30
  • 数据划分:全部数据(34 个回合)作为训练集(train: "0:34"
  • 机器人类型:so_follower

数据文件结构与大小

  • 数据文件大小:100 MB(Parquet 格式)
  • 视频文件大小:200 MB(MP4 格式)
  • 数据路径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 (6,) 动作数据:肩部旋转、肩部升降、肘部弯曲、腕部弯曲、腕部旋转、夹爪位置
observation.state float32 (6,) 观察状态:与动作相同的 6 维关节位置信息
observation.images.overhead_cam video (480, 640, 3) 顶部摄像头视频,AV1 编码,30 fps,无音频
observation.images.wrist_cam video (480, 640, 3) 腕部摄像头视频,AV1 编码,30 fps,无音频
timestamp float32 (1,) 时间戳
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 回合索引
index int64 (1,) 索引
task_index int64 (1,) 任务索引

引用信息

该数据集的 BibTeX 引用信息暂未提供("More Information Needed")。

搜集汇总
数据集介绍
P_P_Green_Cube_Structured_Placement_Generalisation 数据集图片
构建方式
该数据集由LeRobot框架精心构建,旨在捕捉SO-100机械臂在结构化环境中将绿色立方体精准放置到指定位置的操控行为。数据集共含34个episode,总计17993帧,以30fps的频率记录,涵盖单任务场景。数据以Parquet格式存储,其中动作和观测状态均为六维浮点向量,分别对应机械臂的肩部、肘部、腕部及夹爪的关节位置。此外,数据集还包含两个视角的高清视频流:顶置相机和腕部相机,分辨率均为480×640,采用AV1编码,为视觉-运动控制学习提供了丰富的多模态信息。
使用方法
该数据集主要面向机器人操作任务的算法研究与模型训练。研究者可根据LeRobot的API加载数据,并利用其中的视觉观测、关节状态和动作标签构建端到端的模仿学习模型或强化学习环境。数据集的结构化设计允许直接提取训练和验证分割,便于进行模型性能评估。由于动作空间为六维连续控制量,适合使用扩散策略或Transformer等先进算法进行行为克隆。同时,双视角视频流可用于多模态融合或视觉表示学习,而关节状态可被用于状态估计或动力学建模。该数据集的Apache-2.0许可证保证了商业应用和二次开发的自由度。
背景与挑战
背景概述
在机器人操作领域,结构化摆放任务要求机械臂在动态环境中准确、稳定地放置物体,是智能制造与自动化仓储的核心环节。该数据集由fkearsey创建,基于HuggingFace LeRobot框架于近期采集,旨在研究绿色立方体结构化摆放的泛化能力。数据集包含34个演示片段、17993帧图像及同步的六维关节动作数据,涵盖了从肩部到夹爪的完整运动轨迹,利用顶置与腕部摄像头捕捉多视角视觉信息。该资源为模仿学习与强化学习算法提供了高质量的训练样本,推动了机器人从固定程序向数据驱动决策的范式转变,在机器人技能迁移与泛化研究领域具有重要的参考价值。
当前挑战
本数据集着力应对两大核心挑战。其一,领域难题在于机器人需在复杂视觉环境中实现精准的物体识别、姿态估计与操作规划,尤其需泛化至未见过的工作场景、光照变化及物体微小位移,这对模型的鲁棒性提出严苛要求。其二,数据构建艰辛,需手动遥控机械臂完成重复性演示,确保动作一致性与任务完整性,同时同步校准多视角摄像头与六维动作记录,处理传感器噪声与环境干扰,并平衡数据量、存储开销与标注精度,此间任何环节的偏差均可能导致训练失败,影响算法最终的操作效能。
常用场景
经典使用场景
该数据集聚焦于机器人操作中的结构化放置任务,具体场景为绿色立方体的精准投放与位置泛化。在机器人学习领域,此类数据集常被用于训练和评估视觉-运动策略,尤其是在模仿学习框架下(如LeRobot)。数据集包含34个演示回合,每回合记录6维关节角度指令、头顶摄像头与手部摄像头的同步视频流,以及时间戳和帧索引,为研究从像素到动作的映射提供了完整的多模态时间序列数据。其经典使用场景是作为基准测试集,用于验证模仿学习算法在固定任务模板下的泛化能力,即通过改变立方体的初始位置或目标放置区域,考察策略在未见环境中的适应性与鲁棒性,这直接关系到机器人操作技能的迁移性与部署可靠性。
解决学术问题
该数据集主要解决了机器人操作中结构化任务的分割与泛化问题。学术上,它支持研究如何在有限演示下学习可泛化的操作策略,尤其关注位置变化对策略性能的影响。通过提供统一的机器人平台(so_follower)和标准化数据格式(30Hz采样、双视角视觉、关节状态),数据集使研究者能够对比不同算法在任务泛化上的表现,如行为克隆、扩散策略或强化学习微调,从而探讨模型对空间变换的敏感性、数据效率以及视觉特征提取的有效性。其意义在于推动机器人学习从特定环境固化向开放场景适应性的转变,为构建具备通用操作能力的智能体提供数据基础,并促进模仿学习理论在复杂物理交互中的验证与改进。
实际应用
实际应用中,此类结构化放置任务数据集对于工业分拣、仓储物流及服务机器人具有直接价值。在工业场景中,机器人需依据视觉反馈将特定物件(如绿色立方体)精确放置于规定位置,且要求对工件摆放位置的变动具有实时适应能力。该数据集可被用于开发基于视觉引导的机械臂控制系统,通过在真实环境中的演示数据训练策略,使机器人能独立完成置放操作,减少人工示教成本。同时,手部与顶部摄像头的配置便于融合局部和全局视觉信息,增强对遮挡和视角变化的鲁棒性,这对提升生产线的柔性和智能化水平至关重要。此外,该数据集的Apache-2.0许可允许商业应用,加速了研究成果向实际产品的转化。
数据集最近研究
最新研究方向
该数据集聚焦于机器人操作领域中的结构化放置泛化任务,通过SO-100机械臂在绿色立方体操作中的精细动作捕捉,为模仿学习与强化学习算法提供了高保真的训练样本。当前研究前沿关注如何利用此类多视角视觉(顶置相机与腕部相机)与6自由度关节状态数据,提升机器人在未见场景下的泛化能力。结合LeRobot开源框架,该数据集推动了具身智能中少样本学习、视觉-运动控制对齐以及物理交互安全性的发展,为实现机器人在复杂工业与家居环境中的自适应操作奠定了数据基础。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务