Spaces:
Build error
Build error
edits
Browse files
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)
|