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.
- Anaconda with PyTorch and other necessary modules
- For training:
python main.py --phase train
- For inference:
python main.py --phase test
If you are interested in learning more about this project, a step-by-step tutorial is available at - Link
These are the results of the app when tested on a machine with NVIDIA RTX 2060.
Sri Raghu Malireddi / @r4ghu