face parsing.PyTorch

face-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

구문 분석지도를 사용한 얼굴 메이크업

face-makeup.pytorch

	머리카락	말뿐인
원래 입력
색상

참조

• 비 세넷