jingyaogong / PyTorch-DDPM

500 行代码实现降噪扩散模型 DDPM,干净无依赖

Home Page:https://timecat.notion.site/DDPM-b8e2a91927d249fdbcf7c82f2eb6f846

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

500 行代码实现降噪扩散模型 DDPM,干净无依赖

https://timecat.notion.site/DDPM-b8e2a91927d249fdbcf7c82f2eb6f846


Languages

Language:Jupyter Notebook 99.4%Language:Python 0.6%