1
头图

🎈 什么是 StyleGAN

  • GAN 是机器学习中的生成性对抗网络,目标是合成与真实图像无法区分的人工样本,如图像。即改变人脸图像中的特定特征,如姿势、脸型和发型,GAN 的主要挑战就是如何图像变得更加逼真。
  • StyleGAN 是一种基于样式的 GAN 的生成器体系结构,是一个强大的可以控制生成图片属性的框架,它采用了全新的生成模型,分层的属性控制,能够生成 1024×1024 分辨率的人脸图像,并且可以进行属性的精确控制与编辑,可以生成最先进的结果—高分辨率图像,看起来比以前生成的图像更真实。

用 StyleGAN 技术生成的明星人脸图

🎈 如何使用 StyleGAN

安装 Anaconda

安装 CUDA 10.0

安装 cuDNN

安装 VS2019

🎈 下载项目

# 下载项目库
git clone https://github.com/a312863063/generators-with-stylegan2.git

# 安装依赖
pip install tensorflow-gpu==1.14.0
pip install scipy==1.3.3
pip install requests==2.22.0
pip install Pillow==6.2.1    

🎈 修改项目 MSVC

  • 需要打开项目中的文件 dnnlib/tflib/custom_ops.py,第29行所标注的 MSVC 的版本号
  • 这个依据自身电脑安装的版本进行修改便可,一般只需要修改下图的 14.16.27023 的位置即可
compiler_bindir_search_path = [
    'C:/Program Files (x86)/Microsoft Visual Studio/2017/Community/VC/Tools/MSVC/14.16.27023/bin/Hostx64/x64',
    'C:/Program Files (x86)/Microsoft Visual Studio/2019/Community/VC/Tools/MSVC/14.16.27023/bin/Hostx64/x64',
    'C:/Program Files (x86)/Microsoft Visual Studio 14.0/vc/bin',
]

🎈 运行项目

  • 只需要运行 python main.py 即可生成对应的模型的图片
  • main.py 文件中,我们可以看到,可以自定义模型和一次性生成图片的数量
  • 模型需要自行下载,下载完成后放到 networks 目录下,想生成哪种类型的图片,只要更换模型路径即可
  • 现在支持网红脸、明星脸、萌娃脸、超模脸、黄种人脸等模型,下载链接在 networks 文件夹下文本文档中
def main():
    os.makedirs('results/', exist_ok=True)
    os.makedirs('results/generate_codes/', exist_ok=True)

    network_pkl = 'networks/generator_star-stylegan2-config-f.pkl'  # 模型位置
    generate_num = 20  # 生成数量

    generate_images(network_pkl, generate_num)

🎈 运行结果

  • 生成的对应结果存放在 results 文件夹下

AI人脸生成结果


极客飞兔
1.2k 声望649 粉丝