PyTorch-DDPM

500 行代码用 PyTorch 实现降噪扩散模型 DDPM

建议使用 codelab 打开 notebook，可以不用自己配环境了

codelab: DDPM

codelab: Classifier-Free DDPM

DDPM

从随机噪声中降噪生成图片

运行 python ddpm.py 或者打开 notebook DDPM.ipynb

条件控制 DDPM：给定一个数字，根据数字生成图片

运行 python classifier_free_ddpm.py 或者打开 notebook ClassifierFreeDDPM.ipynb

Name		Name	Last commit message	Last commit date
Latest commit History 15 Commits
assets		assets
ClassifierFreeDDPM.ipynb		ClassifierFreeDDPM.ipynb
DDPM.ipynb		DDPM.ipynb
README.md		README.md
classifier_free_ddpm.py		classifier_free_ddpm.py
ddpm.py		ddpm.py