유명화가의 작품을 선택하여, 해당 작품의 화풍으로 이미지를 변환시켜주는 프로그램입니다.
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