styleGAN 학습을 위한 여정 6

5까지를 마지막으로 하고 후기까지 썼는데 그냥 완성해보고 싶어서

5탄에서 학습하던 모델 마저 돌려봄

계속 돌리니까 모양이 잡혔다.

그리고 여러가지 문제들을 수정했다.

 

 

1. RGBA 이미지

RGBA 이미지를 RGB 이미지로 변환 하면 투명한 부분이 이상한 패턴무늬로 바뀌는 경우가 있다.

대부분의 이미지들의 바탕이 흰색이기 때문에 다음 코드로 RGBA 이미지의 투명부분을 흰색으로 처리한 후 RGB이미지로 저장했다.

img = Image.open('이미지 경로')
if img.mode == 'RGBA':
	img.load()
    background = Image.new("RGB", img.size, (255, 255, 255))
    background.paste(img, mask=img.split()[3]) # 3 is the alpha channel
    background.save('이미지 경로')

 (img.load()의 열이 안맞는 건 그냥 버그인듯... 아무리 해도 안고쳐짐)

 

2. 얼굴 형태가 없는 이미지 제거

 

 

3. 초록 머리색의 이미지가 적어서 제대로 얼굴형이 잡히기 않는 것으로 보여서 초록머리 이미지를 보강했다.

 

 

4. FID score가 어느 순간부터 개선이 안되길래 lr_mlp = 0.01 로 변경

 

 

 

5. batch size 키움. 기존에 작은 배치 사이즈로 학습했었음

+근데 배치는 4가 제일 좋은듯 하다. 다른 사람들도 대부분 4 쓰는듯