StarGAN-V2 구현하기

Github : https://github.com/clovaai/stargan-v2

GitHub - clovaai/stargan-v2: StarGAN v2 - Official PyTorch Implementation (CVPR 2020)

예전에 구현을 못한 StarGAN-V2를 다시 구현해보았다.

Github에 나온대로 구현하니까 잘 되었음.

 

단, Data를 불러오는 Bash파일 실행 중 Unzip하는데 오류가 나서 해당 부분만 수정하였음.

Zip file을 다운 후 삭제하는 코드가 있길래 해당 부분을 주석처리로 바꿔서 그냥 수동으로 내가 Unzip하였다..

추가적으로 오류로 인해 수정한 코드는 

./core/checkpoint.py 의 48번 째 줄,

module.module.load_state_dict(module_dict[name]) 을 module.module.load_state_dict(module_dict[name], False) 로 수정하였음.

 

실제 주어진 Data들을 토대로 돌린 결과 아래와 같은 결과를 얻었음. (일부)

 

 

Custom이미지를 넣어서 사용할 수 있길래 여러 연예인 사진들을 넣어서 실행해 보았음.

특징들을 잘따서 나타내는 것을 눈으로 직접 확인가능하였음. (pre-trained model)

Male의 얼굴도 사용하였지만 Female 만큼 괜찮은 결과가 나오진 않았다. 아무래도 사진을 crop하거나 male쪽에선 쉽게 특징을 추출하기 어려워서 일까 하는 생각이 들었다.

어떻게 구현되었는지 paper를 통해서 Review할 계획임.

결과물이 도드라지게 나와서 신기하였음.