给哈登来一个『去胡子』AI 特效!附源码

共 2149字,需浏览 5分钟

 ·

2021-03-18 17:19

点击下面卡片关注AI算法与图像处理”,选择加"星标"或“置顶”

重磅干货,第一时间送达


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

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

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

AI如此多娇,引得无数猛男竞折腰(视频可播放):

今天分享 去胡子特效的算法原理,以及如何玩转这个算法!

No beard

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

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

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

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

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

思想就是,在隐空间(latent space)控制隐藏特征(latent code)来控制图片的属性。

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

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

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

latent code组成的空间就是latent space

StyleFlow就做了这么一个事,解耦特征控制特征

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

算法测试

Github 项目地址:

https://github.com/RameenAbdal/StyleFlow

第一步:搭建测试环境。

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

https://github.com/RameenAbdal/StyleFlow/blob/master/requirements.txt

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

项目地址:

https://github.com/NVlabs/stylegan2

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

https://pan.baidu.com/s/14F1Gww3b8S0Nabn4QTNjNg



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

https://drive.google.com/drive/folders/1QHc-yF5C3DChRwSdZKcx1w6K8JvSxQi7

第三步:在工程目录,运行程序。

python main.py

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

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

话说这李逵没了胡子,也蛮可爱啊!


哈登剃了胡子是这样子吗?



哈哈,恕我直言白了一圈


这么好玩的算法,不来个在看吗?求在看,求转发啦~

      
个人微信(如果没有备注不拉群!
请注明:地区+学校/企业+研究方向+昵称



下载1:何恺明顶会分享


AI算法与图像处理」公众号后台回复:何恺明,即可下载。总共有6份PDF,涉及 ResNet、Mask RCNN等经典工作的总结分析


下载2:终身受益的编程指南:Google编程风格指南


AI算法与图像处理」公众号后台回复:c++,即可下载。历经十年考验,最权威的编程规范!


    
下载3 CVPR2021

AI算法与图像处公众号后台回复:CVPR即可下载1467篇CVPR 2020论文 和 CVPR 2021 最新论文

点亮 ,告诉大家你也在看


浏览 149
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报