Skip to content

LinXueyuanStdio/PyTorch-DDPM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch-DDPM

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

如何实现?可参考个人精简后的公式: https://timecat.notion.site/DDPM-b8e2a91927d249fdbcf7c82f2eb6f846

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

codelab: DDPM

codelab: Classifier-Free DDPM

DDPM

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

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

Classifier-Free DDPM

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

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