/DriveDreamer

[ECCV 2024] DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving

Primary LanguagePythonApache License 2.0Apache-2.0

DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving

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. abs

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

Demo

Diverse Driving Video Generation.

diverse.mp4

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

text_driven.mp4

Future Driving Video Generation with Action Interaction.

action_interact_concat.mp4

Future Driving Action Generation.

action_pred_concat.mp4

DriveDreamer Framework

method

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}
}