ty625911724 / Jittor_StyleGAN3

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Jittor Version of StyleGAN3 (Alias-Free Generative Adversarial Networks)

Teaser image

Alias-Free Generative Adversarial Networks (NeurIPS 2021)
Tero Karras, Miika Aittala, Samuli Laine, Erik Härkönen, Janne Hellsten, Jaakko Lehtinen, Timo Aila
https://nvlabs.github.io/stylegan3

Quick start

Randomly sample face results:

python gen_images.py --outdir=out --trunc=1 --seeds=2 --network=./weights/jt_stylegan3_ffhq_weights_t.pkl

Generate lerp videos:

python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-31 --grid=4x2 --network=./weights/jt_stylegan3_ffhq_weights_t.pkl

Notes

We convert the stylegan3-t-ffhq-1024x1024.pkl to jittor pkl.

The Google Drive link: https://drive.google.com/drive/folders/1Gz-MUclZVSD2cAuyVsIIUoNDiXh3nmzw?usp=sharing

Other original weights dicts could also be load easily to test.

Acknowledgements

This repository borrows partially from the original codes and jittor_ops repository.

About

License:MIT License


Languages

Language:C++ 55.2%Language:Python 44.7%Language:Shell 0.1%