文件名称:pytorch-pretrained-BigGAN:Big具有预先训练的权重和转换脚本的BigGAN的PyTorch实施
文件大小:323KB
文件格式:ZIP
更新时间:2024-06-11 01:24:08
computer-vision neural-network pytorch artificial-intelligence generative-adversarial-network
PyTorch预训练的BigGAN 使用DeepMind的预训练权重对DeepMind的BigGAN模型进行逐个PyTorch重新实现。 介绍 该存储库包含DeepMind的BigGAN的PyTorch互操作实现,并由Andrew Brock,Jeff Donahue和Karen Simonyan发表了论文《 。 DeepMind的此PyTorch实施提供了。 我们还提供了用于从TensorFlow Hub模型下载和转换这些模型的脚本。 此重新实现是从Tensorflow版本的原始计算图完成的,其行为与TensorFlow版本类似(输出差异的方差为1e-5)。 目前该实现仅包含生成器,因为未释放鉴别器的权重(尽管鉴别器的结构与生成器非常相似,因此可以很容易地添加它。告诉我是否要对此进行PR,我很乐意为您提供帮助。) 安装 此仓库已在Python 3.6和PyTorch 1.0.1上
【文件预览】:
pytorch-pretrained-BigGAN-master
----MANIFEST.in(16B)
----pytorch_pretrained_biggan()
--------file_utils.py(8KB)
--------config.py(2KB)
--------utils.py(22KB)
--------model.py(13KB)
--------__init__.py(306B)
--------convert_tf_to_pytorch.py(14KB)
----full_requirements.txt(53B)
----scripts()
--------download_tf_hub_models.sh(498B)
--------convert_tf_hub_models.sh(499B)
----requirements.txt(173B)
----LICENSE(1KB)
----assets()
--------output_0.png(87KB)
--------output_2.png(112KB)
--------output_1.png(93KB)
----setup.py(3KB)
----README.md(11KB)
----.gitignore(1KB)