Fast One-Step Multi-Modal Trajectory Generation and Adaptive Reconstruction via MeanFlow for End-to-End Autonomous Driving
[2026/3/20]We released code and checkpoints.[2026/2/25]We released our paper on arXiv.[2026/2/21]π Accepted to CVPR 2026.
- π TODO List
- ποΈ Model Zoo
- π― Getting Started
- π¦ Data Preparation
- βοΈ Training and Evaluation
- β€οΈ Acknowledgements
- HUGSIM code release (Apr. 2026).
- NAVSIMv2 navtest code release (Apr. 2026).
- Checkpoints release (Mar. 2026).
- Code release (Mar. 2026).
- Paper release (Feb. 2026).
| Method | Backbone | Benchmark | PDMS | Weight Download |
|---|---|---|---|---|
| MeanFuser | ResNet-34 | NAVSIM | 89.0 | Google Drive |
| MeanFuser + BeyondDrive | ResNet-34 | NAVSIM | 90.3 | Google Drive |
| MeanFuser | ResNet-34 | HUGSIM | - | Google Drive |
git clone https://github.com/wjl2244/MeanFuser.git
cd MeanFuserconda create -n meanfuser python=3.9 -y
conda activate meanfuser
pip install -e .NOTE: Please review and agree to the LICENSE file file before downloading the data.
Follow the instructions in the NAVSIM installation guide to download the dataset.
Alternatively, you can download the dataset using Hugging Face with the following commands:
export HF_ENDPOINT="https://huggingface.co"
# export HF_ENDPOINT="http://hf-mirror.com" # Uncomment this line if you are in China
# Install the huggingface_hub tool
pip install -U "huggingface_hub"
# Download the OpenScene dataset
hf download --repo-type dataset OpenDriveLab/OpenScene --local-dir ./navsim_dataset/ --include "openscene-v1.1/*"
# Download the map data
cd download && ./download_maps.shMove the download data to create the following structure.
navsim_workspace/
βββ MeanFuser/
βββ dataset/
β βββ maps/
β βββ navsim_logs/
β β βββ test/
β β βββ trainval/
β βββ sensor_blobs/
β β βββ test/
β β βββ trainval/
βββ cache/
βββ navtest_v1_metric_cache/
βββ traintest_v1_cache/
We provide a script to cache the dataset and metrics.
cd MeanFuser
# Cache the dataset. (navtrain and navtest)
python scripts/evaluation/run_dataset_cache.sh
# Cache the metric.
python scripts/evaluation/run_metric_cache.shPlease download the pre-trained checkpoints from here and place them in the navsim_workspace/MeanFuser/exp/ directory.
cd MeanFuser
bash scripts/evaluation/run_meanfuser_evaluation.shPlease download the ResNet-34 pretrained weights from here. After downloading, update the corresponding path in the configuration file:navsim_workspace/MeanFuser/navsim/agents/meanfuser/meanfuser_config.py
cd MeanFuser
bash scripts/training/run_meanfuser_training.shWe provide a script to visualize the model's planned trajectory.
export NAVSIM_WORKSPACE="xxx/navsim_workspace"
python MeanFuser/tools/visualization_navtest_scenes.pyWe acknowledge all the open-source contributors for the following projects to make this work possible: