比较 DDPM、DDIM、SDE、Flow Matching 等扩散模型,统一 data / utils / output,各扩散过程在 diffusion/ 下按需扩展。
不写统一下载脚本,数据由自己准备、放到 data/ 下即可。
- CIFAR-10 / CIFAR-100
- 自定义数据集(如 256×256 图片)
- 自己下载 / 爬取后,在
data/下建一个子目录(如data/my_256/),把图片放进去(可再分子目录当类别)。 - 在 config 里把
data.dataset_name设为该目录名、data.img_size设为 256 即可。
- 自己下载 / 爬取后,在
如需显式先下载 CIFAR 再训练,可运行:
python -m scripts.data.download_cifar(会下载到 data/,之后训练直接用。)