Print timestamp for debug

app.py

@@ -328,6 +328,8 @@ def run(config):
             )
 
     app.queue().launch(share=False)
+    import datetime
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'app.queue().launch start')
 
 
 if __name__ == "__main__":

common/utils.py

@@ -327,6 +327,8 @@ def run_matching(
     ransac_max_iter=DEFAULT_RANSAC_MAX_ITER,
     choice_estimate_geom=DEFAULT_SETTING_GEOMETRY,
 ):
+    import datetime
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'run_matching start')
     # image0 and image1 is RGB mode
     if image0 is None or image1 is None:
         raise gr.Error("Error: No images found! Please upload two images.")
@@ -368,7 +370,11 @@ def run_matching(
         "Image 0 - Keypoints",
         "Image 1 - Keypoints",
     ]
+    import datetime
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot_images start')
     output_keypoints = plot_images([image0, image1], titles=titles, dpi=300)
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot_images end')
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot_keypoints start')
     if "keypoints0" in pred.keys() and "keypoints1" in pred.keys():
         plot_keypoints([pred["keypoints0"], pred["keypoints1"]])
         text = (
@@ -377,6 +383,7 @@ def run_matching(
         )
         add_text(0, text, fs=15)
     output_keypoints = fig2im(output_keypoints)
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot_keypoints end')
 
     # plot images with raw matches
     titles = [
@@ -384,9 +391,12 @@ def run_matching(
         "Image 1 - Raw matched keypoints",
     ]
 
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot images with raw matches start')
     output_matches_raw, num_matches_raw = display_matches(pred, titles=titles)
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot images with raw matches end')
 
     # if enable_ransac:
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'filter_matches start')
     filter_matches(
         pred,
         ransac_method=ransac_method,
@@ -394,17 +404,21 @@ def run_matching(
         ransac_confidence=ransac_confidence,
         ransac_max_iter=ransac_max_iter,
     )
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'filter_matches end')
 
     # plot images with ransac matches
     titles = [
         "Image 0 - Ransac matched keypoints",
         "Image 1 - Ransac matched keypoints",
     ]
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot images with raw matches start')
     output_matches_ransac, num_matches_ransac = display_matches(
         pred, titles=titles
     )
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot images with raw matches end')
 
     # plot wrapped images
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot wrapped images start')
     geom_info = compute_geom(pred)
     output_wrapped, _ = change_estimate_geom(
         pred["image0_orig"],
@@ -412,6 +426,7 @@ def run_matching(
         {"geom_info": geom_info},
         choice_estimate_geom,
     )
+    print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'plot wrapped images end')
 
     del pred
 

hloc/matchers/gim.py

@@ -38,7 +38,7 @@ class GIM(BaseModel):
             cmd = ["wget", link, "-O", str(model_path)]
             logger.info(f"Downloading the DKMv3 model with `{cmd}`.")
             subprocess.run(cmd, check=True)
-        logger.info(f"Loading GIM model...")
+        # logger.info(f"Loading GIM model...")
         # self.net = DKMv3(path_to_weights=str(model_path), device=device)
 
         model = DKMv3(None, 672, 896, upsample_preds=True)
@@ -97,8 +97,13 @@ class GIM(BaseModel):
         pad_right1 = pad_width1 - pad_left1
         image0 = torch.nn.functional.pad(image0, (pad_left0, pad_right0, pad_top0, pad_bottom0))
         image1 = torch.nn.functional.pad(image1, (pad_left1, pad_right1, pad_top1, pad_bottom1))
+        import datetime
+        print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'self.net.match start')
         dense_matches, dense_certainty = self.net.match(image0, image1)
+        print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'self.net.match end')
+        print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'self.net.sample start')
         sparse_matches, mconf = self.net.sample(dense_matches, dense_certainty, self.conf["max_keypoints"])
+        print(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), 'self.net.sample end')
         height0, width0 = image0.shape[-2:]
         height1, width1 = image1.shape[-2:]
         kpts0 = sparse_matches[:, :2]