YannisK commited on
Commit
c8e8ad1
1 Parent(s): 64e288e
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -65,7 +65,7 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
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)
69
  arange = torch.arange(best2.size(0))
70
  reciprocal = best1[best2]==arange
71
  # check Lowe ratio test
@@ -260,9 +260,9 @@ iface = gr.Interface(
260
  article=article,
261
  css=css,
262
  examples=[
263
- ["chateau_1.png", "chateau_2.png", 2, 100, '55,14,5,4,52,57,40,9'],
264
- ["anafi1.jpeg", "anafi2.jpeg", 4, 50, '99,100,142,213,236'],
265
- ["areopoli1.jpeg", "areopoli2.jpeg", 4, 50, '72,44,142,213,236'],
266
  ]
267
  )
268
  iface.launch(enable_queue=True)
 
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())
69
  arange = torch.arange(best2.size(0))
70
  reciprocal = best1[best2]==arange
71
  # check Lowe ratio test
 
260
  article=article,
261
  css=css,
262
  examples=[
263
+ ["chateau_1.png", "chateau_2.png", 2, 100, '55,14,5,4,52,57,40,9', True],
264
+ ["anafi1.jpeg", "anafi2.jpeg", 4, 50, '99,100,142,213,236', True],
265
+ ["areopoli1.jpeg", "areopoli2.jpeg", 4, 50, '72,44,142,213,236', True],
266
  ]
267
  )
268
  iface.launch(enable_queue=True)