styleGAN 학습을 위한 여정

https://github.com/lucidrains/stylegan2-pytorch

GitHub - lucidrains/stylegan2-pytorch: Simplest working implementation of Stylegan2, state of the art generative adversarial net

위 깃허브 코드를 따라해봤는데 전혀 학습이 안된다..

 

일단 예상되는 유력한 이유로는

1. 위처럼 평면적인 이미지들이라 이미지에 조금만 굴곡이 생기려 하면 손실때문에 다시 평면화가 진행되어서

2. 픽셀이 빽빽한 이미지에서는 augmentation이 노이즈를 추가하는 정도인데 훈련 이미지들이 단색에 단순한 이미지들이라 너무 파괴적으로 작동한 것이 아닐까

 

다른 예상으로는 

실험적인 기능을 너무 많이 켰다던가, 배치 사이즈가 너무 작았고 피라미터들의 설정이 미스였을 수도 있겠다는 생각이 든다.

 

어쨋든 대실패.

 

 

 

 

+

몇번의 실험 후에 augmentation이 문제인걸 알았다.

내가 특정 구도만 나오게 하려고 수작업으로 전처리를 했는데 augmentation에서 crop, rotate 등의 기능이 구도를 망가뜨려 이미지들의 일관성이 없어진 것이었다. 그래서 augmentation으로 색깔만 바꾸도록 했다.

훨씬 적은 훈련 횟수로도 위와 다르게 덩어리들이 잡히는 게 보인다.