文件名称:GAN-pytorch:pytorch中几种GAN算法的实现
文件大小:40.43MB
文件格式:ZIP
更新时间:2024-04-26 15:59:39
Python
素食主义者 一个库,可以轻松地在PyTorch中训练各种现有的GAN(生成对抗网络)。 该库主要针对GAN用户,他们希望将现有的GAN培训技术与自己的生成器/区分器一起使用。 但是,研究人员可能还会发现GAN基类对于更快地实施新的GAN训练技术很有用。 重点是简单性并提供合理的默认值。 如何安装 您需要python 3.5或更高版本。 然后: pip install vegans 如何使用 基本思想是用户提供区分器和生成器网络,而库则负责在选定的GAN设置中训练它们: from vegans.models.GAN import WassersteinGAN from vegans.utils import plot_losses, plot_images generator = ### Your generator (torch.nn.Module) adversariat = ##
【文件预览】:
GAN-pytorch-main
----requirements()
--------requirements.txt(2KB)
----.DS_Store(6KB)
----utils()
--------plot2DModel.py(2KB)
--------utils.py(5KB)
--------layers.py(710B)
--------networks.py(4KB)
----models()
--------GAN.py(431B)
--------conditional()
--------unconditional()
----examples()
--------example_conditional.py(4KB)
--------example_unconditional.py(3KB)
--------example_image_to_image.py(5KB)
--------example_input_formats.py(7KB)
----README.md(7KB)
----data()
--------.DS_Store(6KB)
--------mnist()
--------fashion_mnist()
----.gitignore(2KB)