zhangliangming / DragGAN

Unofficial implementation of "Drag Your GAN: Interactive Point-based Manipulation on the Generative Image Manifold"

DragGAN

💥 Colab Demo

Note for Colab, remember to select a GPU via Runtime/Change runtime type (代码执行程序/更改运行时类型).

Unofficial implementation of Drag Your GAN: Interactive Point-based Manipulation on the Generative Image Manifold

🌟 Features

• Tweak performance.
• Custom Image with GAN inversion.
• Automatically determining the number of iterations.
• Download generated image and generation trajectory.
• Controling generation process with GUI.
• Automatically download stylegan2 checkpoint.
• Support movable region, mutliple handle points.
• Gradio and Colab Demo.

Demo

demo.mp4

Usage

Ensure you have PyTorch, Gradio, and tqdm installed.

pip install -r requirements.txt

Lanuch the Gradio demo

python gradio_app.py

If you have any issuse for downloading the checkpoint, you could manually download it from here and put it into the folder checkpoints.

Acknowledgement

Official DragGAN   StyleGAN2   StyleGAN2-pytorch

Citation

@inproceedings{pan2023draggan,
    title={Drag Your GAN: Interactive Point-based Manipulation on the Generative Image Manifold}, 
    author={Pan, Xingang and Tewari, Ayush, and Leimk{\"u}hler, Thomas and Liu, Lingjie and Meka, Abhimitra and Theobalt, Christian},
    booktitle = {ACM SIGGRAPH 2023 Conference Proceedings},
    year={2023}
}