File size: 1,296 Bytes
4da4d66 b53a790 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
---
license: mit
library_name: pytorch
tags:
- image-generation
- gan
- stylegan
- stylegan3
- nvidia
---
# Anime Faces Generator (StyleGAN3 by NVIDIA)
<img width="679" alt="Generated Faces" src="https://user-images.githubusercontent.com/35907066/161809457-e6467724-5942-4a89-b379-85ddfd6ac86c.png">
This is a [StyleGAN3 PyTorch](https://github.com/NVlabs/stylegan3) model trained on this [Anime Face Dataset](https://github.com/bchao1/Anime-Face-Dataset).
### Usage
Demo on Spaces is not yet implemented.
You can run the model pickle file locally using the instructions in this generator-script-only subset of the StyleGAN3 repo:
- https://github.com/venture-anime/stylegan3-anime-faces-generator
### Dataset & Model Details
The [Anime Face Dataset](https://github.com/bchao1/Anime-Face-Dataset) was created by Mckinsey666.
Training was done in [Paperspace Gradient](https://gradient.run/) on a free `RTX-5000` instance with the following parameters:
- Configuration: `stylegan3-t`
- GPUs: `1`
- Batch Size: `8`
- Gamma: `0.5`
- Final tick: `102`
- Final fid50k_full value (this pickle): `9.26043547642206`.
# Train your own StyleGAN3 on PaperSpace or Colab
You can use the notebooks here for a ready-to-use training pipeline
https://github.com/akiyamasho/stylegan3-training-notebook |