face parsing.PyTorch

face-parsing.pytorch

Isi

• Pelatihan
• Demo
• Referensi

Pelatihan

1. Siapkan Data Pelatihan:-Unduh Dataset Celebamask-HQ

   - Ubah jalur file di prepropess_data.py dan jalankan

python prepropess_data.py

2. Latih model menggunakan dataset celebamask-hq: Cukup jalankan skrip kereta:

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

Jika Anda tidak ingin melatih model, Anda dapat mengunduh model terlatih kami dan menyimpannya di res/cp .

Demo

1. Evaluasi model terlatih menggunakan:

 # evaluate using GPU
python test.py

Wajah wajah menggunakan peta parsing

face-makeup.pytorch

	Rambut	Bibir
Input asli
Warna

Referensi

• Bisenet