Imported from https://github.com/manicman1999/StyleGAN-Tensorflow-2.0