/NTCT

Neural Texture Color Transfer

Primary LanguageJupyter Notebook

NTCT

Neural Texture Color Transfer

Dataset 준비

  • MS COCO

주소: https://cocodataset.org/#download 이미지만 받아오면 될듯 (*근데 크롬에서 다운받으려니깐 안되고 edge 들어가서 다운 누르니깐 안전하지 않은 다운로드라고 뜨는데 무시하고 계속하면 다운로드 됨)

  • WikiArt

주소: https://github.com/lucasdavid/wikiart 이거 git clone해서 다운받은 후 써있는거처럼 command 입력하면 자동으로 다운 받아짐

AdaIN 구현한 github

https://github.com/naoto0804/pytorch-AdaIN

구현 모델 모양

  • 모델1: Untitled (2)

  • 모델2: Untitled

  • 모델3: Untitled (1)

3번은 모델 2가지로 학습해봄 3_1: 기존 decoder 동일 3_2: 기존 decoder의 upsampling 방식을 linear에서 bilinear로, relu에서 leakyrelu로 바꿈

대략적인 실험결과

3번이 괜찮지만 color histogram보다는 좋은지는