上传者: 42128988
|
上传时间: 2021-07-12 10:22:24
|
文件大小: 64.32MB
|
文件类型: ZIP
U-Net GAN PyTorch
PyTorch实施CVPR 2020论文“用于生成对抗网络的基于U网络的鉴别器”。 论文和补充资料可以在找到。 别忘了看一下补充文件(可以在此处找到Tensorflow FID(表S1))。 该代码允许用户重现和扩展研究报告的结果。 报告,复制或扩展结果时,请引用上述论文。
设置
从提供的unetgan.yml文件创建conda环境“ unetgan”。 可以使用training_scripts文件夹中提供的脚本来复制实验(必须手动设置实验文件夹和数据集文件夹)。
争论
解释
--unconditional
如果数据集没有类,请使用此选项(例如CelebA)。
--unet_mixup
使用CutMix。
--slow_mixup
对CutMix增强损失使用预热。
--slow_mixup_epochs
预热的时期数
--full_ba