ionvision / diffusion_models

Minimal standalone example of diffusion model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Diffusion models

Open In Colab

A minimal standalone example of diffusion model. The objective is to understand the forward as well as reverse mapping process of diffusion models. The notebook contains the equations along side the code and some visualizations.

Hope you find it useful!

The forward trajectory:

Forward diffusion

Reverse trajectory:

Reverse diffusion

About

Minimal standalone example of diffusion model

License:MIT License


Languages

Language:Jupyter Notebook 99.5%Language:Python 0.5%