Spaces:
Build error
Build error
edits
Browse files
app.py
CHANGED
@@ -63,6 +63,7 @@ transform = transforms.Compose([
|
|
63 |
def match(query_feat, pos_feat, LoweRatioTh=0.9):
|
64 |
# first perform reciprocal nn
|
65 |
dist = torch.cdist(query_feat, pos_feat)
|
|
|
66 |
best1 = torch.argmin(dist, dim=1)
|
67 |
best2 = torch.argmin(dist, dim=0)
|
68 |
print('best2.size',best2.size())
|
|
|
63 |
def match(query_feat, pos_feat, LoweRatioTh=0.9):
|
64 |
# first perform reciprocal nn
|
65 |
dist = torch.cdist(query_feat, pos_feat)
|
66 |
+
print('dist.size',dist.size())
|
67 |
best1 = torch.argmin(dist, dim=1)
|
68 |
best2 = torch.argmin(dist, dim=0)
|
69 |
print('best2.size',best2.size())
|