当前位置:竞技宝官网>女性时尚>发型>正文

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

2019-02-18 来源:未知 责任编辑:竞技宝官网 点击:

分享到:

  如今,在类似天天P图、美图秀秀等手机APP中,给指定照片或视频中的人物更换头发颜色已经是再正常不过的事情了。那么本文便介绍了该功能背后如AI头发分割模块、头发换色、颜色增强与修正模块等技术原理(附代码)。

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  基于深度学习的目标分割算法已经比较成熟,比较常用的有FCN,SegNet,UNet,PspNet,DenseNet等等。这里我们使用Unet网络来进行头发分割,具体可以参考如下链接:点击打开链接Unet头发分割代码如下:

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  为什么要对头发的颜色进行增强与修正? 先看下面一组图,我们直接使用HSV颜色空间对纯黑色的头发进行染色,目标色是紫色,结果如下:

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  大家可以看到,针对上面这张原图,头发比较黑,在HSV颜色空间进行头发换色之后,效果图中很不明显,只有轻微的颜色变化。

  为什么会出现这种情况?原因如下: 我们以RGB和HSV颜色空间为例,首先来看下HSV和RGB之间的转换公式:

  设 (r, g, b)分别是一个颜色的红、绿和蓝坐标,它们的值是在0到1之间的实数。设max等价于r, g和b中的最大者。设min等于这些值中的最小者。要找到在HSL空间中的 (h, s, l)值,这里的h ∈ [0, 360)度是角度的色相角,而s, l ∈ [0,1]是饱和度和亮度,计算为:

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  我们假设头发为纯黑色,R=G=B=0,那么按照HSV计算公式可以得到H = S = V = 0;

  那么,我们先将红色转换为对应的hsv,然后保留原始黑色头发的V,红色头发的hs,重新组合新的hsV,在转换为RGB颜色空间,即为头发换色之后的效果(hs是颜色属性,v是明度属性,保留原始黑色头发的明度,替换颜色属性以达到换色目的);

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  对于黑色,我们计算的结果是H=S=V=0,由于V=0,因此,p=q=t=0,不管目标颜色的hs值是多少,rgb始终都是0,也就是黑色;

  这样,虽然我们使用了红色,来替换黑色头发,但是,结果却依旧是黑色,结论也就是hsv/hsl颜色空间,无法对黑色换色。

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  与之前HSV颜色空间的结果对比,我们明显可以看到,天天P图和美妆相机的效果要更浓,更好看,而且对近乎黑色的头发进行了完美的换色;

  由于上述原因,我们这里需要对图像中的头发区域进行一定的增强处理:提亮,轻微改变色调;

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  这一步比较简单,保留明度分量不变,将其他颜色、色调分量替换为目标发色就可以了。

  假如我们要将头发染发为一半青色,一般粉红色,那么我们构建如下图所示的颜色MAP:

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  对于头发区域的每一个像素点P,我们将P的RGB转换为HSV颜色空间,得到H/S/V;

  根据P在原图头发区域的位置比例关系,我们在颜色MAP中找到对应位置的像素点D,将D的RGB转换为HSV颜色空间,得到目标颜色的h/s/v;

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  这一步主要是为了突出头发丝的细节,竞技宝官网可以使用锐化算法,如Laplace锐化,USM锐化等等。上述过程基本是模拟美妆相机染发算法的过程,给大家参考一下,最后给出本文算法的一些效果举例:

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码

  本文效果除了实现正常的单色染发,混合色染发之外,还实现了挑染,如最下方一组效果图所示。

  计算头发纹理,根据头发纹理选取需要挑染的头发束,然后对这些头发束与其他头发分开染色即可,具体逻辑这里不再累赘,大家自行研究,这里给出解决思路供大家参考。

  最后,本文算法理论上实时处理是没有问题的,头发分割已经可以实时处理,所以后面基本没有什么耗时操作,使用opengl实现实时染发是没有问题的。

  新智元将于9月20日在北京国家会议中心举办AI WORLD 2018 大会,邀请机器学习教父、CMU教授 Tom Mitchell,迈克思·泰格马克,周志华,陶大程,陈怡然等AI领袖一起关注机器智能与人类命运。

  即日起到8月19日,新智元限量发售若干早鸟票,与全球AI领袖近距离交流,见证全球人工智能产业跨越发展。

竞技宝官网:【AI超级美发师】深度学习算法打造染发特效(附代码