pytorch-pretrained-BigGAN

收录理由

这个仓库把 DeepMind 的 BigGAN 用 PyTorch 重新实现了一遍，和原始 TensorFlow 版本在计算图上保持一致，生成结果的差异大约在 1e-5 量级。它直接提供了 128、256、512 三种分辨率的预训练生成器权重，加载模型就能生成 ImageNet 级别的图像，完全不需要自己训练。仓库里还附带了从 TensorFlow Hub 下载权重并转换的脚本，如果你之前折腾过 TF 到 PyTorch 的模型转换，就知道这能省下多少工夫。有一点得说清楚：由于 DeepMind 只发布了生成器权重，所以这个实现里只有生成器，判别器部分没有包含，不过两者结构相似，真要加也不算太难。无论你是想搞生成艺术、做合成训练数据，还是想研究大型 GAN 的架构细节，这份代码都很精简，值得读一读。