Skip to main content

This is the python library to convert style of content image to given style image

Project description

neuralstyletransfer

This is the python library which takes two image content image and style image and provide the output image looks like a content image but painted in the style of provided reference style image

Usage

pip install neural-style-transfer

Import NeuralStyleTransfer class from Library

from neuralstyletransfer.style_transfer import NeuralStyleTransfer

create class object

nst = NeuralStyleTransfer()

Load the content image and style image from local path

nst.LoadContentImage(content_img_path)
nst.LoadStyleImage(style_img_path)

You can also load image from external URL by providing pathType='url' by default it is setted to 'local'

nst.LoadContentImage(content_url, pathType='url')
nst.LoadStyleImage(style_url, pathType='url')

Use default parameters to train the model

Default parameters are given below we can change it based on requirement by providing values on calling .apply method

  • contentWeight=1e3
  • styleWeight=1e-2
  • epochs=500
output = nst.apply()
output.save('output.jpg')

Content image

content_image

Style image

style_image

Output image

output_image

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

neural-style-transfer-1.0.6.tar.gz (9.0 kB view hashes)

Uploaded Source

Built Distribution

neural_style_transfer-1.0.6-py3-none-any.whl (9.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page