改善扩散
这是的代码库。
用法
README的这一部分将逐步介绍如何训练模型并从模型中取样。
安装
克隆该存储库,然后在您的终端中导航至该存储库。 然后运行:
pip install -e .
这应该安装脚本所依赖的improved_diffusion python软件包。
准备资料
训练代码从图像文件目录中读取图像。 在文件夹中,我们提供了用于为ImageNet,LSUN卧室和CIFAR-10准备这些目录的说明/脚本。
要创建自己的数据集,只需将所有图像转储到扩展名为“ .jpg”,“。jpeg”或“ .png”的目录中即可。 如果您希望训练一个类条件模型,则将文件命名为“ mylabel1_XXX.jpg”,“ mylabel2_YYY.jpg”等,以便数据加载器知道“ mylabel1”和“ mylabel2”是标签。 子目录也会自动枚举,因此可以将图像组织为递归结构(尽管目录名
2024-04-29 11:21:14
45KB
Python
1