哈喽,大家好。
今天继续给大家分享优秀的 AI 项目。
今天分享这个项目使用StyleGAN
算法对人脸进行编辑。
在该项目中,可以改变图片中人物的发型、给人物加胡须、让人物微笑等等。
大致思路是,先用StyleGAN
算法提取人脸特征,再利用人脸属性分类器(如:Nvidia的40个人脸属性分类器),找到可进行人脸编辑的属性,执行编辑后即可将原图变成想要的效果。
项目已在 GitHub 上开源,可直接运行。
下面我介绍下项目核心流程。
1. 提取人脸特征
下载StypeGAN
预训练模型,放在指定目录下。
运行wy_gan_inversion.py
获取人脸w并保存。
这里解释下w的概念,StypeGAN
模型从训练图像数据中提取特征(如:头发、皱纹、眉毛),这些特征存储在每一张图片的latent code
中,一张图片有许多latent code
,一定数量的latent code
组成latent
(一个512维的向量),这个latent
又被称为潜码。
w是将潜码<