directional_pick_place_simpledesk_diverse_targets_tested40_200demos_20260716_molmobot
收藏资源简介:
Directional SimpleDesk Pick-and-Place Diverse Targets (MolmoBot) 是一个机器人模仿学习数据集,专注于桌面物体方向性拾放任务。该数据集包含200个由脚本策略生成的成功演示轨迹,这些轨迹从原始的MolmoSpaces输出转换为MolmoBot/Synthmanip训练格式。核心任务是根据自然语言指令(例如“拿起木杯并将其放在蓝色纸巾盒的左侧”)拾取一个桌面物体,然后从机器人视角将其放置到第二个物体的左侧或右侧。数据集在方向指令上保持平衡:包含100个“左侧”指令演示和100个“右侧”指令演示。这些演示来源于40个固定的初始场景布局(每个方向20个),每个布局提供5个成功演示。拾取物体和参考物体的角色独立变化。数据集包含5类拾取物体(杯、马克杯、纸巾盒、土豆、喷雾瓶)和15类参考物体,确保了目标物体的多样性。每个场景包含拾取物体、参考物体以及另外两个桌面物体,所有场景均使用相同的SimpleDesk环境(编号1088),但物体的身份、角色和位姿各不相同。数据以HDF5轨迹文件和MP4视频文件的形式提供,包含两个摄像头的流数据(腕部摄像头和肩部摄像头)。数据集已划分为训练集(36个布局,180个轨迹,360个视频)和验证集(4个布局,20个轨迹,40个视频),划分基于场景布局以确保独立性。数据已针对MolmoBot的`SynthmanipDataset`加载器进行预处理,并提供了基于训练集计算的归一化统计信息(动作使用分位数归一化,机器人状态使用最小-最大归一化)。该数据集适用于机器人模仿学习、视觉运动策略学习等方向的研究和开发。需要注意的是,数据集规模较小,仅包含成功轨迹,且所有演示共享40个初始布局,因此不能代表任务的全部失败分布或广泛的泛化能力。
Directional SimpleDesk Pick-and-Place Diverse Targets (MolmoBot) is a robot imitation learning dataset focusing on directional pick-and-place tasks for desktop objects. It contains 200 successful demonstration trajectories generated by scripted policies, converted from the original MolmoSpaces output to the MolmoBot/Synthmanip training format. The core task is to pick up a desktop object based on natural language instructions (e.g., Pick up the wooden cup and place it to the left of the blue tissue box) and then place it to the left or right of a second object from the robots perspective. The dataset is balanced in directional instructions: it includes 100 left instruction demonstrations and 100 right instruction demonstrations. These demonstrations originate from 40 fixed initial scene layouts (20 per direction), each providing 5 successful demonstrations. The roles of the pick-up object and reference object vary independently. The dataset includes 5 categories of pick-up objects (cup, mug, tissue box, potato, spray bottle) and 15 categories of reference objects, ensuring diversity in target objects. Each scene contains a pick-up object, a reference object, and two additional desktop objects. All scenes use the same SimpleDesk environment (ID 1088), but the identities, roles, and poses of objects differ. Data is provided in HDF5 trajectory files and MP4 video files, including stream data from two cameras (wrist camera and shoulder camera). The dataset is divided into a training set (36 layouts, 180 trajectories, 360 videos) and a validation set (4 layouts, 20 trajectories, 40 videos), with the division based on scene layouts to ensure independence. The data has been preprocessed for MolmoBots `SynthmanipDataset` loader, and normalization statistics computed from the training set are provided (action uses quantile normalization, robot state uses min-max normalization). This dataset is suitable for research and development in robot imitation learning, visuomotor policy learning, and related areas. It should be noted that the dataset is small in scale, contains only successful trajectories, and all demonstrations share 40 initial layouts, so it does not represent the full distribution of failures or broad generalization capabilities.
数据集概述:Directional SimpleDesk Pick-and-Place Diverse Targets (MolmoBot)
基本信息
- 数据集名称:Directional SimpleDesk Pick-and-Place Diverse Targets
- 机器人平台:Franka(通过MolmoBot/Synthmanip训练格式)
- 任务类型:桌面拾取与放置(Pick-and-Place)
- 指令形式:例如“Pick up the wooden cup and place it to the left of the blue tissue box”
- 方向平衡:100条“left”演示 + 100条“right”演示
数据集规模与划分
| 划分 | Benchmark配置 | HDF5轨迹 | MP4视频 |
|---|---|---|---|
| train | 36 | 180 | 360 |
| val | 4 | 20 | 40 |
| 总计 | 40 | 200 | 400 |
- 验证集划分按场景安全隔离:最后4个场景(10%)保留为验证集,同一场景的所有演示不跨划分。
- 每个场景包含5条成功轨迹,存储在
house_*目录下。 - 每个轨迹包含两个摄像头视角的视频文件:
wrist_camera_zed_mini和droid_shoulder_light_randomization。
任务成功标准
严格成功(strict_success) 需同时满足:
- 拾取物体与参考物体之间的XY轴对齐表面间隙在0到0.06米之间(配置最大0.05米,容差0.01米)。
- 两个物体由同一表面支撑。
- 参考物体从初始位姿移动不超过0.15米,旋转不超过60度。
- 拾取物体位于请求的左或右90度角扇区(以机器人参考桌面坐标系为准)。
宽松成功(lax_success) 仅需满足条件1-3,不要求方向条件。所有200条保留轨迹均以严格成功结束。
物体多样性
- 拾取物体类别(5类):Cup(11)、Mug(10)、TissueBox(7)、Potato(6)、SprayBottle(6)
- 参考物体类别(15类):Bowl、Box、Calculator、Candle、Cup、DishSponge、Egg、Mug、PaperTowel、Potato、RemoteControl、SoapBottle、SprayBottle、TissueBox、Tomato
- 每个场景包含拾取物体、参考物体以及两个额外的桌面物体。
- 所有场景使用同一个SimpleDesk房屋(
1088),但物体身份、角色和位姿各异。
数据采集
- 来源基准筛选:使用脚本化策略
pick_and_place_relative_next_to_search进行筛选,每个场景在5次尝试中至少获得1次严格成功。 - 数据采集:每个场景要求5次严格成功,共200次成功轨迹(来自210次尝试,使用500步步长)。
数据格式与布局
dataset_manifest.json directional_pick_place_simpledesk_diverse_targets_tested40_200demos_20260716_norm_stats.yaml train/ house_/ valid_trajectory_index.json val/ house_/ valid_trajectory_index.json
- 数据已转换为MolmoBot
SynthmanipDataset加载器兼容格式。 - 归一化:使用
franka_joint动作预设和franka_droid摄像头预设,动作采用分位数归一化,状态采用最小-最大归一化。 - 数据维度:动作和机器人状态均为8维;动作块形状为
[16, 8],每样本包含4张图像(每个摄像头2个观测帧)。
验证与完整性
- MolmoSpaces原生验证:零无效轨迹。
- 转换后验证:所有400个引用的摄像头视频文件均已解析并复制,无缺失文件。
- 训练集和验证集均通过了真实的
SynthmanipDataset加载测试。 - 最终发布包含200个HDF5文件和400个MP4文件。
来源与可复现性
- 源基准路径:MolmoSpaces仓库中
FrankaPickAndPlaceRelativeNextToSimpleDeskDiverseTargetsTested40Bench - 哈希值:
- benchmark.json SHA-256:
23511589a1c58c7d3ce27d841a93900ac502a71c68daeca91682ffc6a44e12b9 - 完整leaf基准 SHA-256:
83544065ae11b2c3f46f7463614c6052676621e4334bbb9bc8639d35647fc199
- benchmark.json SHA-256:
- 转换与上传:通过MolmoBot脚本
convert_molmospaces_to_molmobot.py完成,命令与选项详见源运行记录。
限制与注意事项
- 数据集规模较小,来自40个固定场景,仅使用一个SimpleDesk房屋,不具备广泛家庭场景泛化能力。
- 每个场景有5个共享初始排列的演示,不应将200个轨迹误认为200个独立布局。
- 仅包含成功的脚本化演示,不反映任务或规划器的完整失败分布。
- 物体覆盖多样但不均匀,行为受脚本化规划器抓取和运动偏好影响。
- 仅包含“left”和“right”方向,缺少“in front of”和“behind”。
- 未声明独立许可证,场景依赖MolmoSpaces、ProcTHOR、Objaverse及其关联资产,用户需自行遵守相关软件和资产使用条款。




