Spaces:
Build error
Build error
temp
Browse files
app.py
CHANGED
@@ -42,12 +42,12 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
|
|
42 |
|
43 |
# extract features
|
44 |
with torch.no_grad():
|
45 |
-
output1 = net.get_superfeatures(
|
46 |
feats1 = output1[0]
|
47 |
attns1 = output1[1]
|
48 |
strenghts1 = output1[2]
|
49 |
|
50 |
-
output2 = net.get_superfeatures(
|
51 |
feats2 = output2[0]
|
52 |
attns2 = output2[1]
|
53 |
strenghts2 = output2[2]
|
|
|
42 |
|
43 |
# extract features
|
44 |
with torch.no_grad():
|
45 |
+
output1 = net.get_superfeatures(im1_tensor.to(device), scales=scales)
|
46 |
feats1 = output1[0]
|
47 |
attns1 = output1[1]
|
48 |
strenghts1 = output1[2]
|
49 |
|
50 |
+
output2 = net.get_superfeatures(im2_tensor.to(device), scales=scales)
|
51 |
feats2 = output2[0]
|
52 |
attns2 = output2[1]
|
53 |
strenghts2 = output2[2]
|