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