imagen-pytorch VS stylegan2

If google doesnt, someone like lucidrains probably would implement it, just like he did for imagen and muse.

lucidrains/imagen-pytorch: Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch (github.com)

Hi Can someone guide me a little, as to how i can use LAION dataset to train my imagen model? like how i can download the data, and in which format it should be fed to https://github.com/lucidrains/imagen-pytorch code?

If we were to put money into training something, I'd hope we use a better model, like Imagen.

I've been messing around with this open-source implementation. You can get a pretty good idea of the model size by just copying the parameters from the paper.

I'm waiting on this implementation/training of imagen: https://github.com/lucidrains/imagen-pytorch

I'm honestly surprised that they trained a StyleGAN. Recently, the Imagen architecture has been show to be both easier in structure, easier to train, and even faster to produce good results. Combined with the "Elucidating" paper by NVIDIA's Tero Karras you can train a 256px Imagen to tolerable quality within an hour on a RTX 3090.

Here's a PyTorch implementation by the LAION people:

https://github.com/lucidrains/imagen-pytorch

And here's 2 images I sampled after training it for some hours, like 2 hours base model + 4 hours upscaler:

https://imgur.com/a/46EZsJo

I don't know. If you're really curious, you can just try it: https://github.com/NVlabs/stylegan2

StyleGAN2 (Dec 2019) - Karras et al. and Nvidia

> The denoising part of a denoising autoencoder refers to the noise applied to its input

Agree, it converts a noisy image to a denoised image. But the odd thing is, when you put a noisy image into a StyleGAN2 encoder, you get latents which the decoder will turn into a de-noised image. So in practical use, you can take a trained StyleGAN2 encoder/decoder pair and use it as if it was a denoiser.

> These differences lead to learned distributions in the latent space that are entirely different

I also agree there. The training for a denoising auto-encoder and for a GAN network is different, leading to different distributions which are sampled for generating the images. But the architecture is still very similar, meaning the limits of what can be learned should be the same.

> Beyond that the comparison just doesn't work, yes there are two networks but the discriminator doesn't play the role of the AE's encoder at all

Yes, the discriminator in a GAN won't work like an encoder. But if you look at how StyleGAN 1/2 are used in practice, people combine it with a so-called "projection", which is effectively an encoder to convert images to latents. So people use a pipeline of "image to latent encoder" + "latent to image decoder".

That whole pipeline is very similar to an auto-encoder. For example, here's an NVIDIA paper about how they round-trip from image to latent to image with StyleGAN: https://arxiv.org/abs/1912.04958 My interpretation of what they did in that paper is that they effectively trained a StyleGAN-like model with the image L2 loss typically used for training a denoising auto-encoder.

These problems are usually done with CNN Encoder-Decoder frameworks. Usually GAN (Generative Adversarial Networks see StyleGan2).

This is the result of two different models. The first looks like a latent space interpolation of StyleGan2 and the mouth movements are without a doubt from wav2lip.

OP: if you want to run this at higher resolution, you should probably look at running it yourself, using something like this: https://github.com/NVlabs/stylegan2

StyleGAN2 (Dec 2019) - Karras et al. and Nvidia

A few of you might've seen an earlier post of mine about this project (Or the repost that got more upvotes 🙃), and I've improved the code and network since then after more thoroughly reading and understanding the official StyleGAN2 implementation.

So I've been a Pro+ subscriber since around November which I mainly use to train GANs. I have multiple Google accounts, let's call them Account 1, 2, and 3. Accounts 1 and 2 are normal Google accounts and Account 3 is an account I got from my university after I graduated which has unlimited storage.