Conditional Diffusion Model

Modèle de diffusion conditionnelle

Code pour implémenter un DDPM conditionnel formé sur CIFAR10

Aperçu

Ce code modifie un modèle probaitique de diffusion de dénoue (DDPM) de base pour créer un DDPM conditionnel formé sur l'ensemble de données CIFAR-10 qui est capable de générer des images synthétiques à partir d'une combinaison de différentes étiquettes de catégorie d'entrée

Structure du projet

1. cifar_ddpm.py - implémente et entraîne le DDPM conditionnel

2. Image_Generator.py - produit des images génératives en fonction des conditions d'entrée à l'aide du modèle formé

Auteur

Louis Chapo-Saunders