Upload more examples

datasets/show/txt.txt

@@ -15,4 +15,34 @@ event/000289.png
 vis/000422.png
 event/000422.png
 vis/000522.png
-event/000522.png
+event/000522.png
+Medical/MR_PET/MRI_1.png
+Medical/MR_PET/PET_1.png
+Medical/MR_PET/MRI_2.png
+Medical/MR_PET/PET_2.png
+Medical/MR_PET/MRI_3.png
+Medical/MR_PET/PET_3.png
+Medical/SPECT_CT/CT_1.png
+Medical/SPECT_CT/SPECT_1.png
+Medical/SPECT_CT/CT_10.png
+Medical/SPECT_CT/SPECT_10.png
+Medical\Retina\RetinaFix_32.png
+Medical\Retina\RetinaMove_32.png
+Medical\Retina\RetinaFix_43.png
+Medical\Retina\RetinaMove_43.png
+RemoteSensing\SAR_Optical\SO1a.png
+RemoteSensing\SAR_Optical\SO1b.png
+RemoteSensing\SAR_Optical\SO4a.png
+RemoteSensing\SAR_Optical\SO4b.png
+RemoteSensing\Map_Optical\MO3a.png
+RemoteSensing\Map_Optical\MO3b.png
+RemoteSensing\Map_Optical\MO6a.png
+RemoteSensing\Map_Optical\MO6b.png
+RemoteSensing\DayNight\DN4a.png
+RemoteSensing\DayNight\DN4b.png
+RemoteSensing\DayNight\DN5a.png
+RemoteSensing\DayNight\DN5b.png
+RemoteSensing\Infrared_Optical\IO1a.png
+RemoteSensing\Infrared_Optical\IO1b.png
+RemoteSensing\Infrared_Optical\IO3a.png
+RemoteSensing\Infrared_Optical\IO3b.png

ui/app_class.py

@@ -6,7 +6,7 @@ import gradio as gr
 import numpy as np
 from easydict import EasyDict as edict
 from omegaconf import OmegaConf
-import spaces
+# import spaces
 sys.path.append(str(Path(__file__).parents[1]))
 
 from ui.sfm import SfmEngine
@@ -38,16 +38,18 @@ DESCRIPTION = """
     </div>
 </div>
 
-<p style="text-align: center; font-size: 16px; color: #666;">
-    This Space is derived from <a href="https://github.com/Vincentqyw/image-matching-webui" style="color: #007BFF;">Image Matching WebUI</a>.      
+<p style="text-align: center; font-size: 14px; color: #666;">
+    This Space is derived from <a href="https://github.com/Vincentqyw/image-matching-webui" style="color: #007BFF;">Image Matching WebUI</a>.         
+      
+    We are grateful to the authors for their contribution of the source code. 
 </p>
 
-<p style="text-align: center; font-size: 16px; color: #666;">
-    We are grateful to the authors for their contribution of the source code. 
+<p style="text-align: center; font-size: 14px; color: #666;">
+    Here we provide our MINIMA-model in our paper for test and comparison, and this project is undergoing continuous enhancement.
 </p>
 
 <p style="text-align: center; font-size: 16px; color: #666;">
-    Here we provide our MINIMA-model in our paper for test and comparison, and this project is undergoing continuous enhancement.
+    Special thanks to Hugging Face for providing the ZeroGPU for this Space!
 </p>
 """
 

ui/config.yaml

@@ -1,10 +1,10 @@
 server:
   name: "0.0.0.0"
-  port: 7861
+  port: 7860
 
 defaults:
   setting_threshold: 0.1
-  max_keypoints: 2048
+  max_keypoints: 2000
   keypoint_threshold: 0.0005
   enable_ransac: true
   ransac_method: CV2_USAC_MAGSAC
@@ -22,9 +22,12 @@ matcher_zoo:
     feature: superpoint_max
     dense: false
     info:
-      name: MINIMA(LightGlue) #dispaly name
+      name: MINIMA #dispaly name
+      source: "ArXiv 2024"
+      github: https://github.com/LSXI7/MINIMA
+      paper: https://arxiv.org/abs/2412.19412
       project: null
-      display: false
+      display: true
 
   minima(loftr):
     matcher: minima_loftr

ui/utils.py

@@ -36,7 +36,7 @@ warnings.simplefilter("ignore")
 ROOT = Path(__file__).parent.parent
 # some default values
 DEFAULT_SETTING_THRESHOLD = 0.1
-DEFAULT_SETTING_MAX_FEATURES = 2048
+DEFAULT_SETTING_MAX_FEATURES = 2000
 DEFAULT_DEFAULT_KEYPOINT_THRESHOLD = 0.0005
 DEFAULT_ENABLE_RANSAC = True
 DEFAULT_RANSAC_METHOD = "CV2_USAC_MAGSAC"
@@ -197,8 +197,8 @@ def gen_examples():
     random.seed(1)
     example_matchers = [
         "superpoint+minima(lightglue)",
-        # "minima(loftr)",
-        # "minima(RoMa)",
+        "minima(loftr)",
+        "minima(RoMa)",
     ]
 
     def distribute_elements(A, B):
@@ -251,7 +251,7 @@ def gen_examples():
                 match_setting_threshold,
                 match_setting_max_features,
                 detect_keypoints_threshold,
-                mt,
+                mt if 'Medical' not in pair[0] else 'minima(RoMa)',
                 # enable_ransac,
                 ransac_method,
                 ransac_reproj_threshold,