Turning my Dog into a Person with TensorFlow and StyleGAN2-ADA on Windows
StyleGAN: The Gateway Drug
Earlier this year, I started looking into something called StyleGAN which is a machine-learning project that was made public by Nvidia in early 2019.
These people are not real – they were generated by StyleGAN
This naturally led me to find stylegan2 which added improvements to image quality, and then to the even newer stylegan2-ada which adds even more improvements when working with smaller data-sets.
Getting Started (on Windows)
One of the coolest things about ML projects is that many are hosted online using google colab, meaning that you don't need to install or configure anything on your computer. And that's fun and all, but I wanted to get my hands dirty and make use of some idle GPUs.
Requirements
First, a couple things need to be installed:
- Nvidia CUDA Toolkit
- Anaconda python
- Download at least one pre-trained network (PKL file)
- Clone StyleGAN2-ADA repository
Experiment: Turn my dog into a person
The first thing I thought to do, mainly because so many data sets are based on human faces, was to see what would happen if I provided an image of my dog as the input:
