Pegasus Image Generating GAN
- Category: ML model
- Client: Durham University
- Project date: April 2021
Summary
A DCGAN implemented in PyTorch used to turn images of planes and horses in CIFAR-10 into pegasus images The DCGAN was then experimented on to see how different modifications effect results.
STAR Breakdown
- Situation: In order to see the effect of hybridizing different modifications on a DCGAN, I wanted to see how realistic I could get a picture of pegasus made with AI.
- Task: I was asked to develop a ML model for creating pegasus images but was not instructed specifically how nor how I should experiment on it to try and tweak results.
- Action: I first explored the data in the CIFAR-10 dataset and isolated the images of horses, planes and birds. I decided to use planes instead of birds as a method of implementing wings due to many bird images not having spread wings nor being white in colour. I then created the layered convolutions in the DCGAN and fed the images into it. The resulting image batches were promising and so I began tweaking various metrics to try and improve the result.
- Result: My written report on the experiments discusses the experiments made and their effects as well as contains the final generated images.