/Seeing-and-Hearing

[CVPR 2024] Seeing and Hearing: Open-domain Visual-Audio Generation with Diffusion Latent Aligners

Primary LanguagePythonOtherNOASSERTION

Seeing-and-Hearing

This repository includes official codes for "Seeing and Hearing: Open-domain Visual-Audio Generation with Diffusion Latent Aligners (CVPR 2024)".

🔮 Method

Framework Overview. Our approach is versatile and can tackle four tasks: joint video-audio generation (Joint-VA), video-to-audio (V2A), audio-to-video (A2V), and image-to-audio (I2A). By leveraging a multimodal binder, e.g., pretrained ImageBind, we establish a connection between isolated generative models that are designed for generating a single modality. This enables us to achieve both bidirectional conditional and joint video/audio generation.

🔆 Abstract

Video and audio content creation serves as the core technique for the movie industry and professional users. Recently, existing diffusion-based methods tackle video and audio generation separately, which hinders the technique transfer from academia to industry. In this work, we aim at filling the gap, with a carefully designed optimization-based framework for cross-visual-audio and joint-visual-audio generation. We observe the powerful generation ability of off-the-shelf video or audio generation models. Thus, instead of training the giant models from scratch, we propose to bridge the existing strong models with a shared latent representation space. Specifically, we propose a multimodality latent aligner with the pre-trained ImageBind model. Our latent aligner shares a similar core as the classifier guidance that guides the diffusion denoising process during inference time. Through carefully designed optimization strategy and loss functions, we show the superior performance of our method on joint video-audio generation, visual-steered audio generation, and audio-steered visual generation tasks. The project website can be found at https://yzxing87.github.io/Seeing-and-Hearing/.

Seeing and Hearing: Open-domain Visual-Audio Generation with Diffusion Latent Aligners
Yazhou Xing* 1, Yingqing He* 1, Zeyue Tian* 1, Xintao Wang2, Qifeng Chen1 (* indicates equal contribution)
1HKUST, 2ARC Lab, Tencent PCG

[Paper] [Project Page]

TODO

  • open-source v2a codes
  • open-source evaluation codes
  • open-source joint-va, a2v, and i2v codes

⚙️ Code

Installation

Please follow the instructions below:

git clone https://github.com/yzxing87/Seeing-and-Hearing.git
cd Seeing-and-Hearing/v2a
conda create -n seeing python=3.10
conda activate seeing
pip install -r pip_env3.txt

pip install git+https://github.com/yzxing87/transformers

pip install "git+https://github.com/facebookresearch/pytorchvideo.git"

pip install transformers_stream_generator

V2A generation

  1. download necessary checkpoints

    (1) Download audioldm ckpts from the huggingface repo. Put the ckpts to ckpt/.

    mkdir ckpt; cd ckpt
    git clone https://huggingface.co/cvssp/audioldm-m-full
    cd ..
    

    (2) Download imagebind ckpts imagebind_huge.pth from the official repo. Put the ckpts to imagebind/.checkpoints/.

    mkdir imagebind/.checkpoints; cd imagebind/.checkpoints
    wget https://dl.fbaipublicfiles.com/imagebind/imagebind_huge.pth
    cd ../..
    

    (3) Download qwen ckpts from the huggingface repo. Put the ckpts to qwen_ckpt/.

    mkdir qwen_ckpt; cd qwen_ckpt
    git clone https://huggingface.co/Qwen/Qwen-VL-Chat
    cd ..
    
  2. follow pipeline.sh

🤗 Citation

@inproceedings{xing24seeing,
  title     = {Seeing and Hearing: Open-domain Visual-Audio Generation with Diffusion Latent Aligners},
  author    = {Xing, Yazhou and He, Yingqing and Tian, Zeyue and Wang, Xintao and Chen, Qifeng},
  booktitle = {CVPR},
  year      = {2024}
}

📭 Contact

If you have any comments or questions, feel free to contact Yazhou Xing, Yingqing He or Zeyue Tian.

License

Please follow CC-BY-NC.