YannisK commited on
Commit
dd99365
1 Parent(s): 6bc6828
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -120,8 +120,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
120
  attns2 = output2[1][0]
121
  strenghts2 = output2[2][0]
122
 
123
- feats1n = F.normalize(feats1, dim=1)
124
- feats2n = F.normalize(feats2, dim=1)
125
  print('feats1n.shape', feats1n.shape)
126
  ind_match = match(feats1n, feats2n)
127
  print('ind', ind_match)
 
120
  attns2 = output2[1][0]
121
  strenghts2 = output2[2][0]
122
 
123
+ feats1n = F.normalize(torch.squeeze(feats1), dim=1)
124
+ feats2n = F.normalize(torch.squeeze(feats2), dim=1)
125
  print('feats1n.shape', feats1n.shape)
126
  ind_match = match(feats1n, feats2n)
127
  print('ind', ind_match)