WorldEngine
收藏资源简介:
WorldEngine数据集是一个专注于自动驾驶领域物理AI后训练的开源数据集,由香港大学OpenDriveLab、华为和上海创新研究院联合开发。该数据集系统性地解决了自动驾驶中长尾安全关键数据稀缺的问题,具有以下核心特点:1) 采用模块化数据结构,包含原始数据(nuPlan和OpenScene基础数据集)、算法引擎(用于端到端模型训练和评估的预处理数据)和仿真引擎(闭环仿真环境场景资产)三大模块;2) 数据形式多样,涵盖传感器数据、高清地图、标注信息、预处理标注、模型检查点、PDM指标缓存以及3D高斯泼溅(3DGS)渲染所需的场景资产;3) 通过行为世界模型(BWM)生成多样化交通变体,将稀疏的安全关键事件扩展为可学习的密集分布;4) 已在量产ADAS平台上进行验证,基于80,000+小时驾驶日志训练,可降低碰撞率高达45.5%,并在200公里道路测试中实现零脱离。数据集采用CC-BY-NC-SA-4.0许可,允许修改和分发但禁止商业用途。
The WorldEngine dataset is an open-source resource focused on post-training physical AI for autonomous driving, jointly developed by OpenDriveLab at The University of Hong Kong, Huawei, and Shanghai Innovation Research Institute. This dataset systematically addresses the critical shortage of long-tail safety-critical data in autonomous driving, and boasts the following core features: 1) Modular data structure: It includes three core modules, namely raw data (built upon the nuPlan and OpenScene foundational datasets), algorithm engine (preprocessed data for end-to-end model training and evaluation), and simulation engine (closed-loop simulation environment scene assets); 2) Diverse data modalities: It covers a wide spectrum of data types, including sensor data, high-definition maps, annotation information, preprocessed annotations, model checkpoints, PDM metric caches, and scene assets required for 3D Gaussian Splatting (3DGS) rendering; 3) Expanded dense distribution of safety-critical data: It generates diverse traffic variants via the Behavior World Model (BWM), thereby transforming sparse safety-critical events into learnable dense distributions; 4) Validated on mass-produced ADAS platforms: Trained on more than 80,000 hours of driving logs, it has been validated on mass-produced ADAS platforms, achieving up to 45.5% reduction in collision rates and zero disengagements during 200-kilometer road tests. The dataset is licensed under CC-BY-NC-SA-4.0, which permits modification and redistribution but forbids commercial usage.
WorldEngine 数据集概述
数据集基本信息
- 名称: WorldEngine
- 发布方: OpenDriveLab (The University of Hong Kong), Huawei Inc., Shanghai Innovation Institute (SII)
- 许可证: CC-BY-NC-SA-4.0
- 语言: 英语
- 标签: Robotics
- 官方数据发布日期: 2026年4月9日
核心特点
- 定位: 一个用于自动驾驶中物理人工智能的后训练框架,系统性解决自动驾驶中长尾安全关键数据稀缺问题。
- 数据驱动的长尾场景发现: 通过预训练智能体自动从真实世界驾驶日志中识别易出错的场景,无需人工设计或合成扰动。
- 基于3D高斯泼溅的逼真交互式仿真: 每个发现的场景被重建为一个完全可控、可实时渲染的仿真环境。
- 行为驱动的场景生成: 利用行为世界模型从长尾场景中泛化和合成多样化的交通变化,将稀疏的安全关键事件扩展为密集、可学习的分布。
- 基于强化学习的后训练: 在安全关键场景上进行后训练,其性能显著优于仅增加预训练数据,效果相当于预训练数据量增加约10倍。
- 生产规模验证: 部署于一个量产ADAS平台,基于超过80,000小时的驾驶日志进行训练,将碰撞率降低高达45.5%,并在200公里道路测试中实现零次接管。
数据集结构概述
数据集采用模块化数据结构,主要包含以下部分:
1. 原始数据 (data/raw/)
- nuPlan原始数据集: 包含高清地图、传感器数据(相机图像和激光雷达)以及训练/验证/测试划分。
- OpenScene-v1.1数据集 (基于nuPlan): 包含训练和测试的传感器数据及元数据。
2. AlgEngine数据 (data/alg_engine/)
用于端到端模型训练和评估的数据,包括:
- openscene-synthetic/: SimEngine生成的合成数据(需生成)。
- ckpts/: 预训练模型检查点。
- pdms_cache/: 预计算的PDM指标缓存。
- merged_infos_navformer/: 预处理的标注信息。
- test_8192_kmeans.npy: 用于PDM的K-means聚类数据。
3. SimEngine数据 (data/sim_engine/)
用于闭环仿真的数据,包括:
- assets/: 仿真场景资源(需提取),包含navtest、navtrain和navtest_failures场景资产。
- scenarios/: 场景配置文件,包含原始记录的场景和通过行为世界模型增强的场景。
环境设置
需配置以下环境变量:
WORLDENGINE_ROOT: 指向WorldEngine根目录的路径。NUPLAN_MAPS_ROOT: 指向nuPlan地图目录的路径。PYTHONPATH: 包含WorldEngine根目录。
使用步骤
- 下载数据集: 通过Hugging Face Hub或Git克隆。
- 提取场景资产: 提取
data/sim_engine/assets/目录下的分卷压缩文件。 - 设置环境变量: 配置
WORLDENGINE_ROOT及相关路径。 - 创建符号链接 (可选): 链接已下载的原始数据集以避免数据重复。
- 验证安装: 运行快速测试脚本。
引用
若使用本项目,请考虑引用相关论文。若使用渲染资产(MTGS),请额外引用MTGS相关论文。若使用行为世界模型生成的场景数据,请额外引用相关论文。
许可证与使用条款
- 许可证: CC-BY-NC-SA-4.0
- 允许: 修改、分发、私人使用。
- 要求: 署名、以相同方式共享。
- 限制: 禁止商业使用;必须保留版权和许可证声明。
相关资源链接
- 项目主页: https://github.com/OpenDriveLab/WorldEngine
- Hugging Face数据集页面: https://huggingface.co/datasets/OpenDriveLab/WorldEngine
- ModelScope数据集页面: https://www.modelscope.cn/datasets/OpenDriveLab/WorldEngine
- 完整文档: https://github.com/OpenDriveLab/WorldEngine/tree/main/docs
- 场景重建: https://github.com/OpenDriveLab/MTGS
- 问题反馈: https://github.com/OpenDriveLab/WorldEngine/issues
- 讨论区: https://huggingface.co/datasets/OpenDriveLab/WorldEngine/discussions




