YannisK commited on
Commit
66b64bf
1 Parent(s): 881aeeb
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -136,6 +136,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
136
  print(attns1.shape, attns2.shape)
137
  print(strenghts1.shape, strenghts2.shape)
138
 
 
 
139
  # Store all binary SF att maps to show them all at once in the end
140
  all_att_bin1 = []
141
  all_att_bin2 = []
@@ -242,9 +244,8 @@ 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
- "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"),
 
136
  print(attns1.shape, attns2.shape)
137
  print(strenghts1.shape, strenghts2.shape)
138
 
139
+ # if only_matching:
140
+
141
  # Store all binary SF att maps to show them all at once in the end
142
  all_att_bin1 = []
143
  all_att_bin2 = []
 
244
  gr.inputs.Image(type="pil", label="Second Image"),
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"),