How to Train StyleGAN2-ADA with Custom Dataset

Learn how to train an AI to generate any images you want

Fathy Rashad
Towards Data Science
5 min readJun 7, 2021

--

Generated Bike Interpolation [Image by Author]

Have you always wondered how to train your own generative model? The first time I discovered GAN applications like this face generating website, I always wondered how to train GAN on other things. Luckily, I recently had the opportunity to train a bike generating model as part of my research. In this article, I will document my experience on how to train StyleGAN2-ADA on your own images.

StyleGAN is one of the most popular generative models by NVIDIA. Multiple version of StlyeGAN has been released and we will be using the latest version which is StyleGAN2-ADA. To avoid redundancy, I won’t explain StyleGAN as there are many articles that have explained it really well.

--

--