文件名称：stylegan2-ada-pytorch:StyleGAN2-ADA-官方PyTorch实施
文件大小：1.12MB
文件格式：ZIP
更新时间：2024-03-27 11:09:23
Python
StyleGAN2-ADA —官方PyTorch实施 用有限的数据训练生成对抗网络Tero Karras，Miika Aittala，Janne Hellsten，Samuli Laine，Jaakko Lehtinen，Timo Aila 摘要：使用太少的数据训练生成对抗网络（GAN）通常会导致判别器过度拟合，从而导致训练分散。 我们提出了一种自适应鉴别器增强机制，该机制可显着稳定有限数据环境中的训练。 该方法不需要更改丢失功能或网络体系结构，并且适用于从头开始训练以及在其他数据集上对现有GAN进行微调时。 我们在几个数据集上证明，仅使用几千个训练图像就可以取得良好的效果，并且通常将StyleGAN2的结果与较少的图像数量级进行匹配。 我们希望这将为GAN开辟新的应用程序域。 我们还发现，实际上，广泛使用的CIFAR-10只是一个有限的数据基准，并将记录FID从5.59提高到2.4
【文件预览】：
stylegan2-ada-pytorch-main
----train.py(24KB)
----.github()
--------ISSUE_TEMPLATE()
----LICENSE.txt(4KB)
----dataset_tool.py(17KB)
----generate.py(5KB)
----metrics()
--------perceptual_path_length.py(5KB)
--------inception_score.py(2KB)
--------precision_recall.py(4KB)
--------__init__.py(435B)
--------frechet_inception_distance.py(2KB)
--------metric_main.py(6KB)
--------kernel_inception_distance.py(2KB)
--------metric_utils.py(12KB)
----Dockerfile(897B)
----dnnlib()
--------util.py(16KB)
--------__init__.py(478B)
----calc_metrics.py(8KB)
----README.md(25KB)
----projector.py(9KB)
----docker_run.sh(1KB)
----torch_utils()
--------ops()
--------misc.py(11KB)
--------custom_ops.py(6KB)
--------__init__.py(438B)
--------persistence.py(9KB)
--------training_stats.py(10KB)
----training()
--------augment.py(26KB)
--------loss.py(7KB)
--------dataset.py(8KB)
--------__init__.py(435B)
--------training_loop.py(21KB)
--------networks.py(37KB)
----docs()
--------license.html(5KB)
--------stylegan2-ada-training-curves.png(496KB)
--------stylegan2-ada-teaser-1024x252.png(531KB)
--------dataset-tool-help.txt(2KB)
--------train-help.txt(4KB)
----.gitignore(21B)
----legacy.py(16KB)
----style_mixing.py(5KB)