improved-diffusion:发布改进的降噪扩散概率模型

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

文件下载

资源详情

[{"title":"( 25 个子文件 45KB ) improved-diffusion:发布改进的降噪扩散概率模型","children":[{"title":"improved-diffusion-main","children":[{"title":"setup.py <span style='color:#111;'> 168B </span>","children":null,"spread":false},{"title":"improved_diffusion","children":[{"title":"resample.py <span style='color:#111;'> 5.56KB </span>","children":null,"spread":false},{"title":"respace.py <span style='color:#111;'> 4.80KB </span>","children":null,"spread":false},{"title":"gaussian_diffusion.py <span style='color:#111;'> 31.19KB </span>","children":null,"spread":false},{"title":"train_util.py <span style='color:#111;'> 12.33KB </span>","children":null,"spread":false},{"title":"dist_util.py <span style='color:#111;'> 1.91KB </span>","children":null,"spread":false},{"title":"__init__.py <span style='color:#111;'> 74B </span>","children":null,"spread":false},{"title":"script_util.py <span style='color:#111;'> 7.68KB </span>","children":null,"spread":false},{"title":"fp16_util.py <span style='color:#111;'> 2.23KB </span>","children":null,"spread":false},{"title":"unet.py <span style='color:#111;'> 18.38KB </span>","children":null,"spread":false},{"title":"nn.py <span style='color:#111;'> 4.90KB </span>","children":null,"spread":false},{"title":"image_datasets.py <span style='color:#111;'> 4.02KB </span>","children":null,"spread":false},{"title":"logger.py <span style='color:#111;'> 13.65KB </span>","children":null,"spread":false},{"title":"losses.py <span style='color:#111;'> 2.47KB </span>","children":null,"spread":false}],"spread":false},{"title":".gitignore <span style='color:#111;'> 24B </span>","children":null,"spread":false},{"title":"datasets","children":[{"title":"lsun_bedroom.py <span style='color:#111;'> 1.68KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 2.59KB </span>","children":null,"spread":false},{"title":"cifar10.py <span style='color:#111;'> 944B </span>","children":null,"spread":false}],"spread":true},{"title":"LICENSE <span style='color:#111;'> 1.04KB </span>","children":null,"spread":false},{"title":"scripts","children":[{"title":"image_nll.py <span style='color:#111;'> 2.87KB </span>","children":null,"spread":false},{"title":"image_train.py <span style='color:#111;'> 2.25KB </span>","children":null,"spread":false},{"title":"image_sample.py <span style='color:#111;'> 3.27KB </span>","children":null,"spread":false},{"title":"super_res_train.py <span style='color:#111;'> 2.64KB </span>","children":null,"spread":false},{"title":"super_res_sample.py <span style='color:#111;'> 3.59KB </span>","children":null,"spread":false}],"spread":true},{"title":"README.md <span style='color:#111;'> 8.03KB </span>","children":null,"spread":false}],"spread":true}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明