viuts/wgan-animefaces

Anime faces WGAN-GP trained on pytorch

Anime faces WGAN-GP Generator

Step 1

Train the model with pytorch, the script is avaliable in train.ipynb, dataset can be downloaded from kaggle

Step 2

Transfer the model to ONNX -> Tensorflow -> TFJS, script is avaliable in ToONNX.ipynb

Step 3

Place the trained model in playground/public/converted

Step 4

Run npm start in playground