kampta / face-seg

Semantic segmentation for hair, face and background

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