PeterZhouSZ / StyleCariGAN

Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation" (SIGGRAPH 2021)

StyleCariGAN in PyTorch

Official implementation of StyleCariGAN:Caricature Generation via StyleGAN Feature Map Modulation in PyTorch

Requirements

• PyTorch 1.3.1
• CUDA 10.1/10.2

Usage

First download pre-trained model weights:

bash ./download.sh

Test

python test.py

This will generate caricatures from latent codes corresponding to photos in ./examples. You can see the results in ./results directory.

When you generate caricatures from your photos, you need to include latent code inversion:

python test.py --invert_images

This will invert latent codes from input photos and generate caricatures from latent codes.