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

progan cyclegan

Browse Source
This commit is contained in:
Aladdin Persson
2021-03-11 15:55:27 +01:00
parent c67e7f88a6
commit 8cbaf3ebc3
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ML/Pytorch/GANs/CycleGAN/README.md
View File

@@ -14,7 +14,7 @@ The model was trained on Zebra<->Horses dataset.
The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/suyashdamle/cyclegan).
### Download pretrained weights
Pretrained weights [will upload soon]().
Pretrained weights [download](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/CycleGAN_weights.zip).
Extract the zip file and put the pth.tar files in the directory with all the python files. Make sure you put LOAD_MODEL=True in the config.py file.