发型识别大总结 您所在的位置:网站首页 ai头发脸型识别 发型识别大总结

发型识别大总结

2024-06-09 16:15| 来源: 网络整理| 查看: 265

Hair detection, segmentation, and hairstyle classification in the wild

头发检测:

基于patch滑动窗口+CaffeNet-fc7 +Random Forest

头发分割:

LTP特征(Linear Ternary Pattern )+SVM

只对不确定的位置的像素进行分类,比如下图的蓝色部分。

发型识别:

LTP特征(Linear Ternary Pattern )+Random Forest

整体流程:

 

FIGARO, HAIR DETECTION AND SEGMENTATION IN THE WILD

头发检测:

将图片分patch,然后基于每个patch提取HOG+LTP特征进行分类

头发分割:

基于Felzenszwalb and Huttenlocher 这样的基于图的算法,对头发检测后的mask进行分割。

数据集:

Figaro-1k

http://projects.i-ctm.eu/en/project/figaro-1k

数据在谷歌图片和Flickr收集。包含7种发型类别(straight, wavy, curly, kinky, braids, dreadlocks, short ),每一个类别包含150张图,一共1050张图,包含840种视角。数据集提供了发型分割的mask。

 

Human Hair Segmentation In The Wild Using Deep Shape Prior

Github:https://yozey.github.io/Hair-Segmentation-in-the-wild

bilibili:https://www.bilibili.com/video/av66067281

网络结构:

网络整体基于resnet18主干网络的deeplabv3,然后再加一个Unet结构。deeplabv3负责头发形状(shape prior )的学习,生成的特征图和原图concat后输入UNet网络,输出最终分割结果。

 

实验结果:

 

其他资源:

Github:

https://github.com/YBIGTA/pytorch-hair-segmentation

可能安装错误:

ImportError: No module named 'ignite.engine'

解决方法:

pip install pytorch-ignite

实际运行:

python3 demo.py



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有