homediffusionddpm

View code on Github

#

Denoising Diffusion Probabilistic Models (DDPM)

This is a PyTorch implementation/tutorial of the paper Denoising Diffusion Probabilistic Models.

In simple terms, we get an image from data and add noise step by step. Then We train a model to predict that noise at each step and use the model to generate images.

Here is the UNet model that predicts the noise and training code. This file can generate samples and interpolations from a trained model.

labml.ai