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
