본문 바로가기

Data-science/deep learning

[pytorch] 유명인 합성 stylegan2 freezeD + freezeG ( freezing layer 시행 착오)

728x90

Training From FFHQ

  no freezeG freezeG (끝 단 3 블럭만 학습)  freezeG (끝 단 2블럭만 학습)  freezeG (끝 단 1블럭만 학습) 
no freezeD x      
freezeD (끝 단 3개만 학습)  경과 후 이미지 일그러지는
현상
시도중 (gpu4) 시도중 (Gpu 1)
생성 이미지가 본질적으로 ffhq에 가까워, custom 이미지 생성이 어려워 보임 

(10k step까지 학습)
생성 이미지가 본질적으로 ffhq에 가까워, custom 이미지 생성이 어려워 보임
(10k step까지 학습)
freezeD (끝 단 2개만 학습) ?      

 

왼쪽부터 순서대로 0k, 4k, 7k step 학습 결과
왼쪽부터 순서대로  10k, 17k, 23k step 학습 결과