Conditional Diffusion Model

Modelo condicional-difusão

Código para implementar um DDPM condicional treinado no CIFAR10

Visão geral

Este código modifica um modelo probatístico de difusão básico (DDPM) para criar um DDPM condicional treinado no conjunto de dados CIFAR-10, capaz de gerar imagens sintéticas a partir de uma combinação de diferentes rótulos de categoria de entrada

Estrutura do projeto

1. cifar_ddpm.py - implementa e treina DDPM condicional

2. image_generator.py - produz imagens generativas de acordo com as condições de entrada usando o modelo treinado

Autor

Louis Chapo-Saunders