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- .gitignore +0 -2
- README.md +1 -1
- 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
|
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
![]() |