Vittorio Pippi commited on
Commit
dd7de19
·
1 Parent(s): 7ad9600

Update .gitignore and README.md; remove unused image files and fix style image resizing

Browse files
Files changed (3) hide show
  1. .gitignore +0 -2
  2. README.md +1 -1
  3. sample.png +0 -0
.gitignore CHANGED
@@ -1,8 +1,6 @@
1
  checkpoints
2
  test.py
3
  model.py
4
- sample.png
5
  visual_prompting.py
6
  test.png
7
- sample2.png
8
  output.png
 
1
  checkpoints
2
  test.py
3
  model.py
 
4
  visual_prompting.py
5
  test.png
 
6
  output.png
README.md CHANGED
@@ -72,7 +72,7 @@ style_img = Image.open("my_style_image.png").convert("RGB")
72
 
73
  # Convert the style image to a suitable tensor
74
  style_img = F.to_tensor(style_img)
75
- style_img = F.resize(style_img, (64, 64)) # Example resize
76
  style_img = F.normalize(style_img, [0.5], [0.5]) # Normalize to [-1, 1]
77
  style_img = style_img.unsqueeze(0).cuda()
78
 
 
72
 
73
  # Convert the style image to a suitable tensor
74
  style_img = F.to_tensor(style_img)
75
+ style_img = F.resize((style_img.width * 64 // style_img.height, 64)) # Example resize
76
  style_img = F.normalize(style_img, [0.5], [0.5]) # Normalize to [-1, 1]
77
  style_img = style_img.unsqueeze(0).cuda()
78
 
sample.png ADDED