Skip to content

A Generative Adversarial Network for creating Birds: optimised to run on Google's Tensor Processing Units

Notifications You must be signed in to change notification settings

RaghavGirgaonkar/Bird-GAN-Optimised-for-TPUs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 

Repository files navigation

Bird-GAN-Optimised-for-TPUs

A Deep Convolutional Generative Adversarial Network for creating Birds: optimised to run on Google's Tensor Processing Units.

The network was trained on the Bird Dataset on Kaggle which has over 58000 images of birds across 400 different species.

The architecture of the GAN was inspired by this Face-GAN.

The Google Colab notebook can be run here, if you plan to use your own dataset you will need Google Cloud storage and will need to upload your dataset as a Tensorflow record file to run this notebook on TPUs.

Any comments or suggestions are welcome and feel free to contact us for the same! We'd also love to see any interesting birds that you may generate. Happy training!

A few generated birds, 128x128 resolution

Generated Birds

Training montage for 600 epochs, 64X64 images

bird_gan.mp4

About

A Generative Adversarial Network for creating Birds: optimised to run on Google's Tensor Processing Units

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published