Project Page | Paper

Abstract

World models, especially in autonomous driving, are trending and drawing extensive attention due to its capacity for comprehending driving environments. The established world model holds immense potential for the generation of high-quality driving videos, and driving policies for safe maneuvering. However, a critical limitation in relevant research lies in its predominant focus on gaming environments or simulated settings, thereby lacking the representation of real-world driving scenarios. Therefore, we introduce DriveDreamer, a pioneering world model entirely derived from real-world driving scenarios. Regarding that modeling the world in intricate driving scenes entails an overwhelming search space, we propose harnessing the powerful diffusion model to construct a comprehensive representation of the complex environment. Furthermore, we introduce a two-stage training pipeline. In the initial phase, DriveDreamer acquires a deep understanding of structured traffic constraints, while the subsequent stage equips it with the ability to anticipate future states. The proposed DriveDreamer is the first world model established from real-world driving scenarios. We instantiate DriveDreamer on the challenging nuScenes benchmark, and extensive experiments verify that DriveDreamer empowers precise, controllable video generation that faithfully captures the structural constraints of real-world traffic scenarios.
Additionally, DriveDreamer enables the generation of realistic and reasonable driving policies, opening avenues for interaction and practical applications.

News

• [2024/11/02] 🚀 Research code for DriveDreamer is realsed! For future update, please follow GigaAI-research.
• [2024/10/17] 🚀We release the DriveDreamer4D project! (Key features: 4D reconstruction, complex maneuvers rendering)
• [2024/07/02] 🎉DriveDreamer is accepted for ECCV'24! Rank 15 in Most Influential ECCV'24 Papers.
• [2024/03/11] 🚀We release the DriveDreamer-2 project! (Key features: multi-view video generation, user-friendly with LLM)
• [2023/09/17] 🚀We release the DriveDreamer project!

Getting Started

• Installation

• Prepare Dataset & Env

• Train, Test, Visualization

Demo

Diverse Driving Video Generation.

Driving Video Generation with Traffic Condition and Different Text Prompts (Sunny, Rainy, Night).

Future Driving Video Generation with Action Interaction.

Future Driving Action Generation.

DriveDreamer Framework

Bibtex

If this work is helpful for your research, please consider citing the following BibTeX entry.

@article{wang2023drivedreamer,
  title={Drivedreamer: Towards real-world-driven world models for autonomous driving},
  author={Wang, Xiaofeng and Zhu, Zheng and Huang, Guan and Chen, Xinze and Zhu, Jiagang and Lu, Jiwen},
  journal={arXiv preprint arXiv:2309.09777},
  year={2023}
}