Skip to content

My notebook reimplementation of a U-Net diffusion model based on the original paper

Notifications You must be signed in to change notification settings

AnanyaHN27/diffusionModel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Diffusion Model Implementation

This notebook implements a diffusion model from scratch, inspired by this guide. The goal of this project is to demonstrate the process of building and training a basic diffusion model on the MNIST dataset without any pre-trained models or specialized purposes.

Contents

  • Introduction: A diffusion model is a type of generative model that learns to reverse a noising process applied to data, here demonstrated on MNIST images.
  • Architecture: This implementation follows the architecture and methodology described in the linked guide, providing insight into how diffusion models work on a fundamental level.

Getting Started

Prerequisites

Ensure you have the following libraries installed:

  • tensorflow
  • numpy
  • tqdm

You can install these using:

pip install tensorflow numpy tqdm

About

My notebook reimplementation of a U-Net diffusion model based on the original paper

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published