John Guerrerio commited on
Commit
4e4e1e5
1 Parent(s): b947961

updated readme and added sample images

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. President.jpg +3 -0
  3. README.md +10 -1
  4. 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

  • SHA256: 9e2e51ccd4e079ef888009ff0779de7957ef2e89b98c2070ceb51bc32431f037
  • Pointer size: 130 Bytes
  • Size of remote file: 23.5 kB
README.md CHANGED
@@ -8,5 +8,14 @@ sdk_version: 4.27.0
8
  app_file: app.py
9
  pinned: false
10
  ---
 
 
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
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

  • SHA256: fbdc491884716cbee1894fdb959f9d1f050695ec3098f1c7a905bc43081e34c4
  • Pointer size: 132 Bytes
  • Size of remote file: 1.71 MB