Spaces:
Sleeping
Sleeping
John Guerrerio
commited on
Commit
·
4e4e1e5
1
Parent(s):
b947961
updated readme and added sample images
Browse files- .gitattributes +1 -0
- President.jpg +3 -0
- README.md +10 -1
- SelfPortrait.jpg +3 -0
.gitattributes
CHANGED
|
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
*.psd filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
*.psd filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
*.jpg filter=lfs diff=lfs merge=lfs -text
|
President.jpg
ADDED
|
Git LFS Details
|
README.md
CHANGED
|
@@ -8,5 +8,14 @@ sdk_version: 4.27.0
|
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
---
|
|
|
|
|
|
|
| 11 |
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
---
|
| 11 |
+
# Neural Style Transfer Deployment
|
| 12 |
+
### DALI 2024S - Machine Learning Track
|
| 13 |
|
| 14 |
+
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.\
|
| 15 |
+
Importantly, this algorithm is running on a CPU so keep training time in mind when setting the number of epochs.
|
| 16 |
+
|
| 17 |
+
## Files
|
| 18 |
+
- app.py: A copy of the interface I defined in the Python notebook
|
| 19 |
+
- vgg19-dcbb9e9d.pth: The vgg19 pre-trained weights (so I don't have to query them each time I load the model)
|
| 20 |
+
- President.jpg: Sample content image
|
| 21 |
+
- SelfPortrait.jpg: Sample style image
|
SelfPortrait.jpg
ADDED
|
Git LFS Details
|