YannisK commited on
Commit
64e288e
1 Parent(s): 66b64bf
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -65,6 +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
  arange = torch.arange(best2.size(0))
69
  reciprocal = best1[best2]==arange
70
  # check Lowe ratio test
@@ -245,7 +246,7 @@ iface = gr.Interface(
245
  gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
246
  gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
247
  gr.inputs.Textbox(lines=1, default="", label="SF IDs to show (comma separated numbers from 0-255; typing 'rX' will return X random SFs", optional=True),
248
- gr.inputs.Checkbox(default=True, label="Show only matching", optional=False),
249
  ],
250
  outputs=[
251
  gr.outputs.Image(type="plot", label="First Image SFs"),
 
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
 
246
  gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
247
  gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
248
  gr.inputs.Textbox(lines=1, default="", label="SF IDs to show (comma separated numbers from 0-255; typing 'rX' will return X random SFs", optional=True),
249
+ gr.inputs.Checkbox(default=True, label="Show only matching SFs", optional=False),
250
  ],
251
  outputs=[
252
  gr.outputs.Image(type="plot", label="First Image SFs"),