Skip to content

This repository implements a Denoising Diffusion Probabilistic Model (DDPM) for image generation using TensorFlow and Keras. The model is trained on the Oxford-IIIT Pet dataset and includes a custom-implemented time-aware U-Net model with ResNet blocks and self-attention mechanism.

Notifications You must be signed in to change notification settings

Shreehar01/Image-Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This repository implements a Denoising Diffusion Probabilistic Model (DDPM) for image generation using TensorFlow and Keras. The model is trained on the Oxford-IIIT Pet dataset and includes a custom-implemented time-aware U-Net model with ResNet blocks and self-attention mechanism.

How to Run

  1. Clone the repository and dataset:

    git clone https://github.com/Shreehar01/Image-Generator.git
    cd Image-Generator
    git clone https://github.com/ml4py/dataset-iiit-pet.git
  2. Install dependencies:

    pip install -r requirements.txt
  3. Start training:

    python main.py

    The script will create a directory named epoch_outputs and save generated images there every 10 epochs.

About

This repository implements a Denoising Diffusion Probabilistic Model (DDPM) for image generation using TensorFlow and Keras. The model is trained on the Oxford-IIIT Pet dataset and includes a custom-implemented time-aware U-Net model with ResNet blocks and self-attention mechanism.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages