face parsing.PyTorch

Face-Parsing.pytorch

Contenu

• Entraînement
• Démo
• Références

Entraînement

1. Préparer les données de formation: - Télécharger le jeu de données Celebamask-HQ

   - Modifier le chemin du fichier dans le prepropess_data.py et exécuter

python prepropess_data.py

2. Train le modèle à l'aide de l'ensemble de données Celebamask-HQ: il suffit d'exécuter le script de train:

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

Si vous ne souhaitez pas former le modèle, vous pouvez télécharger notre modèle pré-formé et l'enregistrer dans res/cp .

Démo

1. Évaluez le modèle qualifié en utilisant:

 # evaluate using GPU
python test.py

Maquillage face à l'aide de cartes d'analyse

face-makeup.pytorch

	Cheveux	Lèvre
Entrée d'origine
Couleur

Références

• Bisenet