Face Alignment

[기대되는 실행 내용]

• psp 인코더에 입력하기에 좋은 형태로 얼굴을 align

• 변환된 얼굴을 기존 얼굴 각도로 synthesis

align 하는 과정에서 나오는 각도 정보를 저장하여 합성 시에 적용

 

[참고 코드]

stylegan2/align_images.py at master · justinpinkney/stylegan2

toonify/toonify-yourself.ipynb at master · justinpinkney/toonify

 

[colab demo]

%tensorflow_version 1.x

git clone https://github.com/justinpinkney/stylegan2
%cd stylegan2
mkdir raw
mkdir aligned

# !wget https://images.complex.com/complex/image/upload/c_limit,w_680/f_auto,fl_progressive,pg_1,q_auto/bxj0w4zedsrdolbrkrwq.jpg -O raw/saltbae.jpg
python align_images.py raw aligned

 

[workstation]

• 전용 컨테이너 생성 (tensorflow 1 버전)

docker pull nvcr.io/nvidia/tensorflow:21.02-tf1-py3
NV_GPU=0 docker run --runtime=nvidia --name bernice-tf1 --shm-size 4G -it -v $(pwd):/workspace -v $(readlink -f disk1):/disk1 nvcr.io/nvidia/tensorflow:21.02-tf1-py3 /bin/bash

• 라이브러리 설치

pip install cmake
pip install opencv-contrib-python dlib

• 실행

python align_images.py raw aligned