遇见数据集

ccwatson/mw_spatial_cardinal_and_diagonal_15cm_strict_50deg_close_high_approach_improved_mw_target_v2

收藏
Hugging Face2026-04-30 更新2026-05-03 收录
官方服务:

资源简介:

该数据集是使用LeRobot创建的,主要用于机器人技术相关任务。数据集包含1600个episodes,总计196926帧数据,涉及16个不同的任务。数据内容包括机器人状态观察(4维浮点数组)、环境状态观察(39维浮点数组)、两个视角的图像数据(224x224x3的RGB图像)、机器人动作(4维浮点数组)以及时间戳、帧索引等元数据。数据以parquet格式存储,帧率为24fps,训练集包含所有episodes。

This dataset was created using LeRobot and is primarily intended for robotics-related tasks. It contains 1600 episodes with a total of 196926 frames, covering 16 different tasks. The data includes robot state observations (4-dimensional float32 array), environment state observations (39-dimensional float32 array), image data from two perspectives (224x224x3 RGB images), robot actions (4-dimensional float32 array), as well as metadata such as timestamps and frame indices. The data is stored in parquet format with a frame rate of 24fps, and the training set includes all episodes.

提供机构:
ccwatson
搜集汇总
数据集介绍
ccwatson/mw_spatial_cardinal_and_diagonal_15cm_strict_50deg_close_high_approach_improved_mw_target_v2 数据集图片
构建方式
该数据集基于LeRobot框架构建,源自MetaWorld机器人仿真环境,专注于空间方位感知与抓取任务。数据集包含1600个演示片段,共计196926帧,覆盖16种不同的任务类型。每个演示通过高精度15厘米分辨率与50度视角约束下的近距靠近策略进行采集,确保示教动作的准确性与一致性。数据以parquet格式存储于chunk分片结构中,每个片段独立保存,便于分布式加载与处理。
特点
数据集的核心特点在于其多模态感知与高保真度。观测特征融合了4维机器人状态、39维环境状态以及双视角视觉图像(corner4与gripperPOV),均为224x224的RGB图像,提供丰富的空间信息。动作空间为4维连续控制信号,准确反映机械臂的位姿调整。所有特征以float32精度存储,保证数值稳定性。此外,数据集通过严格的帧率控制(24fps)与分片管理,为模仿学习与策略泛化研究提供了高质量的训练基础。
使用方法
数据集可直接通过LeRobot库加载,使用默认配置读取data目录下的parquet文件。用户可依据episode_index与task_index进行任务筛选,或按frame_index顺序采样。训练时,建议将observation.state与corner4.image/gripperPOV.image作为输入,actions作为输出,构建行为克隆或强化学习模型。数据集已预设100%训练集划分(splits.train覆盖0至1599号片段),无需额外拆分。需注意,数据集未包含视频文件,所有图像数据均内嵌于parquet中。
背景与挑战
背景概述
该数据集名为 mw_spatial_cardinal_and_diagonal_15cm_strict_50deg_close_high_approach_improved_mw_target_v2,依托于LeRobot框架构建,由MetaWorld机器人平台提供技术支持,创建于近年来机器人学习领域快速发展阶段。数据集包含1600个演示回合、近20万帧数据,覆盖16种任务,专注于机械臂在空间中的精确操作,如沿主方向和对角线方向以15厘米步长、50度视角逼近目标物体。其核心研究问题在于如何通过高精度视觉和状态信息,训练机器人掌握复杂且精细的空间交互技能。该数据集对机器人模仿学习、策略泛化以及行为克隆等领域具有重要推动力,为研究人员提供了标准化的仿真训练与评估环境。
当前挑战
数据集的领域挑战在于,机器人从精准空间操作中泛化至非结构化场景时,面临任务多样性不足和视角敏感性问题,当前16种任务虽覆盖基础方向,但难以应对真实世界中无限变体。构建过程中,挑战包括高一致性演示采集的困难,如确保每次操作在15厘米步长和50度视角下的严格复现,同时需平衡数据量(20万帧)与计算资源,并克服视觉特征(如corner4.image和gripperPOV.image)与动作空间(4维动作)之间的对齐噪声。此外,缺乏动态环境适应能力,如物体位置或光照变化,限制了模型在复杂现实部署中的鲁棒性。
常用场景
经典使用场景
在多模态机器人操控研究中,mw_spatial_cardinal_and_diagonal_15cm_strict_50deg_close_high_approach_improved_mw_target_v2数据集尤为引人瞩目。它基于MetaWorld仿真环境构建,聚焦于机械臂在严格空间约束下的精准操控任务,涵盖16个子任务,共1600个演示片段。该数据集通过多视角视觉输入(如corner4和gripperPOV相机)与低维状态信息(4维观测状态和39维环境状态)的有机融合,为模仿学习和强化学习提供了高保真的训练数据。数据集以24帧/秒的采样频率和统一的parquet格式存储,确保了时间维度的连续性与数据读取的高效性,是机器人技能习得与泛化研究的理想基石。
实际应用
从实际应用角度审视,该数据集在智能制造、医疗辅助和家庭服务机器人领域展现出广阔前景。在精密装配场景中,机械臂可凭借数据集中的高精度轨迹数据,自主习得微米级地螺丝拧紧或元器件插拔操作。针对医疗领域的血管介入手术,数据集蕴含的柔性力矩调节策略可迁移至导管机器人的路径规划。此外,在非结构化家庭环境中,数据集所训练的视觉伺服模型能使机器人稳定完成抓取、放置等日常任务,其抗干扰能力已在光照变化与背景杂乱的测试中得到验证,推动了机器人从实验室向真实世界的跨越。
衍生相关工作
该数据集孕育了一系列具有影响力的衍生研究,其中最经典的工作包括基于扩散策略的机器人操控模型,其利用数据集中多视角图像与状态序列,生成了平滑的动作轨迹并显著提升了任务成功率。另一项开创性研究是层级式强化学习框架,通过分解数据集中的复杂任务为子技能库,将学习效率提升了数倍。此外,跨具身泛化能力的突破也得益于此数据集,研究者通过对比不同机械臂在相同数据集上的表现,揭示了视觉特征与动作参数之间的可迁移表征,为通用机器人智能体的构建奠定了实证基础。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务