Context Encoder
이 중간 결과는 Deepfake 의 faceswap 이 아닌, GAN 을 이용하는 context encoder 를 pytorch 로 돌려보는 과정이다. (https://arxiv.org/abs/1604.07379)
데이터는 유명한사람들 얼굴 모아놓은 셋이고, 윗줄은 사진 가운데를 마스크로 가린 입력이고, 두번째 줄이 학습된 u-net 으로 합성한 이미지이고, 아랫줄이 원본사진이다. 총24825장의 얼굴 이미지를 랜덤 마스크로 가린채, 원본 사진과 GAN 방식으로 학습을 시킨다.
첨부한 이미지는 173 epoch 후의 결과이다. deepfake 의 방식보다 코딩이 간단하다. 하지만, DGX1V 에서 2일이 걸린 학습의 결과라는 것이 문제이다.
지금까지 감상........ u-net 은 훌륭하다, 하지만 무척 느리다.
교수님 지난번 A100 으로 돌리신 deepfake 영상도 이곳에 올리시면 어떠실런지요!!
답글삭제