实时发型分割+换色

 

论文：Real-time Hair Segmentation and Recoloring on Mobile GPUs

Github：https://github.com/google/mediapipe

CVPR2019

论文提出了一个手机端的实时的发型分割+换色的方案，使用手机端的gpu可以达到30-100FPS。

 

网络结构：

整体结构为hourglass 沙漏型结构。输出部分还加入了denseNet的层。

网络输入为rgb+mask共四个通道的输入，其中，mask为视频中前一帧的分割结果。

 

整体流程：

 

数据集label:

 

运行效果：