DragGAN

Wild implementation of DragGAN

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

Features

• Support movable region.
• Mutliple handle points.
• Automatically download stylegan2 checkpoint.
• Gradio Demo.
• Workable version.

Demo video

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