Face Segmentation

Semantic segmentation for hair, face and background

Barebones version of this repository.

Dataset

Labeled faces in the wild. Get it from here.

Training

python train.py \
    --data-folder data/lfw \
    --pre-trained weights/mobilenet_v2.pth.tar \
    --output-folder scratch/ \
    --num-epochs=50 \
    --batch-size=32

Testing

python test.py --data-folder data/samples/ --pre-trained checkpoints/model.pt

Results