Implementation of 2D Neural Style Transfer in a manner proposed in an original paper, along with optimization for preserving content image colors.
- Pillow : Python Imaging Library
- matplotlib : Plotting library for the Python programming language and its numerical mathematics extension NumPy
- NumPy : Fundamental package for scientific computing with Python
- OpenCV : Library of programming functions mainly aimed at real-time computer vision
- PyTorch : Open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing
Just run main.py in initialModel package.
If COLOR_HISTOGRAM_MATCHING variable is set to True, then it will do the color preservation optimization with technique: Color Histogram Matching.
If LUMINANCE_ONLY_TRANSFER variable is set to True, then it will do the color preservation optimization with technique: Luminance Only Transfer.
For given inputs:
Outputs with different ratios:
