Indi Model Training Repository

Introduction

This repository contains the code for training the Indi model, an image processing model designed for image to image translation. This is a personal implementation of the Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restoration (InDI) paper by Delbracio et al 2023. The code is organized into three main files:

• training.py: The main script for training the model.
• indi_diffusion.py: Contains the Indi class and utility functions.
• modules.py: Includes essential modules and classes used in the model.

Examples

Getting Started

Prerequisites

List of libraries to run the project:

• Python 3.8+
• PyTorch
• PIL
• NumPy
• tqdm

Installation

pip install torch numpy pillow tqdm

Results and Output

This implementation gives you an idea on how the InDI architeture work and it is made up for image to image translation tasks. For more information about it read the paper https://doi.org/10.48550/arXiv.2303.11435

Contact

Contact me at [email protected]

References

• https://doi.org/10.48550/arXiv.2303.11435
• https://github.com/dome272/Diffusion-Models-pytorch