当前位置：首页 > article >正文

开源Stylegan人脸生成预训练模型

article 2024/9/27 7:25:58

最近在研究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