Skip to content
Style Transfer Application using PyTorch
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.vscode
loaders
models
results
.gitignore
LICENSE
Profile-Ops.ipynb
README.md
main.py
net.py
utils.py
videocapture_async.py

README.md

StyleTransfer-PyTorch

This is the implementation of Style Transfer application using PyTorch.

The app fetches image from your webcam and perform style transfer @ (average) 21.3 FPS.

Requirements

  • Windows/Linux/MacOS
  • Anaconda with PyTorch and other necessary modules

Usage

  • For training: python main.py --phase train
  • For inference: python main.py --phase test

Tutorial

If you are interested in learning more about this project, a step-by-step tutorial is available at - Link

Results

These are the results of the app when tested on a machine with NVIDIA RTX 2060.

Result 1

Author

Sri Raghu Malireddi / @r4ghu

You can’t perform that action at this time.