Nueral Style Transfer

Pytorch implementation of Nueral style transfer algorithm , it is used to apply artistic styles to content images . Content is the layout or the sketch and Style being the painting or the colors.

Examples

Getting Started

Prerequisite

• Python 3+ (tested on python 3.9)
• pytorch
• Nvidia Gpu(not necessary but it will significantly boost your training speed)

Installation

1. Clone the repo

   git clone https://github.com/abhinav-TB/Neural-Style-Transfer.git

2. Install python packages

   pip install -r requirements.txt

Training

Command line Arguments

Parameter	Type	Description
-s	string	Required. path to style image
-c	string	Required. path to content image
-e	string	Required. Number of training loops
-o	string	Output folder path

start training

python main.py

Example

python main.py -s ./style/style5.jpg -c ./content/trees.jpg -e 2500 

Contributing

Contributions are what makes the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.

1. Fork the Project
2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
3. Commit your Changes (git commit -m 'Add some AmazingFeature')
4. Push to the Branch (git push origin feature/AmazingFeature)
5. Open a Pull Request

License

Distributed under the MIT License. See LICENSE for more information.

Contact

@abhiGamez on Twitter

Acknowledgements

• A Neural Algorithm of Artistic Styles
• Neural Style Transfer: A Review