Diffusion Model

介绍

扩散模型复现，仅用于个人学习

软件架构

dataste.py 用来处理处理集，使用mnist手写数据集

ddpm.py 实现扩散模型

network.py 实现unet神经网络

main.py 实现训练

diffusionmodel.ipynb 可视化加噪去噪过程

参考文献

https://segmentfault.com/a/1190000043744225#item-1

https://blog.csdn.net/DFCED/article/details/132394895

https://blog.csdn.net/wshzd/article/details/132347788