update

README.md

@@ -4,10 +4,11 @@ emoji: 📚
 colorFrom: red
 colorTo: yellow
 sdk: gradio
+sdk_version: 3.0.5
 app_file: app.py
 pinned: false
 ---
 
 # Convolutional Hough Matching Networks
 
-A demo for Convolutional Hough Matching Networks. [[Paper](https://arxiv.org/abs/2109.05221)] [[Official Github Repo](https://github.com/juhongm999/chm.git)]
+A demo for Convolutional Hough Matching Networks. [[Paper](https://arxiv.org/abs/2109.05221)] [[Official Github Repo](https://github.com/juhongm999/chm.git)]

app.py

@@ -26,7 +26,13 @@ from model import chmnet
 from model.base.geometry import Geometry
 
 # Downloading the Model
-gdown.download(id="1zsJRlAsoOn5F0GTCprSFYwDDfV85xDy6", output="pas_psi.pt", quiet=False)
+# gdown.download(id="1zsJRlAsoOn5F0GTCprSFYwDDfV85xDy6", output="pas_psi.pt", quiet=False)
+
+md5 = "6b7b4d7bad7f89600fac340d6aa7708b"
+
+gdown.cached_download(
+    url="1zsJRlAsoOn5F0GTCprSFYwDDfV85xDy6", path="pas_psi.pt", quiet=False, md5=md5
+)
 
 # Model Initialization
 args = dict(
@@ -189,27 +195,56 @@ def generate_correspondences(
     )
 
 
-# GRADIO APP
-title = "Correspondence Matching with Convolutional Hough Matching Networks "
-description = "Performs keypoint transform from a 7x7 gird on the source image to the target image. Use the sliders to adjust the grid."
-article = "<p style='text-align: center'><a href='https://github.com/juhongm999/chm' target='_blank'>Original Github Repo</a></p>"
-
-iface = gr.Interface(
+# Gradio App
+main = gr.Interface(
     fn=generate_correspondences,
     inputs=[
-        gr.inputs.Image(shape=(240, 240), type="pil"),
-        gr.inputs.Image(shape=(240, 240), type="pil"),
-        gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label="Min X"),
-        gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label="Max X"),
-        gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label="Min Y"),
-        gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label="Max Y"),
+        gr.Image(shape=(240, 240), type="pil"),
+        gr.Image(shape=(240, 240), type="pil"),
+        gr.Slider(minimum=1, maximum=240, step=1, default=15, label="Min X"),
+        gr.Slider(minimum=1, maximum=240, step=1, default=215, label="Max X"),
+        gr.Slider(minimum=1, maximum=240, step=1, default=15, label="Min Y"),
+        gr.Slider(minimum=1, maximum=240, step=1, default=215, label="Max Y"),
+    ],
+    allow_flagging="never",
+    outputs="plot",
+    examples=[
+        ["./examples/sample1.jpeg", "./examples/sample2.jpeg", 17, 223, 17, 223],
+        [
+            "./examples/Red_Winged_Blackbird_0012_6015.jpg",
+            "./examples/Red_Winged_Blackbird_0025_5342.jpg",
+            17,
+            223,
+            17,
+            223,
+        ],
+        [
+            "./examples/Yellow_Headed_Blackbird_0026_8545.jpg",
+            "./examples/Yellow_Headed_Blackbird_0020_8549.jpg",
+            17,
+            223,
+            17,
+            223,
+        ],
     ],
-    outputs=[gr.outputs.Plot(type="matplotlib")],
-    enable_queue=True,
-    title=title,
-    description=description,
-    article=article,
-    examples=[["sample1.jpeg", "sample2.jpeg", 15, 215, 15, 215]],
 )
 
-iface.launch()
+
+blocks = gr.Blocks()
+with blocks:
+
+    gr.Markdown(
+        """
+# Correspondence Matching with Convolutional Hough Matching Networks
+Performs keypoint transform from a 7x7 gird on the source image to the target image. Use the sliders to adjust the grid.
+[Original Paper](https://arxiv.org/abs/2103.16831) - [Github Page](https://github.com/juhongm999/chm)
+    """
+    )
+
+    gr.TabbedInterface([main], ["Main"])
+
+
+blocks.launch(
+    debug=True,
+    enable_queue=False,
+)

requirements.txt

@@ -1,11 +1,12 @@
-gradio==2.4.5
-matplotlib==3.4.3
-numpy==1.21.2
+gradio==3.0.5
 pandas==1.3.4
-Pillow==8.4.0
 requests==2.26.0
 scipy==1.7.1
-tensorboardX==2.4.1
-torch==1.10.0
-torchvision==0.11.1
+tensorboardX==2.5
+matplotlib==3.5.2
+numpy==1.21.5
+Pillow==9.0.1
+scikit-image==0.19.2
+torch==1.11.0
+torchvision==0.12.0
 gdown