pix2pix (2) - 데이터 학습 및 테스트

https://github.com/affinelayer/pix2pix-tensorflow

위의 소스를 활용하였다. 

학습환경

 - OS : Ubuntu 16.0.4

 - Python 3.6, Tensorflow 1.4.1

 - GPU : Geforce 1080Ti

기본 설정이 되어있는데로 학습시켰으며 200 epoch 학습시켰다.

논문에도 나와 있지만 별도의 parameter tweak 없이도 잘 돌아간다고 하니 건드릴게 없었다

아래는 tensorboard를 통해 확인가능한 중간결과물이다. 물론 트레이닝셋 표본 안에서 돌린것이니 잘나왔다고 생각할 수 있겠다.

다음은 실제 드로잉 이미지로 테스트해보자

아래 사진에서 오른쪽 이미지는 인터넷에서 누군가 그린 펜더 이미지이다.

그냥 팬더에 가까운 사진은 꽤 그럴사 하지 않은가?

쿵푸팬더는 살짝 아쉽다. 쿵푸팬더의 이미지도 학습데이터에 있었다면

좀 더 자연스러운 결과가나올 것으로 기대된다.

다음은 팬더가 아닌 인테리어 이미지로 테스트 한 것이다.

꽤 그럴듯한 결과를 연출해준다. 이미지 크기는 250pixel 정도의 작은 이미지인데 품질이 썩 만족스럽지는 않다

NVIDIA에서 공개한 pix2pixHD 를 활용하면 fullHD급의 자연스러운 이미지도 가능한데

이를 학습시키기 위해서는 기본 GPU 메모리 12GB 이상이 필요하다.

게임용으로 가장 비싼 geforce 1080Ti 가 11GB의 메모리인데.... 어쩌면 그들의 상술인지도 모르겠다.

고오급 GPU에서만 돌아가는 멋진 소스코드를 공개하고... 장비벽을 불러일으키는 전략?