face parsing.PyTorch

الوجه 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

مكياج وجه باستخدام خرائط التحليل

الوجه makeup.pytorch

	شعر	شفة
المدخلات الأصلية
لون

مراجع

• Bisenet