YannisK commited on
Commit
eee2bf2
1 Parent(s): c8e8ad1
Files changed (1) hide show
  1. app.py +1 -0
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())