flaxmodels
stylegan2-pytorch
flaxmodels | stylegan2-pytorch | |
---|---|---|
1 | 3 | |
222 | 2,665 | |
- | - | |
3.1 | 0.0 | |
9 months ago | 6 months ago | |
Python | Python | |
- | MIT License |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
flaxmodels
-
[P] Training StyleGAN2 in Jax (FFHQ and Anime Faces)
a while ago I posted here regarding a repository containing some pretrained models implemented in Jax/Flax. I decided to add training code for these models. Here is the training code for StyleGAN2: https://github.com/matthias-wright/flaxmodels/tree/main/training/stylegan2
stylegan2-pytorch
- Converting a .pkl file to a .pt for StyleGan2 Model
-
I'm stumped with installing PyTorch.
Originally I wanted to run https://github.com/JCBrouwer/maua-stylegan2. I was trying to run the convert_weight.py but it resulted in shape mismatch errors in torch torch.Size([1, 512, 4, 4]) vs torch.Size([1]), so I tried the version here https://github.com/rosinality/stylegan2-pytorch/blob/master/convert_weight.py and the result was the same.
-
[D] Node Collapse with StyleGAN2 - ada
Used this implementation (https://github.com/rosinality/stylegan2-pytorch) to better understand the code. IMO it is written way more clearly than the official implementation. You should spend a lot of time reviewing the code until you understand what each line is doing. It isn't helpful that their aren't very many comments, but you should be able to recognize different architectures and calculations from the paper. Understanding the details is key. Don't let your eyes glaze over any part of it.
What are some alternatives?
jax-resnet - Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
PaddleGAN - PaddlePaddle GAN library, including lots of interesting applications like First-Order motion transfer, Wav2Lip, picture repair, image editing, photo2cartoon, image style transfer, GPEN, and so on.
long-range-arena - Long Range Arena for Benchmarking Efficient Transformers
stylegan-waifu-generator - Generate your waifu with styleGAN, stylegan老婆生成器
jax-models - Unofficial JAX implementations of deep learning research papers
PyTorch-StudioGAN - StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.
ffhq-dataset - Flickr-Faces-HQ Dataset (FFHQ)
anycost-gan - [CVPR 2021] Anycost GANs for Interactive Image Synthesis and Editing
transformers - 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
BasicSR - Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.
prompt-tuning - Original Implementation of Prompt Tuning from Lester, et al, 2021
maua-stylegan2 - This is the repo for my experiments with StyleGAN2. There are many like it, but this one is mine. Contains code for the paper Audio-reactive Latent Interpolations with StyleGAN.