![cec97b64cc984218837b77ba28304cdc.png](https://img-blog.csdnimg.cn/img_convert/cec97b64cc984218837b77ba28304cdc.png)
魔栗 发自 凹非寺
量子位 报道 | 公众号 QbitAI
你知道么,和BigGAN一起玩耍,会上瘾的。
比如,生成了一只狗,再生成了一只汉堡。
![1c49f6c0896a3bbe67c8cbf203a58f9d.png](https://img-blog.csdnimg.cn/img_convert/1c49f6c0896a3bbe67c8cbf203a58f9d.png)
那么,狗 × 汉堡 = ?
![2c15888760da9437e4a0c18c4a7d9ad5.png](https://img-blog.csdnimg.cn/img_convert/2c15888760da9437e4a0c18c4a7d9ad5.png)
一看就是亲生的。
现在,身为PyTorch用户的你,也可以拥有一只BigGAN,而且不用自己训练,便能直接玩耍。
一向以造福人类为己任的抱抱脸 (Hugging Face) 团队,用PyTorch复现了这个“史上最强”GAN。
团队开源了预训练模型,只要pip install一下,你有什么大胆想法,就可以实施了。
推特用户纷纷表示欢迎:
![83c2eb5e5b1a27020305a514ed51b027.png](https://img-blog.csdnimg.cn/img_convert/83c2eb5e5b1a27020305a514ed51b027.png)
还原度极高
开源项目里有三个模型,是不同分辨率的bigGAN:
128×128,256×256,512×512。
抱抱脸团队说,模型的参数都是BigGAN的爸爸DeepMind官方训练的成果。
![496f980b952b8dfdea1279eb85cc2786.png](https://img-blog.csdnimg.cn/img_convert/496f980b952b8dfdea1279eb85cc2786.png)
团队说,他们是用官方的原始计算图 (Computation Graph) 来复现的,与原模型的表现几乎无差:输出差异 (Output Difference) 的方差在10^-5级。
官方模型是放在TensorFlow Hub上,抱抱脸还提供了把TF模型转成PyTorch模型时,用到的脚本。
更加温柔的是,最后会显示生成效果:
![a4c22a29b6fd0ca180278363b071e8c5.png](https://img-blog.csdnimg.cn/img_convert/a4c22a29b6fd0ca180278363b071e8c5.png)
肉眼看去,成果喜人。
食用方法
如果只是想随意玩耍的话,pip install就够了。
如果要用前面提到的转换脚本,以及ImageNet实用程序的话,就要再安装一些依赖项。记得要用full_requirements.txt来装:
1
128×128模型,有5040多万参数;256×256模型,有5590多万参数;512×512模型,有5620多万参数。三个模型,大小都在200~Mb。
安装之后,正式开始食用:
1
到这里,图像就愉快地生成了。
那么,你有大胆的想法了么?
不如,“红酒烩鸡”?
![3413d7656fb8f2ca833191fe68a9719e.png](https://img-blog.csdnimg.cn/img_convert/3413d7656fb8f2ca833191fe68a9719e.png)
代码传送门:
https://github.com/huggingface/pytorch-pretrained-BigGAN
— 完 —
量子位 · QbitAI
վ'ᴗ' ի 追踪AI技术和产品新动态
戳右上角「+关注」获取最新资讯↗↗
如果喜欢,请分享or点赞吧~比心❤