Skip to content

WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation

License

Notifications You must be signed in to change notification settings

fudan-zvg/WoVoGen

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation

WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation,
Jiachen Lu, Ze Huang, Jiahui Zhang, Zeyu Yang, Li Zhang
Fudan University

Official implementation of "WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation".

  • Our WoVoGen is crafted to generate future world volumes (i.e., HD maps and occupancy) and high-quality multi-camera street-view images, with the input of past world-volumes. For comprehensive control, editing, and diverse sample generation, please refer to the article.

Abstract

Generating multi-camera street-view videos is critical for augmenting autonomous driving datasets, addressing the urgent demand for extensive and varied data. Due to the limitations in diversity and challenges in handling lighting conditions, traditional rendering-based methods are increasingly being supplanted by diffusion-based methods. However, a significant challenge in diffusion-based methods is ensuring that the generated sensor data preserve both intra-world consistency and inter-sensor coherence. To address these challenges, we combine an additional explicit world volume and propose the World Volume-aware Multi-camera Driving Scene Generator (WoVoGen). This system is specifically designed to leverage 4D world volume as a foundational element for video generation. Our model operates in two distinct phases: (i) envisioning the future 4D temporal world volume based on vehicle control sequences, and (ii) generating multi-camera videos, informed by this envisioned 4D temporal world volume and sensor interconnectivity. The incorporation of the 4D world volume empowers WoVoGen not only to generate high-quality street-view videos in response to vehicle control inputs but also to facilitate scene editing tasks.

🛠️ Pipeline


🎞️ Controlling and Editing

  • WoVoGen empowers diverse and controllable scene generation. Altering the random seed allows for the generation of various scenarios. Additionally, adjustment to weather (such as sunny, rainy, night, etc.) and location (Singapore, Boston, etc.) within the prompt enables the modification of weather conditions and city styles within the generated scene.

  • Top: the ability to selectively add or remove specific objects (such as trees, buildings, cars, etc., highlighted by red circles in the figure) within the world volume empowers the precise and coherent generation of diverse driving scenarios across multiple cameras.
  • Bottom: due to WoVoGen advanced 3D understanding capabilities, the rotation of perspectives across multiple cameras can be achieved by modifying the camera's extrinsic parameters. This functionality enables the generation of driving scenes from any desired angle.

  • More controlling and editing samples generated by WoVoGen.

🚗 Scene Generation

  • WoVoGen excels in producing future world volumes with temporal consistency. Subsequently, it utilizes the world volume-aware 2D image features derived from the world model's outputs to synthesize a driving video with both multi-camera consistency and temporal consistency.


Get started

Requirements

  • Python == 3.9
  • torch == 2.1.0

Generate world Volumes

python tools/nuscenes_convertor.py \
    --nusc_root=$NUSC_ROOT \
    --nusc_occ_root=$NUSC_OCC_ROOT \
    --out_root=$OUT_ROOT \
    --vae  # add image latent to world volumes

Generate object guidance

python tools/clip_convertor.py

Single frame generation

# Add weights to the model
python tools/add_weight.py \
    models/v2-1_512-ema-pruned.ckpt \
    models/wovogen_single.ckpt

# Train the single frame model
python train_single_frame.py \
    --config_path models/cldm_v21_c64_256x448_6cat_clip_local_high_dim.yaml \
    --resume-path=models/wovogen_single.ckpt

📜 BibTeX

@article{lu2023wovogen,
  title={WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation},
  author={Lu, Jiachen and Huang, Ze and Zhang, Jiahui and Yang, Zeyu and Zhang, Li},
  journal={arXiv preprint arXiv:2312.02934},
  year={2023},
}

Acknowledgements

Here's how you can add a URL link to your README:

This includes a clickable link to the Stability AI GitHub repository. You can use this format to add more links or modify as needed!

About

WoVoGen: World Volume-aware Diffusion for Controllable Multi-camera Driving Scene Generation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages