GAN encoders in PyTorch that could match PGGAN, StyleGAN v1/v2, and BigGAN
MTV-TSA
MTV-TSA: Adaptable GAN Encoders for Image Reconstruction via Multi-type Latent Vectors with Two-scale Attentions.
This is the official code release for “Adaptable GAN Encoders for Image Reconstruction via Multi-type Latent Vectors with Two-scale Attentions”.
The code contains a set of encoders that match pre-trained GANs (PGGAN, StyleGANv1, StyleGANv2, BigGAN) via multi-scale vectors with two-scale attentions.
Usage
- training encoder with center attentions (align image)
python E_align.py
- training encoder with Gram-based attentions (misalign image)
python E_mis_align.py
-
embedding real images to latent space (using StyleGANv1 and w).
a. You can put real images at ‘./checkpoint/realimg_file/’ (default file as args.img_dir)
b. You should