文件名称:pytorch-gan-collections:DCGAN,WGAN-GP和SNGAN的PyTorch实施
文件大小:22KB
文件格式:ZIP
更新时间:2024-05-19 04:05:26
pytorch dcgan wgan-gp sngan fid-score
GAN的集合 无监督GAN的Pytorch实现。 有关使用pytorch计算初始分数和FID的更多信息,请参见 楷模 直流电源 威根 WGAN-GP SN-GAN 要求 初始化metric模型git submodule update --init 安装python包pip install -U pip setuptools pip install -r requirements.txt 结果 模型 数据集 起始分数 FID 直流电源 CIFAR10 6.04(0.05) 47.90 WGAN(CNN) CIFAR10 6.64(0.6) 33.27 WGAN-GP(CNN) CIFAR10 7.47(0.06) 24.00 WGAN-GP(ResNet) CIFAR10 7.74(0.10) 21.89 SNGAN(CNN) CIFAR10 7
【文件预览】:
pytorch-gan-collections-master
----dcgan.py(8KB)
----source()
--------models()
--------utils.py(891B)
--------losses.py(2KB)
----Readme.md(4KB)
----.gitmodules(102B)
----metrics()
----requirements.txt(75B)
----config()
--------WGANGP_CIFAR10_CNN.txt(286B)
--------WGANGP_CIFAR10_RES.txt(286B)
--------SNGAN_CIFAR10_CNN.txt(274B)
--------DCGAN_CIFAR10.txt(268B)
--------SNGAN_CIFAR10_RES.txt(273B)
--------WGAN_CIFAR10_CNN.txt(279B)
----wgan.py(8KB)
----tools()
--------plotcurve.py(1KB)
--------sample2gif.py(923B)
----sngan.py(8KB)
----wgangp.py(9KB)
----.gitignore(103B)