Skip to content

Commit 2fff9d2

Browse files
committed
CycleGAN + Folder structure changes
1 parent 1c7703a commit 2fff9d2

File tree

8 files changed

+535
-7
lines changed

8 files changed

+535
-7
lines changed
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

Making art with cats (style transfer)/CycleGAN.py

+524
Large diffs are not rendered by default.

README.md

+11-7
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,22 @@
22

33
This repository is a "toy" project so I can gain experience building deep neural networks. My first goal is learning to generate pictures of cats with Generative Adversarial Networks (^._.^).
44

5-
**Update** Many people were concerned about whether the generated images are really novel or rather just copies of the training dataset so I made a python function that output the 5 most similar training (real) images to the given generated (fake) images. See: https://github.com/AlexiaJM/Generative-model-most-similar-images.
6-
75
![](/images/DCGAN_220epochs.gif)
86

7+
**Updates**
8+
* 17/07/2017 : Added CycleGAN for style transfer, so far doesn't work well with the CAT dataset and it's so slow (20-24h to fully train it) that optimizing it is very difficult. I'm leaving it here for those interested but I probably will try fast neural style instead.
9+
* 15/07/2017 : Many people were concerned about whether the generated images are really novel or rather just copies of the training dataset so I made a python function that output the 5 most similar training (real) images to the given generated (fake) images. See: https://github.com/AlexiaJM/Generative-model-most-similar-images.
10+
911
**Objectives (so far)**
1012
* Generate images of cats using various types of Generative Adversarial Networks (GAN)
1113
* use **DCGAN** (Done)
1214
* use **WGAN** (Done)
1315
* use **WGAN-GP** (Done)
1416
* use **LSGAN** (Done)
1517
* use **BEGAN**
16-
* Transform real cats into art pieces using **CycleGAN**
18+
* Transform real cats into art pieces
19+
* use **CycleGAN** (Code done)
20+
* use **Fast neural style**
1721
* Various/Others
1822
* Preprocess cat images so we get aligned cat faces for much better GAN convergence (Done)
1923
* Separate cats by size to be better able to generate cats of certain sizes (Done)
@@ -23,17 +27,17 @@ This repository is a "toy" project so I can gain experience building deep neural
2327
* Add multi-gpu and non-CUDA option (Done)
2428
* Option to load previous models (Done)
2529
* Add log to output (Done)
26-
* Dropouts in G
2730
* L2 Regularization (Done)
2831
* Try making higher resolutions pictures
2932
* 128 x 128 (Done)
30-
* 256 x 256
33+
* 256 x 256 (Need more higher res data to do this)
3134
* Try Self-Normalizing Neural Networks (SELU) as per https://arxiv.org/abs/1706.02515 (Done)
35+
* Add option to generate more cat pictures (Done)
36+
* Add option to see the 5 nearest real cats images and thus be able to tell if the generated cats are really unique (Done)
37+
* Dropouts in G
3238
* Try adding Frechet Inception Distance (FID) as per https://arxiv.org/pdf/1706.08500.pdf
3339
* Try soft and noisy labels as per https://github.com/soumith/ganhacks
3440
* Try adding decaying noise to input as per https://github.com/soumith/ganhacks
35-
* Add option to generate more cat pictures
36-
* Add option to see the 5 nearest real cats images and thus be able to tell if the generated cats are really unique
3741

3842
**Needed**
3943

File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)