skytnt/fbanime-gan
By skytnt · ↓ 0 · ♥ 9 · Updated 2022-12-12
What's in this repo
This page lists the files in skytnt/fbanime-gan, including encoder.onnx, g_mapping.onnx, g_synthesis.onnx and more.
- 4 ONNX models (.onnx files)
- 10 files, including 4 .onnx, 2 .pkl, 1 .md, 1 .ipynb
- Last updated 2022-12-12
ONNX files
| File | Size | Download |
|---|---|---|
| encoder.onnx | — | Download |
| g_mapping.onnx | — | Download |
| g_synthesis.onnx | — | Download |
| waifu_dect.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
Model info
Project fbanimegan
fbanime.pkl
StyleGan2 model trained with official StyleGan3. But I modified the code (networks_stylegan2.py and dataset.py) to support non-square resolutions.
FID: 1.4
fbanime_fp32.pkl
fp32 version of fbanime.pkl
Note: Fp16 version (fbanime.pkl) only works on gpu. And fp32 version works on gpu and cpu.
g_mapping.onnx
onnx format mapping network of fbanime_fp32.pkl
g_synthesis.onnx
onnx format synthesis network of fbanime_fp32.pkl
encoder.onnx
e4e model trained with encoder4editing-stylegan3. I add support for official StyleGan2 model and change backbone to ResNet-34 in restyle-encoder.
waifu_dect.onnx
YOLOv5 model trained with official YOLOv5
Usage
see demo
Dataset
fbanimehq v2.0
Model info
Created: 2022-08-10
Updated: 2022-12-12
View on Hugging Face