谷歌“史上最强GAN”,现在有了PyTorch预训练版,可直接玩耍 | 代码( 二 )

128×128,256×256,512×512。

抱抱脸团队说,模型的参数都是BigGAN的爸爸DeepMind官方训练的成果。

谷歌“史上最强GAN”,现在有了PyTorch预训练版,可直接玩耍 | 代码

团队说,他们是用官方的原始计算图 (Computation Graph) 来复现的,与原模型的表现几乎无差:输出差异 (Output Difference) 的方差在10^-5级。

官方模型是放在TensorFlow Hub上,抱抱脸还提供了把TF模型转成PyTorch模型时,用到的脚本。

更加温柔的是,最后会显示生成效果:

谷歌“史上最强GAN”,现在有了PyTorch预训练版,可直接玩耍 | 代码

肉眼看去,成果喜人。

食用方法如果只是想随意玩耍的话,pip install就够了。

如果要用前面提到的转换脚本,以及ImageNet实用程序的话,就要再安装一些依赖项。记得要用full_requirements.txt来装:

推荐阅读