face parsing.PyTorch

face-parsing.pytorch

Inhalt

• Ausbildung
• Demo
• Referenzen

Ausbildung

1. Erstellen Sie Trainingsdaten:-Download Celebamask-HQ-Datensatz

   - Ändern Sie den Dateipfad in der prepropess_data.py und führen

python prepropess_data.py

2. Trainieren Sie das Modell mit Celebamask-HQ-Datensatz: Führen Sie einfach das Zugskript aus:

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

Wenn Sie das Modell nicht trainieren möchten, können Sie unser vorgebildetes Modell herunterladen und es in res/cp speichern.

Demo

1. Bewerten Sie das ausgebildete Modell mit:

 # evaluate using GPU
python test.py

Gesichts Make -up mit Parsing -Karten

face-makeUp.pytorch

	Haar	Lippe
Originaleingabe
Farbe

Referenzen

• Bisenet