Search icon
Arrow left icon
All Products
Best Sellers
New Releases
Books
Videos
Audiobooks
Learning Hub
Newsletters
Free Learning
Arrow right icon
Deep Learning with PyTorch Lightning

You're reading from  Deep Learning with PyTorch Lightning

Product type Book
Published in Apr 2022
Publisher Packt
ISBN-13 9781800561618
Pages 366 pages
Edition 1st Edition
Languages
Author (1):
Kunal Sawarkar Kunal Sawarkar
Profile icon Kunal Sawarkar

Table of Contents (15) Chapters

Preface Section 1: Kickstarting with PyTorch Lightning
Chapter 1: PyTorch Lightning Adventure Chapter 2: Getting off the Ground with the First Deep Learning Model Chapter 3: Transfer Learning Using Pre-Trained Models Chapter 4: Ready-to-Cook Models from Lightning Flash Section 2: Solving using PyTorch Lightning
Chapter 5: Time Series Models Chapter 6: Deep Generative Models Chapter 7: Semi-Supervised Learning Chapter 8: Self-Supervised Learning Section 3: Advanced Topics
Chapter 9: Deploying and Scoring Models Chapter 10: Scaling and Managing Training Other Books You May Enjoy

Creating new butterfly species using a GAN

In this section, we are going to use the same GAN model that we built in the previous section with a minor tweak to generate new species of butterflies.

Since we are following the same steps here, we will keep the description concise and observe the outputs. (The full code can be found in the GitHub repository for this chapter.)

We will first try with the previous architecture that we used for generating food images (which is 4 convolution, 1 fully connected layer, and 5 transposed convolution layers). We will then try another architecture with 5 convolution layers and 5 transposed convolution layers:

  1. Download the dataset:
    dataset_url =  'https://www.kaggle.com/gpiosenka/butterfly-images40-species'
    od.download(dataset_url)
  2. Initialize the variables for the images:
    image_size = 64
    batch_size = 128
    normalize = [(0.5, 0.5, 0.5), (0.5, 0.5, 0.5)]
    latent_size = 256
    butterfly_data_directory = "/content/butterfly...
lock icon The rest of the chapter is locked
Register for a free Packt account to unlock a world of extra content!
A free Packt account unlocks extra newsletters, articles, discounted offers, and much more. Start advancing your knowledge today.
Unlock this book and the full library FREE for 7 days
Get unlimited access to 7000+ expert-authored eBooks and videos courses covering every tech area you can think of
Renews at $15.99/month. Cancel anytime}