edits

app.py

@@ -91,12 +91,12 @@ 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, random_mode=False, sf_ids='', only_matching=True):
+def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6, threshold=50, random_mode=False, sf_ids='', only_matching=True):
     print('im1:', im1.size)
     print('im2:', im2.size)
 
     net = net_sfm
-    if model_fn == "ImageNet":
+    if Imagenet_model:
         net = net_imagenet
 
     # dataset_ = ImgDataset(images=[im1, im2], imsize=1024)
@@ -246,14 +246,6 @@ title = "Visualizing Super-features"
 description = "This is a visualization demo for the ICLR 2022 paper <b><a href='https://github.com/naver/fire' target='_blank'>Learning Super-Features for Image Retrieval</a></p></b>" 
 article = "<p style='text-align: center'><a href='https://github.com/naver/fire' target='_blank'>Original Github Repo</a></p>"
 
-
-# css = ".output-image, .input-image {height: 40rem !important; width: 100% !important;}"
-# css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"
-# css = ".output_image, .input_image {hieght: 1000px !important}"
-css = ".input_image, .input_image {height: 600px !important; width: 600px !important;} "
-# css = ".output-image, .input-image {height: 40rem !important; width: 100% !important;}"
-
-
 iface = gr.Interface(
     fn=generate_matching_superfeatures,
     inputs=[
@@ -261,11 +253,11 @@ iface = gr.Interface(
 #        gr.inputs.Image(shape=(1024, 1024), type="pil", label="Second Image"),
         gr.inputs.Image(type="pil", label="First Image"),
         gr.inputs.Image(type="pil", label="Second Image"),
-        gr.inputs.Radio(["ImageNet", "SfM-120k (landmarks)"], label="Model", optional=False),
+        gr.inputs.Checkbox(default=False, label="Model trained on ImageNet (Default: SfM-120k)"),
         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=False, label="Show random (matching) SFs"),
+        gr.inputs.Textbox(lines=1, default="", label="...or show specific SF IDs:", optional=True),
         gr.inputs.Checkbox(default=True, label="Show only matching SFs"),
         ],
     outputs=[
@@ -278,11 +270,10 @@ iface = gr.Interface(
     layout="horizontal",
     description=description,
     article=article,
-    css=css,
     examples=[
-        ["chateau_1.png", "chateau_2.png", 3, 150, 'r8', True],
-        ["anafi1.jpeg", "anafi2.jpeg", 4, 150, 'r8', True],
-        ["areopoli1.jpeg", "areopoli2.jpeg", 4, 150, 'r8', True],
+        ["chateau_1.png", "chateau_2.png", False, 3, 150, True, '', True],
+        ["anafi1.jpeg", "anafi2.jpeg", False, 4, 150, True, '', True],
+        ["areopoli1.jpeg", "areopoli2.jpeg", False, 4, 150, True, '', True],
     ]
 )
 iface.launch(enable_queue=True)