Book Image

Generative Adversarial Networks Cookbook

By : Josh Kalin
Book Image

Generative Adversarial Networks Cookbook

By: Josh Kalin

Overview of this book

Developing Generative Adversarial Networks (GANs) is a complex task, and it is often hard to find code that is easy to understand. This book leads you through eight different examples of modern GAN implementations, including CycleGAN, simGAN, DCGAN, and 2D image to 3D model generation. Each chapter contains useful recipes to build on a common architecture in Python, TensorFlow and Keras to explore increasingly difficult GAN architectures in an easy-to-read format. The book starts by covering the different types of GAN architecture to help you understand how the model works. This book also contains intuitive recipes to help you work with use cases involving DCGAN, Pix2Pix, and so on. To understand these complex applications, you will take different real-world data sets and put them to use. By the end of this book, you will be equipped to deal with the challenges and issues that you may face while working with GAN models, thanks to easy-to-follow code solutions that you can implement right away.
Table of Contents (17 chapters)
Title Page
Copyright and Credits
About Packt
Dedication
Contributors
Preface
Dedication2
Index

Training this model


Training is always an adventure—there are so many pitfalls when developing GAN architectures. In this training class, we aim to provide a simple setup to train a GAN that takes a 2D image and creates a 3D model.

Getting ready

This is the final recipe in our chapter, so we've got a few files to create—the train.py, run.py, and run.sh files. Before continuing, check to make sure you have the exact same directory structure in your directory:

├── data
├── docker
│   ├── build.sh
│   ├── clean.sh
│   ├── Dockerfile
│   └── kaggle.json
├── out
├── README.md
├── run_autoencoder.sh
├── run.sh
└── src
    ├── discriminator.py
    ├── encoder_model.h5
    ├── encoder.py
    ├── gan.py
    ├── generator.py
    ├── run.py
    ├── train.py
    ├── x_test_encoded.npy
    └── x_train_encoded.npy

How to do it...

There are a few steps to successfully training this model—preparing the class, importing the data, training, plotting, and running the training code. This recipe will cover all of...