tmp

app.py

@@ -37,13 +37,20 @@ transform = transforms.Compose([
         ])
 
 
-# which sf
-sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
+# sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
 
 col = plt.get_cmap('tab10')
 
-def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
+def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids=''):
 
+    # which sf 
+    sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
+    if sf_ids.lower().startswith('r'):
+        n_sf_ids = int(sf_ids[1:])
+        sf_idx_ = np.random.randint(256, size=n_sf_ids)
+    elif sf_ids != '':
+        sf_idx_ = map(int, sf_ids.strip().split(','))
+    
     im1_tensor = transform(im1).unsqueeze(0)
     im2_tensor = transform(im2).unsqueeze(0)
 
@@ -74,7 +81,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
         att_heat = np.array(attns1[0,i,:,:].numpy(), dtype=np.float32)
         att_heat = np.uint8(att_heat / np.max(att_heat[:]) * 255.0)
         att_heat_bin  = np.where(att_heat>threshold, 255, 0)
-        print(att_heat_bin)
+        # print(att_heat_bin)
         all_att_bin1.append(att_heat_bin)
 
         att_heat = np.array(attns2[0,i,:,:].numpy(), dtype=np.float32)
@@ -164,10 +171,11 @@ css = ".input_image, .input_image {height: 600px !important; width: 600px !impor
 iface = gr.Interface(
     fn=generate_matching_superfeatures,
     inputs=[
-        gr.inputs.Image(shape=(1024, 1024), type="pil"),
-        gr.inputs.Image(shape=(1024, 1024), type="pil"),
+        gr.inputs.Image(shape=(1024, 1024), type="pil", label="First Image"),
+        gr.inputs.Image(shape=(1024, 1024), type="pil", label="Second Image"),
         gr.inputs.Slider(minimum=1, maximum=7, step=1, default=3, label="Scale"),
-        gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold")],
+        gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
+        gr.inputs.Textbox(lines=1, default="", label="Super-feature IDs to show", optional=True)],
     outputs=["plot", "plot"],
     # outputs=gr.outputs.Image(shape=(1024,2048), type="plot"),
     title=title,