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())
|