StyleCariGAN in PyTorch

April 30, 2021 ยท View on GitHub

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.