์ ๋ช ํ๊ฐ์ ์ํ์ ์ ํํ์ฌ, ํด๋น ์ํ์ ํํ์ผ๋ก ์ด๋ฏธ์ง๋ฅผ ๋ณํ์์ผ์ฃผ๋ ํ๋ก๊ทธ๋จ์ ๋๋ค.
tensorflow ๋ฐ tensorflowhub - magenta/arbitrary-image-stylization-v1-256 ๋ชจ๋ธ์ ํ์ฉํ์ฌ ์ด๋ฏธ์ง๋ฅผ ์ ๋ช ํ๊ฐ์ ์ํ์ ํํ์ผ๋ก ๋ณํ์์ผ์ค๋ค.
ํ๋ก์ ํธ ๊ฒฝ๋ก๋ก ์ด๋ํ์ฌ ๋ค์ ๋ช ๋ น์ด๋ฅผ ์ ๋ ฅํ๋ค.
python style_transfer.py --content_img="./image/content/sample.png" --style_img="./image/portrait/sample_1.jpg"
- --content_img=[์ปจํ ์ธ ์ด๋ฏธ์ง ๊ฒฝ๋ก]
- --style_img=[์คํ์ผ ์ด๋ฏธ์ง ๊ฒฝ๋ก]
- --output_img=[๊ฒฐ๊ณผ ์ด๋ฏธ์ง ๊ฒฝ๋ก]
- --resize_width=[๊ฒฐ๊ณผ ์ด๋ฏธ์ง ํฌ๊ธฐ]
- --model=[๋ชจ๋ธ ๋ฒ์ ]
๋ค์ ํ๊ฒฝ ๋ฐ ๋ชจ๋์ด ์ค์น๋์ด์ผ ํ๋ก๊ทธ๋จ์ ์คํ ํ ์ ์์ต๋๋ค.
- Python 3.8 ์ด์
- TensorFlow 2.2 ์ด์
- TensorFlow_Hub
- numpy
- pillow
- ๋ชจ๋ v1 : magenta/arbitrary-image-stylization-v1-256/1
- ๋ชจ๋ v2 : magenta/arbitrary-image-stylization-v1-256/2