home diffusion ddpm

View code on Github

去噪扩散概率模型 (DDPM)

这是《去噪扩散概率模型》论文的 PyTorch 实现/教程。

简而言之，我们从数据中获取图像并逐步添加噪点。然后，我们训练一个模型来预测每个步骤的噪声，并使用该模型生成图像。

这是预测噪声和训练代码的 UNet 模型。此文件可以从经过训练的模型生成样本和插值。