Generative Modelling: Generating Realistic Images with Hardware Limitations and Interpolating Between Them.
Submitted as part of the degree of Msci Natural Sciences (3rd year) to the Board of Examiners in the Department of Computer Sciences, Durham University. This summative assignment was assessed and marked by the professor of the module in question:
"This is a perfect GAN-based submission, where you’ve clearly done a significant amount of research by reading of state-of-the-art peer-reviewed literature from top venues (CVPR, ICCV, NeurIPS). The samples have excellent diversity with very good coverage and few issues, usually only noticable in the backgrounds at the sides. The model samples are amongst the best in the class - I’ve used several of them as exemplars in the general feedback. The interpolations clearly show the model captures the data distribution, with a smooth and intuitive generated images between a diverse set of samples." - Dr Chris G. Willcocks
This paper proposes using a StyleGAN2 architecture in combination with an adaptive discriminator augmentation (ADA) mechanism trained on the FFHQ dataset. This mechanism passes the images shown to the discriminator through an augmentation pipeline and dynamically adjusts the strength of augmentation. We produce realistic interpolation and achieve a reasonable balance between diversity and quality of images
All models were implemented with the goal of generating diverse and realistic images from the FFHQ dataset given severe computational limitations (access to one NVIDIA TITAN Xp or an NVIDIA GeForce RTX 2080 for a maximum of three training days).
- styleGAN2_paper.pdf - Paper reporting findings and methodology of the study, "Generating Images using StyleGAN2 + ADA".
- source_code/ - Folder containing the modified version of the state-of-the-art StyleGAN2 + ADA proposed in ICLR 2021 (this source-code has yet to be cleaned and commented, apologies). Credits for paper and code go to:
- StyleGAN2 + ADA paper - Training Generative Adversarial Networks with Limited Data.
- lucidrains' original implementation - https://github.com/NVlabs/stylegan2-ada-pytorch.