tmp

app.py

@@ -42,7 +42,8 @@ transform = transforms.Compose([
 col = plt.get_cmap('tab10')
 
 def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids=''):
-
+    print('im1:', im1.size)
+    print('im2:', im2.size)
     # which sf 
     sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
     if sf_ids.lower().startswith('r'):
@@ -173,10 +174,10 @@ 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", label="First Image"),
-        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.Image(shape=(1024, 1024), type="pil", label="First Image"),
+#        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.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
         gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
         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)],