face parsing.PyTorch

FACE-PARSING.PYTORCH

Conteúdo

• Treinamento
• Demonstração
• Referências

Treinamento

1. Prepare dados de treinamento:-Faça o download do conjunto de dados Celebamask-HQ

   - Alterar o caminho do arquivo no prepropess_data.py e executar

python prepropess_data.py

2. Treine o modelo usando o conjunto de dados Celebamask-HQ: Basta executar o script do trem:

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

Se você não deseja treinar o modelo, pode baixar nosso modelo pré-treinado e salvá-lo em res/cp .

Demonstração

1. Avalie o modelo treinado usando:

 # evaluate using GPU
python test.py

Maquiagem de rosto usando mapas de análise

FACE-MAKEUP.PYTORCH

	Cabelo	Lábio
Entrada original
Cor

Referências

• Bisenet