face parsing.PyTorch

face-parsing.pytorch

內容

• 訓練
• 演示
• 參考

訓練

1. 準備培訓數據： - 下載celebamask-HQ數據集

   - 更改prepropess_data.py中的文件路徑並運行

python prepropess_data.py

2. 使用Celebamask-HQ數據集訓練模型：只需運行火車腳本：

    $ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py

如果您不想訓練該模型，則可以下載我們的預培訓模型並將其保存在res/cp中。

演示

1. 使用以下方式評估訓練的模型：

 # evaluate using GPU
python test.py

使用解析圖

face-makeup.pytorch

	頭髮	唇
原始輸入
顏色

參考

• Bisenet