This repository contains code for training Deep Image Prior (DIP) and Denoising Diffusion Probabilistic Models (DDPM) on images stored in the /imgs
directory.
To install the necessary dependencies, use the provided requirements.txt
file. You can do this by running:
pip install -r requirements.txt
To train the Deep Image Prior (DIP) model, run the DIP_train.py
script:
python DIP_train.py
To train the Denoising Diffusion Probabilistic Model (DDPM), run the main.py
script:
python main.py
Your project directory should look like this:
/imgs
- image1.png
- image2.png
...
DIP_train.py
main.py
requirements.txt
README.md
This project is inspired by and references code from the following repositories: