Conditional Diffusion Model

조건부 분해 모델

CIFAR10에서 훈련 된 조건부 DDPM을 구현하기위한 코드

개요

이 코드는 기본 거부 확산 프로모어 모델 (DDPM)을 수정하여 CIFAR-10 데이터 세트에서 교육 된 조건부 DDPM을 생성하여 다른 입력 범주 레이블의 조합에서 합성 이미지를 생성 할 수 있습니다.

프로젝트 구조

1. CIFAR_DDPM.PY- 조건부 DDPM을 구현하고 훈련시킵니다

2. image_generator.py- 훈련 된 모델을 사용한 입력 조건에 따라 생성 이미지를 생성합니다.

작가

루이스 샤포 사우 더스