Skip to content

MuhammedBuyukkinaci/FlowerGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FlowerGAN

FlowerGAN : A DCGAN implementation in TensorFlow in Python 3 on 102 Category Flower Dataset.

Dependencies

pip install -r requirements.txt

or

pip3 install -r requirements.txt

Training

git clone https://github.com/MuhammedBuyukkinaci/FlowerGAN.git

cd ./FlowerGAN

python FlowerGAN.py

Notebook

You can download .ipynb file from here.

jupyter lab or jupyter notebook

Data

No MNIST or CIFAR-10.

This is a repository containing datasets of 8189 flower pictures belonging to 102 different categories. We aren't interested in

categories because GAN's is an UNSUPERVISED Machine Learning model.

Download .tgz extension version from here or .npy extension version from here. It is about 95 MB.

If you downloaded the dataset, extract files from 102flowers.tgz .Then put it in FlowerGAN folder.

If you download .npy file from Dropbox, put flower_photos.npy in FlowerGAN folder.

GPU

I trained on GTX 1050. 1 epoch lasted 6-7 minutes. I left my laptop overnight and obtained outputs in the morning.

If you don't want to wait for one month, use a GPU.

Architecture

Images are resized to (64,64,3) . The architecture is below:

alt text

Generated Photos

Each picture contains 16 generated photos. I trained it 100 epochs and obtained outputs 1 in 2 epochs.

alt text