YannisK commited on
Commit
36eb9c8
1 Parent(s): a15c1b9
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -84,7 +84,7 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
84
 
85
  col = plt.get_cmap('tab10')
86
 
87
- def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids=''):
88
  print('im1:', im1.size)
89
  print('im2:', im2.size)
90
  # which sf
@@ -242,7 +242,10 @@ iface = gr.Interface(
242
  gr.inputs.Image(type="pil", label="Second Image"),
243
  gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
244
  gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
245
- 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)],
 
 
 
246
  outputs=[
247
  gr.outputs.Image(type="plot", label="First Image SFs"),
248
  gr.outputs.Image(type="plot", label="Second Image SFs"),
 
84
 
85
  col = plt.get_cmap('tab10')
86
 
87
+ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='', only_matching):
88
  print('im1:', im1.size)
89
  print('im2:', im2.size)
90
  # which sf
 
242
  gr.inputs.Image(type="pil", label="Second Image"),
243
  gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
244
  gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
245
+ "checkbox",
246
+ 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),
247
+ gr.inputs.Checkbox(default=True, label="Show only matching", optional=False)
248
+ ],
249
  outputs=[
250
  gr.outputs.Image(type="plot", label="First Image SFs"),
251
  gr.outputs.Image(type="plot", label="Second Image SFs"),