본문 바로가기

딥러닝

[GAN] 인공지능으로 화장시키기

GAN을 이용하여 화장하는 인공지능 모델을 만들어 화장을 시켜보자

출처 영상 

: https://www.youtube.com/watch?v=Uhn3dmW1uuM&list=PLRJOE4yoSM0SaeirRMcInTprh7i1ogXuR&index=5  

GAN을 이용한 결과

 

원리

Generator : 생성모델 (얼굴 —> 화장 )

Discriminator : 분류 모델 ( 진짜인지 가짜인지)

[ 3가지 오차 함수]

cycle consistency loss : 내 얼굴 → 화장 —> 내얼굴로 다시 generate 했을 때 내얼굴이 입력과 결과가 같은지 (오차 최소화)

perceptual loss: 내 얼굴이 내 얼굴로 나올 수 있도록 하여 메이크업 시키도록 하는 (오차 최소화)

makeup loss: 얼굴에서 주요 부위 (입술, 눈) 등을 잘라내어 메이크업 한 눈과 안한 눈의 오차

 

 

전체 코드 : 

https://colab.research.google.com/drive/1yqh7x9bwHu486CJAaS1cB3ZSIjD3KkvH?usp=sharing 

 

화장하는_gan.ipynb의 사본

Colaboratory notebook

colab.research.google.com

 

코드 결과: