데이터 사이언스/AI

Style Transfer

B1001101 2021. 2. 9. 16:31

논문

 

Papers with Code - A Neural Algorithm of Artistic Style

Implemented in 248 code libraries.

paperswithcode.com


설치 및 사용방법 (Mac 기준)

1) neural-style

$ git clone https://github.com/jcjohnson/neural-style 

2) torch7 & loadcaffe

$ git clone https://github.com/torch/distro.git ~/torch --recursive
$ cd ~/torch; bash install-deps;
$ ./install.sh
  • cmake: command not found 에러 뜰 때: cmake 설치
$ brew install cmake
$ cd
$ vim .bash_profile (vim 말고 다른 텍스트 에디터 사용해도 됨)
  -> 맨 아래에 다음 코드 추가 (<you> 자리에는 사용자 이름 넣음)
  PATH=$PATH\:/Users/<you>/torch/install/bin ; export PATH
$ source .bash_profile
$ echo $PATH (저장되었는지 확인)
$ brew install protobuf
$ luarocks install loadcaffe
  • libjpeg, image 설치
$ brew install 
$ luarocks install image
 

Torch / Lua after installation is not working

I have followed the following approach in order to install Torch in my machine (Mac). http://torch.ch/docs/getting-started.html#_ When I am done with the installation, I type: $ luarocks install...

stackoverflow.com

3) VGG 모델

$ sh models/download_models.sh

4) 사용방법

  • 기본 사용방법
$ th neural_style.lua -style_image <스타일이미지.jpg> -content_image <변환할이미지.jpg>
  • 옵션
    • -output_image: 저장할 이미지 이름
    • -gpu: 사용할 GPU 아이디 지정, CPU 사용하려면 -1로 설정
    • 기타 옵션: github.com/jcjohnson/neural-style 참고
 

jcjohnson/neural-style

Torch implementation of neural style algorithm. Contribute to jcjohnson/neural-style development by creating an account on GitHub.

github.com