手撕Diffusion系列 - 第三期 - Dataset 目录 手撕Diffusion系列 - 第三期 - Dataset DDPM 原理图 MNIST数据集介绍 主要特点 数据预处理 Dataset 代码 Part1 引入相关库函数 Part2 MNIST数据集的预处理以及获取 Part3 测试 参考 DDPM 原理图 DDPM包括两个过程:前向过程(forward process)和反向过程(reverse process),其中前向过程又称为扩散过程(diffusion process),如下图所示。无论是前向过程还是反向过程都是一个参数化的马尔可夫链(Markov chain),其中反向过程可以用来生成图片。 DDPM 整体大概流程 图中,由高斯随机噪声 x T x_T xT 生成原始图片 x 0 x_0 x