edits

app.py

@@ -91,7 +91,7 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
 
 col = plt.get_cmap('tab10')
 
-def generate_matching_superfeatures(im1, im2, model_fn, scale_id=6, threshold=50, sf_ids='', only_matching=True):
+def generate_matching_superfeatures(im1, im2, model_fn, scale_id=6, threshold=50, random_mode=False, sf_ids='', only_matching=True):
     print('im1:', im1.size)
     print('im2:', im2.size)
 
@@ -142,13 +142,10 @@ def generate_matching_superfeatures(im1, im2, model_fn, scale_id=6, threshold=50
     
     # which sf 
     sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
-    random_mode = False
-    n_sf_ids = 0
-    if sf_ids.lower().startswith('r'):
-        random_mode = True
-        n_sf_ids = int(sf_ids[1:])
+    n_sf_ids = 8
+    if random_mode or sf_ids == '':
         sf_idx_ = np.random.randint(256, size=n_sf_ids)
-    elif sf_ids != '':
+    else:
         sf_idx_ = map(int, sf_ids.strip().split(','))
         
     if only_matching:
@@ -267,8 +264,9 @@ iface = gr.Interface(
         gr.inputs.Radio(["ImageNet", "SfM-120k (landmarks)"], label="Model", optional=False),
         gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
         gr.inputs.Slider(minimum=0, maximum=255, step=25, default=150, label="Binarization Threshold"),
+        gr.inputs.Checkbox(default=False, label="Show random matching SFs"),
         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),
-        gr.inputs.Checkbox(default=True, label="Show only matching SFs", optional=False),
+        gr.inputs.Checkbox(default=True, label="Show only matching SFs"),
         ],
     outputs=[
         gr.outputs.Image(type="plot", label="First Image SFs"),