AdaRoute
收藏资源简介:
AdaRoute Dataset v1 是一个用于学习新障碍物周围动作条件机器人动力学的匹配干预数据集。它包含来自固定策略 pi0.5 的原始 LIBERO-10 任务 rollout、直接障碍物失败案例、运动规划器桥接轨迹以及随后的策略延续数据。数据集规模包含 375 个匹配的根轨迹(300 个基础根轨迹和 75 个受控重试),总计 1,529 个符合模式的 HDF5 轨迹,磁盘占用约 98 GiB。数据覆盖 LIBERO-10 的任务 0、8 和 9,包含 764 个无碰撞执行的规划器桥接、307 个规划后成功的策略延续以及 328 个被标记为“卡住”的候选延续。障碍物为固定圆柱体、横杆和墙壁,其姿态和颜色有所变化。每个根轨迹在相同任务状态、物体偏移和障碍物设置下,包含五个分支变体:无障碍物的干净策略执行、直接在有障碍物环境下执行的策略、以及三种不同规划器桥接(最短路径、最高清空度、与阶段兼容的端点)后接策略延续的组合。数据每帧存储双视角 256x256 RGB 图像(agent_rgb 和 wrist_rgb)、模拟器生成的二值障碍物掩码、末端执行器和机械臂的本体感知数据(位置、速度)、归一化的 7 维动作指令、实际末端执行器位移、动作来源标签、奖励、完成标志、卡住标志和障碍物接触标签,以及特权模拟状态(关节位置、速度、物体和障碍物姿态)。RGB 图像、本体感知、任务文本和动作可作为可部署的动作条件世界模型(AC-WM)的输入;而模拟器状态、接触信息、奖励和真实掩码仅用于离线监督和评估。数据集旨在支持 AC-WM 训练、动作跟随评估、物理交互建模、碰撞/卡住预测、规划器端点评分以及策略-规划器交接研究。数据集以原始 HDF5 格式分发,以保留帧对齐的图像、掩码和物理交互数据的完整性。
AdaRoute Dataset v1 is a matched intervention dataset for learning action-conditioned robot dynamics around novel obstacles. It contains raw LIBERO-10 task rollouts from a fixed policy pi0.5, direct obstacle failure cases, motion planner bridge trajectories, and subsequent policy continuation data. The dataset comprises 375 matched root trajectories (300 base root trajectories and 75 controlled retries), totaling 1,529 conforming HDF5 trajectories, occupying approximately 98 GiB on disk. It covers tasks 0, 8, and 9 of LIBERO-10, including 764 collision-free planner bridges, 307 successful policy continuations after planning, and 328 candidate continuations marked as stuck. Obstacles are fixed cylinders, crossbars, and walls with varying poses and colors. Each root trajectory, under the same task state, object offset, and obstacle configuration, contains five branch variants: clean policy execution without obstacles, direct policy execution in the presence of obstacles, and three combinations of different planner bridges (shortest path, highest clearance, stage-compatible endpoint) followed by policy continuation. Per frame, the dataset stores dual-view 256x256 RGB images (agent_rgb and wrist_rgb), simulator-generated binary obstacle masks, end-effector and arm proprioception (position, velocity), normalized 7-dimensional action commands, actual end-effector displacement, action source label, reward, completion flag, stuck flag, obstacle contact label, and privileged simulator state (joint positions, velocities, object and obstacle poses). RGB images, proprioception, task text, and actions can serve as input for deployable action-conditioned world models (AC-WM); while simulator state, contact information, reward, and ground-truth masks are used only for offline supervision and evaluation. The dataset is intended to support AC-WM training, action-following evaluation, physical interaction modeling, collision/stuck prediction, planner endpoint scoring, and policy-planner handover research. The dataset is distributed in raw HDF5 format to preserve frame-aligned integrity of images, masks, and physical interaction data.
AdaRoute Dataset v1 概述
数据集名称:AdaRoute Dataset v1
语言:英语
标签:机器人学、机器人操作、具身智能、动作条件世界模型、运动规划、LIBERO
数据规模:1K < n < 10K
数据集简介
AdaRoute v1 是一个用于学习在新型障碍物周围进行动作条件机器人动力学的匹配干预数据集。包含来自冻结 pi0.5 策略的原始 LIBERO-10 轨迹、直接障碍物故障、运动规划器桥接以及后续策略延续。
项目代码与收集详情:https://github.com/Loule0-0/AdaRoute
发布摘要
- 匹配根节点数量:375 个,其中 300 个基础根节点,75 个控制重试
- 轨迹数量:1,529 条符合模式的 HDF5 轨迹
- 磁盘占用:约 98 GiB(104,247,555,293 字节)
- 任务分布:LIBERO-10 任务 0、8、9,分别包含 526、494、509 条轨迹
- 规划器桥接:764 条无碰撞的执行规划器桥接
- 策略延续:307 次成功规划后策略延续
- 卡住延续:328 条被标记为卡住的候选延续
- 障碍物类型:固定圆柱、条形和墙壁障碍物,具有变化的姿态和颜色
匹配分支结构
每个根节点共享相同的任务状态、物体偏移和障碍物,包含以下分支:
- clean_policy:无障碍物下的 pi0.5
- obstacle_policy:pi0.5 直接围绕障碍物执行
- candidate_01_shortest:最短可行规划器桥接,后接 pi0.5
- candidate_02_clearance:最高间隙桥接,后接 pi0.5
- candidate_03_recovery_prior:阶段兼容端点,后接 pi0.5
部分根节点被有意保留为负规划覆盖数据,包含少于三个可行规划器分支,但每条存储轨迹都是完整的。
存储信号
每帧存储内容:
- 观测:
observations/agent_rgb和observations/wrist_rgb(256×256 RGB) - 障碍物掩码:
observations/*_obstacle_mask(模拟器真实二进制障碍物掩码) - 本体感:
proprio/eef_*、gripper_qpos、arm_qpos、arm_qvel - 动作:
actions/commanded_osc(归一化 7D LIBERO EEF 增量姿态及抓手)、actions/realized_eef_delta、actions/source - 标签:
labels/reward、done、stuck以及障碍物接触标签 - 特权信息:
privileged/sim_qpos、sim_qvel、物体姿态和障碍物姿态
RGB、本体感、任务文本和动作可作为可部署的 AC-WM 输入。模拟器状态、接触、奖励和真实掩码仅用于离线监督和评估标签。障碍物掩码不作为在线触发输入。
注意:commanded_osc[i] 是产生观测 i 的动作,而非来自该观测的下一个动作。
存储库结构
text README.md manifests/ adaroute_train_v1.jsonl adaroute_train_v1_retries_v4.jsonl adaroute_train_v1_with_retries.jsonl train/ task_00/<root>/<branch>/{trajectory.h5,metadata.json} task_08/<root>/<branch>/{trajectory.h5,metadata.json} task_09/<root>/<branch>/{trajectory.h5,metadata.json} scene_files/ audit_final.json audit_final.csv file_index.tsv sample_episode_inspection.json
下载方式
bash hf download Jaqen0-0/AdaRoute --repo-type dataset --local-dir data/adaroute_train_v1
数据集以原始 HDF5 格式发布,而非扁平化为 Hugging Face datasets 表,以避免帧对齐图像、掩码、物理交互和特权评估标签的有损或重复编码。
最小读取示例
python import h5py
path = "train/task_00/<root>/<branch>/trajectory.h5" with h5py.File(path, "r") as episode: agent_rgb = episode["observations/agent_rgb"][0] wrist_rgb = episode["observations/wrist_rgb"][0] action = episode["actions/commanded_osc"][0] obstacle_mask = episode["observations/agent_obstacle_mask"][0]
预期用途
该数据集支持以下研究方向:
- AC-WM 训练
- 动作跟随评估
- 物理交互建模
- 碰撞/卡住预测
- 规划器端点评分
- 策略-规划器交接研究
在使用时,不应将特权模拟器字段作为可部署输入用于报告真实机器人兼容结果。
许可说明
该数据集未声明独立许可证。用户需自行遵守 AdaRoute、LIBERO、MuJoCo、OpenPI 及底层模拟资产的许可条款。





