Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
rosinality committed Jul 6, 2019
2 parents b7a058f + 6486533 commit 33bb528
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,20 +1,31 @@
# Style-Based GAN in PyTorch

##### Update (2019/07/04)

* Now trainer uses pre-resized lmdb dataset for more stable data loading and training.
* Model architecture is now more closely matches with official implementation.

Implementation of A Style-Based Generator Architecture for Generative Adversarial Networks (https://arxiv.org/abs/1812.04948) in PyTorch

Usage:

You should prepare lmdb dataset

> python prepare_data.py --out LMDB_PATH --n_worker N_WORKER DATASET_PATH
This will convert images to jpeg and pre-resizes it. (For example, 8/16/32/64/128/256/512/1024) Then you can train StyleGAN.

for celebA

> python train.py --mixing -d {folder} PATH
> python train.py --mixing LMDB_PATH
for FFHQ

> python train.py --mixing --loss r1 --sched -d {folder}
> python train.py --mixing --loss r1 --sched LMDB_PATH
## Sample

![Sample of the model trained on FFHQ](doc/sample_ffhq_new.png)
![Style mixing sample of the model trained on FFHQ](doc/sample_mixing_ffhq_new.png)

512px sample from the generator trained on FFHQ.
512px sample from the generator trained on FFHQ.

0 comments on commit 33bb528

Please sign in to comment.