Spaces:
Sleeping
Sleeping
File size: 954 Bytes
bd7e239 4e4e1e5 bd7e239 4e4e1e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
---
title: StyleTransfer
emoji: 🐢
colorFrom: indigo
colorTo: red
sdk: gradio
sdk_version: 4.27.0
app_file: app.py
pinned: false
---
# Neural Style Transfer Deployment
Deployment of the neural style transfer algorithm. Users can input their own style and content images and train the algorithm with their own hyperparameters. If you need a pair of images to expiriment with, I reccomend President.jpg as the content image and SelfPortrait.jpg as the style image. I get good results with 5 epochs, a learning rate of 0.01, a content loss of 1, and a style loss of 0.01.\
Importantly, this algorithm is running on a CPU so keep training time in mind when setting the number of epochs.
## Files
- app.py: A copy of the interface I defined in the Python notebook
- vgg19-dcbb9e9d.pth: The vgg19 pre-trained weights (so I don't have to query them each time I load the model)
- President.jpg: Sample content image
- SelfPortrait.jpg: Sample style image
|