Feature/bug fix - Githubissues

Taeu / Style-Your-Hair

Official Pytorch implementation of "Style Your Hair: Latent Optimization for Pose-Invariant Hairstyle Transfer via Local-Style-Aware Hair Alignment (ECCV 2022)"

295 stars 40 forks source link

Feature/bug fix #2

Closed myoons closed 2 years ago

myoons commented 2 years ago

is_reconed 가 True 일 경우 source 와 target 이미지의 npy (W+) , npz (FS) 가 존재하는지 확인
flip_check 가 False 인 경우 is_flip 변수가 할당이 안되어 에러가 나던 버그 해결
FaceAlignment 모듈에 args.device 인자로 추가
CPU 로만 Inference 하던 경우 생기는 버그 해결 (GPU 사용이 불가능한 환경에서 테스트 완료)