最近在研究Stylegan对抗式图像生成网络,使用了网络的一些预训练模型生成相应的图像,感觉非常有趣。下面开源一些我找到了预训练模型和代码,供大家一起玩。
Stylegan2官方给出的是TensorFlow版本的,费了半天劲找出了pytorch版本
这里给出下载链接:https://pan.baidu.com/s/1ktGqfnGxtYSr-5ao1_se6w 提取码:5g7z
生成命令如下:
python generate.py --outdir=生成图片的文件夹 --seeds=0-图像数目 --class=1 --network=模型名称
同时,使用命令执行生成图片的时候,记得使用命令台,使用run的方式没法生成,这其中的难受,试过的都知道。
开源模型:
黄种人脸预训练模型(generator_yellow-stylegan2-config-f.pkl):
下载链接:https://pan.baidu.com/s/1gyWgNZngheZnZJMmYQpXVg 提取码:bljc
网红脸预训练模型(generator_wanghong-stylegan2-config-f.pkl):
下载链接:https://pan.baidu.com/s/1GN-SzsDBdthijFQEnXuRmA 提取码:6xxa
萌娃脸预训练模型(generator_baby-stylegan2-config-f.pkl):
下载地址:https://pan.baidu.com/s/10qWI_g5b0AQId06BbmO4jg 提取码:5nj8
明星脸预训练模型(generator_star-stylegan2-config-f.pkl):
下载链接:https://pan.baidu.com/s/19_Iku6a7nUjStCK1eBDMNw 提取码:k4e7
超模脸生成器(generator_model-stylegan2-config-f.pkl):
下载链接:https://pan.baidu.com/s/18OMIDsX_vdYPsj2CfEN0Mw 提取码:t8xm