Auto-Hair (segmentation)

install packages

• pip install tensorflow==1.14
• pip install torch
• pip install torchvision
• pip install opencv-python

download model

1. segmentation model

• http://download.tensorflow.org/models/deeplabv3_pascal_trainval_2018_01_04.tar.gz

2. matting model

• https://github.com/huochaitiantang/pytorch-deep-image-matting/releases/download/v1.4/stage1_sad_54.4.pth

source

1. segmentation

• https://github.com/tensorflow/models/tree/master/research/deeplab

2. matting

• https://github.com/huochaitiantang/pytorch-deep-image-matting

3. trimap

• https://github.com/lnugraha/trimap_generator