Python+AI智能编辑人脸

哈喽，大家好。

今天继续给大家分享优秀的 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是将潜码解耦后得到的。由于潜码的特征之间的耦合性比较大。例如：头发长度和男子气概，按照潜码的特征分布，这两个特征之间有紧密的联系。实际上，短发男子、长发男子都可以有很强的男子气概，因此需要将特征解耦。

项目