add: output file

common/app_class.py

@@ -254,6 +254,9 @@ class ImageMatchingApp:
                         with gr.Accordion(
                             "Open for More: Matches Statistics", open=False
                         ):
+                            output_pred = gr.File(
+                                label="Outputs", elem_id="download"
+                            )
                             matches_result_info = gr.JSON(
                                 label="Matches Statistics"
                             )
@@ -299,6 +302,7 @@ class ImageMatchingApp:
                         geometry_result,
                         output_wrapped,
                         state_cache,
+                        output_pred,
                     ]
                     # button callbacks
                     button_run.click(
@@ -327,6 +331,7 @@ class ImageMatchingApp:
                         ransac_confidence,
                         ransac_max_iter,
                         choice_geometry_type,
+                        output_pred,
                     ]
                     button_reset.click(
                         fn=self.ui_reset_state,
@@ -349,6 +354,7 @@ class ImageMatchingApp:
                             output_matches_ransac,
                             matches_result_info,
                             output_wrapped,
+                            output_pred,
                         ],
                     )
 
@@ -492,6 +498,7 @@ class ImageMatchingApp:
             ],  # ransac_confidence: float
             self.cfg["defaults"]["ransac_max_iter"],  # ransac_max_iter: int
             self.cfg["defaults"]["setting_geometry"],  # geometry: str
+            None,  # predictions
         )
 
     def display_supported_algorithms(self, style="tab"):

common/utils.py

@@ -24,6 +24,8 @@ from .viz import (
 import time
 import matplotlib.pyplot as plt
 import warnings
+import tempfile
+import pickle
 
 warnings.simplefilter("ignore")
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -774,6 +776,14 @@ def run_ransac(
 
     num_matches_raw = state_cache["num_matches_raw"]
     state_cache["wrapped_image"] = warped_image
+
+    # tmp_state_cache = tempfile.NamedTemporaryFile(suffix='.pkl', delete=False)
+    tmp_state_cache = "output.pkl"
+    with open(tmp_state_cache, "wb") as f:
+        pickle.dump(state_cache, f)
+
+    logger.info(f"Dump results done!")
+
     return (
         output_matches_ransac,
         {
@@ -781,6 +791,7 @@ def run_ransac(
             "num_matches_ransac": num_matches_ransac,
         },
         output_wrapped,
+        tmp_state_cache,
     )
 
 
@@ -961,6 +972,12 @@ def run_matching(
     state_cache["num_matches_raw"] = num_matches_raw
     state_cache["num_matches_ransac"] = num_matches_ransac
     state_cache["wrapped_image"] = warped_image
+
+    # tmp_state_cache = tempfile.NamedTemporaryFile(suffix='.pkl', delete=False)
+    tmp_state_cache = "output.pkl"
+    with open(tmp_state_cache, "wb") as f:
+        pickle.dump(state_cache, f)
+    logger.info(f"Dump results done!")
     return (
         output_keypoints,
         output_matches_raw,
@@ -978,6 +995,7 @@ def run_matching(
         },
         output_wrapped,
         state_cache,
+        tmp_state_cache,
     )
 
 

hloc/match_features.py

@@ -386,7 +386,7 @@ def match_images(model, feat0, feat1):
         "mkeypoints1": mkpts1,
         "mkeypoints0_orig": mkpts0_origin.numpy(),
         "mkeypoints1_orig": mkpts1_origin.numpy(),
-        "mconf": mconfid,
+        "mconf": mconfid.numpy(),
     }
     del feat0, feat1, desc0, desc1, kpts0, kpts1, kpts0_origin, kpts1_origin
     torch.cuda.empty_cache()