main.py를 통해 염색 이미지를 생성할 수 있는 방법은 2가지 입니다.
아래 방법을 통해 이미지를 생성하기 위해서는 Face parsing, SEAN의 pretrained 모델을 다운 받아, pretrained_network 폴더에 담아주세요.
pip install -r requirements.txt
위 명령어를 통해 필요한 패키지들을 설치할 수 있습니다.
: 머리 색을 가져오고싶은 reference 이미지가 있는 경우
1) data/ref, data/src 폴더 생성
2) data/ref 폴더에는 reference 이미지를, data/src 폴더에는 source 이미지를 저장
3) python main.py --mode custom
: pkl 파일(Python 3.8 이상에서만 가능)에 미리 저장된 색을 이용해 염색하고자 하는 경우 (지원 색상 - black, brown, blond, red, blue)
1) data/src 폴더 생성
2) data/src 폴더에 source 이미지 저장
3) python main.py --mode {color_name}
ex. python main.py --mode black
+pkl 파일 색상 코드 변경 또는 새로운 색상 추가하기
1) data/ref 폴더 생성
2) data/ref 폴더에 reference 이미지 저장
3) python main.py --mode save_color --color_name {color_name}
ex. python main.py --mode save_color --color_name pink