Skip to content

pattrickx/StyleTransfer

Repository files navigation

StyleTransfer

This project is based on the VGG19 Network, to extract style features from an image and insert it into another image, below you can see an example, where The Starry Night painting was passed as a style image and as a target image a picture of hoover tower at night, the image on the right is the result after 1000 epochs, using the project's default hyperparameters.

⚠️ If you don't have a gpu with cuda, the style transfer execution time will be much longer

Prerequisites

Python >=3.8.10

How to Install

sudo pip3 install -r requirements.txt 

How to Use

if you are going to test with the example images, just run:

python3 main.py

but if you want to use with your own images change the path to the variables shown below in the main.py file:

style = "your style image"
img = "your base image"

and next run:

python3 main.py
Based on

About

Open source style transfer project, based on VGG19

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages