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

cyclegan, progan

Browse Source
This commit is contained in:
Aladdin Persson
2021-03-11 15:50:44 +01:00
parent 91b1fd156c
commit 2c53205f12
27 changed files with 276 additions and 238 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,17 +4,17 @@ A clean, simple and readable implementation of CycleGAN in PyTorch. I've tried t
## Results
The model was trained on Zebra<->Horses dataset.
|1st row: Input / 2nd row: Generated / 3rd row: Target|
|1st column: Input / 2nd column: Generated / 3rd row: Re-converted|
|:---:|
|![](results/results_anime.png)|
|![](results/results_maps.png)|
|![](results/horse_results.png)|
|![](results/zebra_results.png)|
### Horses and Zebras Dataset
The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/suyashdamle/cyclegan).
### Download pretrained weights
Pretrained weights for Satellite image to Google Map [will upload soon]().
Pretrained weights [will upload soon]().
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.

BIN
ML/Pytorch/GANs/CycleGAN/results/horse_results.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 817 KiB

BIN
ML/Pytorch/GANs/CycleGAN/results/zebra_results.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 816 KiB