让猛男娇羞的AI算法

àì夳堔傛蜴生んèń 2022-11-12 02:34 162阅读 0赞

![让猛男娇羞的AI算法][AI]

> 本文 GitHub [https://github.com/Jack-Cherish/PythonPark][https_github.com_Jack-Cherish_PythonPark] 已收录，有技术干货文章，整理的学习资料，一线大厂面试经验分享等，欢迎 Star 和 完善。

大家好，我是 Jack。

最近又火了一个AI算法，猛男都hold不住的AI特效。

想像一下，魁梧的身躯，浓密的络腮胡的猛男。

![20210322103143803.gif][]

在AI特效的加持下，“华丽转身”瞬间变成了一个可爱的“小胖子”。

![20210322103150408.gif][]

AI如此多娇，引得无数猛男竞折腰（**视频可播放**）：

让猛男娇羞的AI算法

其实，**这个去胡子特效算法很早就有了**。

就像今年2月份突然火的「**蚂蚁呀嘿**」一样，早在去年8月份我就出过技术教程：

[特朗普和蒙娜丽莎深情合唱《Unravel》][Unravel]

这次火起来的去胡子特效，我也在今年1月份就出过类似的技术教程。

这些爆款的背后，都是**早已发表的算法的应用落地**。

我们从不缺优秀的算法，缺的是将算法巧妙用起来的idea，以及算法的快速落地能力。

**去胡子特效**就是**人脸属性编辑算法**的巧妙应用。

类似技术，不光可以去掉胡子，也可以**变秃变强**：

[秃头生成器，Hairstyle Transfer 你值得拥有！][Hairstyle Transfer]

今天就带大家**回顾一下**，**去胡子特效**的算法原理，以及如何玩转这个算法！

## No beard ##

猛男如果想要尝试这「去胡子特效」，不想跑算法，可以直接用一款名为Snapchat的App。

![让猛男娇羞的AI算法][AI 1]

在对话框里输入no beard，选中下面第一个特效即可：

![让猛男娇羞的AI算法][AI 2]

Snapchat这款App需要科学上网，否则无法使用。

接下来说的就是我曾经写过的StyleFlow，**胡须、头发、光照角度、人脸角度、年龄、眼镜、表情**等多维角度都可以单独调节（**视频可播放**）：

StyleFlow，牛逼！

StyleFlow 就是基于StyleGAN的人脸属性编辑算法。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI_size_16_color_FFFFFF_t_70][]

思想就是，在隐空间（latent space）控制隐藏特征（latent code）来控制图片的属性。

简单点解释就是，一副人脸图片，是由多维特征组成的，比如年龄、性别、光照、肤色、发质等。

为了更好的对数据进行分类或生成，需要对数据的特征进行表示，但是数据有很多特征。

这些特征之间相互关联，耦合性较高，导致模型很难弄清楚它们之间的关联，使得学习效率低下，因此需要寻找到这些表面特征之下隐藏的深层次的关系，将这些关系进行解耦，得到的隐藏特征，即latent code。

由latent code组成的空间就是latent space。

StyleFlow就做了这么一个事，**解耦特征，控制特征**。

算法在人脸和汽车数据集上，都取得了非常不错的效果。

## 算法测试 ##

Github 项目地址：

[https://github.com/RameenAbdal/StyleFlow][https_github.com_RameenAbdal_StyleFlow]

第一步：搭建测试环境。

需要安装PyQt5、Tensorflow等第三方库，根据requirements.txt安装即可。

[https://github.com/RameenAbdal/StyleFlow/blob/master/requirements.txt][https_github.com_RameenAbdal_StyleFlow_blob_master_requirements.txt]

此外，还需要配置StyleGAN2的环境。

项目地址：

[https://github.com/NVlabs/stylegan2][https_github.com_NVlabs_stylegan2]

为了方便大家下载，我从Google Drive下载了控制人脸的属性的权重文件，放到了百度网盘（提取码：jack）：

[https://pan.baidu.com/s/1-kLG2Fzx0MM88SSYqefMNg][https_pan.baidu.com_s_1-kLG2Fzx0MM88SSYqefMNg]

更多的控制其他物体属性的权重文件，可在从Google Drive下载。

[https://drive.google.com/drive/folders/1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7][https_drive.google.com_drive_folders_1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7]

第三步：在工程目录，运行程序。

python main.py

这个去胡子特效无非就是将，控制胡子的latent code单独拿出来，做了一个端到端的效果。

为了保证效果的稳定，还会做一些工程上的优化，例如添加**下巴区域**的mask，只替换该区域。

话说这李逵没了胡子，也蛮可爱啊！

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI_size_16_color_FFFFFF_t_70 1][]

我是超多干货的 Jack，我们下期见。

> 文章转自我的公众号：
> 
> [https://mp.weixin.qq.com/s/UJltI0jOTyN0Ofn\_Uy1J\_g][https_mp.weixin.qq.com_s_UJltI0jOTyN0Ofn_Uy1J_g]

[AI]: /images/20221022/38d2810722fb47a483818a576589597a.png
[https_github.com_Jack-Cherish_PythonPark]: https://github.com/Jack-Cherish/PythonPark
[20210322103143803.gif]: /images/20221022/49f20a0851e44f589737f7ec11f77bf2.png
[20210322103150408.gif]: /images/20221022/802d10bdee814a19a2e09b6baf2063d7.png
[Unravel]: https://mp.weixin.qq.com/s?__biz=MzIxODg1OTk1MA==&mid=2247485548&idx=1&sn=4f2834743dd6a44d7e4968467ef20f73&scene=21#wechat_redirect
[Hairstyle Transfer]: https://mp.weixin.qq.com/s?__biz=MzIxODg1OTk1MA==&mid=2247485384&idx=1&sn=288859b5369ae45688eb59eb9a455aa2&scene=21#wechat_redirect
[AI 1]: /images/20221022/734977454e4942ef8dd31441f6d883ee.png
[AI 2]: /images/20221022/bd9017a25aed4d4cb5dbd339ee7a06ff.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20210322103339888.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI=,size_16,color_FFFFFF,t_70
[https_github.com_RameenAbdal_StyleFlow]: https://github.com/RameenAbdal/StyleFlow
[https_github.com_RameenAbdal_StyleFlow_blob_master_requirements.txt]: https://github.com/RameenAbdal/StyleFlow/blob/master/requirements.txt
[https_github.com_NVlabs_stylegan2]: https://github.com/NVlabs/stylegan2
[https_pan.baidu.com_s_1-kLG2Fzx0MM88SSYqefMNg]: https://pan.baidu.com/s/1-kLG2Fzx0MM88SSYqefMNg
[https_drive.google.com_drive_folders_1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7]: https://drive.google.com/drive/folders/1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20210322103426396.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2M0MDY0OTU3NjI=,size_16,color_FFFFFF,t_70
[https_mp.weixin.qq.com_s_UJltI0jOTyN0Ofn_Uy1J_g]: https://mp.weixin.qq.com/s/UJltI0jOTyN0Ofn_Uy1J_g