digital-forensics-labs/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-04-10 12:33:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated progan

Browse Source
This commit is contained in:
Aladdin Persson
2021-03-21 12:19:18 +01:00
parent c72d1d6a31
commit 59b1de7bfe
5 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ML/Pytorch/GANs/ProGAN/README.md
View File

@@ -4,16 +4,15 @@ A clean, simple and readable implementation of ProGAN in PyTorch. I've tried to
## Results
The model was trained on the Celeb-HQ dataset up to 256x256 image size. After that point I felt it was enough as it would take quite a while to train to 1024^2.
|First is 64 random examples (not cherry picked) and second is more cherry picked examples. |
|First is some more cherrypicked examples and second is just sampled from random latent vectors|
|:---:|
|![](results/64_examples.png)|
|![](results/result1.png)|
|![](results/64_examples.png)|
### Celeb-HQ dataset
The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/lamsimon/celebahq).
### Download pretrained weights
Pretrained weights [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/ProGAN_weights.zip).