zllrunning / face-parsing.PyTorch

Using modified BiSeNet for face parsing in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

face-parsing.PyTorch

Contents

Training

  1. Prepare training data: -- download CelebAMask-HQ dataset

    -- change file path in the prepropess_data.py and run

python prepropess_data.py
  1. Train the model using CelebAMask-HQ dataset: Just run the train script:
    $ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py

If you do not wish to train the model, you can download our pre-trained model and save it in res/cp.

Demo

  1. Evaluate the trained model using:
# evaluate using GPU
python test.py

Face makeup using parsing maps

face-makeup.PyTorch

  Hair Lip
Original Input Original Input Original Input
Color Color Color

References

ezoic increase your site revenue

About

Using modified BiSeNet for face parsing in PyTorch

License:MIT License


Languages

Language:Python 66.6%Language:Cuda 20.8%Language:C++ 11.5%Language:C 1.1%