YannisK commited on
Commit
2ada6f2
·
1 Parent(s): 880da41
Files changed (1) hide show
  1. app.py +2 -2
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(im1.to(device), scales=scales)
46
  feats1 = output1[0]
47
  attns1 = output1[1]
48
  strenghts1 = output1[2]
49
 
50
- output2 = net.get_superfeatures(im2.to(device), scales=scales)
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]