Alias Free Gan Pytorch Save

Unofficial implementation of Alias-Free Generative Adversarial Networks. (https://arxiv.org/abs/2106.12423) in PyTorch

Project README

alias-free-gan-pytorch

Unofficial implementation of Alias-Free Generative Adversarial Networks. (https://arxiv.org/abs/2106.12423) This implementation contains a lot of my guesses, so I think there are many differences to the official implementations

Usage

First create lmdb datasets:

python prepare_data.py --out LMDB_PATH --n_worker N_WORKER --size SIZE1,SIZE2,SIZE3,... DATASET_PATH

This will convert images to jpeg and pre-resizes it. This implementation does not use progressive growing, but you can create multiple resolution datasets using size arguments with comma separated lists, for the cases that you want to try another resolutions later.

Then you can train model in distributed settings

python train.py --n_gpu N_GPU --conf config/config-t.jsonnet training.batch=BATCH_SIZE path=LMDB_PATH

train.py supports Weights & Biases logging. If you want to use it, add wandb=true arguments to the script.

Sample

Latent translation sample 1 Latent translation sample 2 Latent translation sample 3 Latent translation sample 4 Latent translation sample 5 Latent translation sample 6

Open Source Agenda is not affiliated with "Alias Free Gan Pytorch" Project. README Source: rosinality/alias-free-gan-pytorch
Stars
497
Open Issues
16
Last Commit
8 months ago
License
MIT

Open Source Agenda Badge

Open Source Agenda Rating