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