face parsing.PyTorch

pare.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

การแต่งหน้าใบหน้าโดยใช้แผนที่การแยกวิเคราะห์

face-makeup.pytorch

	ผม	ริมฝีปาก
อินพุตดั้งเดิม
สี

การอ้างอิง

• Bisenet