본문 바로가기

AI/Deep Learning

Style Transfer 알고리즘 | Image Style Transfer Using Convolutional Neural Networks (CVPR 2016)

728x90

들어가며


포스팅된 글은 아래에 첨부된 논문과 유튜브 영상을 바탕으로 공부를 한 내용입니다.

이외에도 구글링을 통해 다양한 글들을 참고하며 공부한 내용을 포스팅 한 것 입니다.

더보기

논문 제목: Image Style Transfer Using Convolutional Neural Networks (CVPR 2016)

논문 링크: https://rn-unison.github.io/articulos/style_transfer.pdf

유튜브 링크: https://www.youtube.com/watch?v=va3e2c4uKJk 

 

 

 

Style Transfer 알고리즘


출처: https://www.youtube.com/watch?v=va3e2c4uKJk

1. 스타일 손실 (Style Loss)

- 스타일 이미지 a 로 Gram matrix A 생성

- 노이즈 이미지 x 로 Gram matrix G 생성

- A와 G의 차를 감소시키며, 스타일 손실 (Style Loss) 생성

 

2. 콘텐츠 손실 (Content Loss)

- 콘텐츠 이미지 p 로 feature map P 생성

- 노이즈 이미지 x 로 feature map F 생성

- P와 F의 차를 감소시키며, 콘텐츠 손실 (Content Loss) 생성

 

3. Total Loss

- 스타일 손실 (Style Loss)콘텐츠 손실 (Content Loss) 에 적절한 가중치를 각각 적용시켜서 Total Loss 생성

 

 

 

마치며


이제 클론 코딩하러 가야지