본문 바로가기

Data-science/deep learning

MBTI 성격유형별 관상 stylegan2 morphing image

728x90

Trainer를 뜯어보기 전까지는 Gs가 GeneratorSynthesis의 약자인 줄 알았다...

학습중인 Generator들의 moving average라고 표현하는 게 맞다. Gs가 명시되지 않으면, 가중치들의 moving average로 만들어진다고 한다.

3 이미지 합성 결과.

1024 x 1024 pretrained model이라 입력도 같은 size로 처리해야 함.

위 이미지를 뽑는데 시간이 5분 가량 소요. 256 x 256 pretrained 모델을 찾고 해당 model에서 위를 진행해봐야 함

256 x 256 pretrained 모델 발견, 사용할 수 있게 변경 완료.

256 이미지로 변경후, 3분 가량 소요.

label (MBTI 정보) 주입 후, 해당 label 이미지 끼리 합성후 저장하는 projection code 작성 필요. 

(현재는 전체 MBTI 정보가 섞이는 중)

왠지 얼굴이 많이 외계인처럼 나왔다.

개별 이미지들이 완전히 projection 되지 않아서다.

데이터는 소량밖에 없는 상황, freeze D와 augmentation 기법인 ada를 써서 추가학습이 필요하다. 이는 개별 이미지들 mapping을 제대로 시켜줄 것이다. 그 후 latent 단에서 합성을 하여 자연스러운 이미지를 구축할 계획