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]
|