From 48047d7a4de6fb3c10d585029fa3e97e6b1417f2 Mon Sep 17 00:00:00 2001
From: gperda <giulioperda@gmail.com>
Date: Wed, 27 Nov 2024 15:37:18 +0100
Subject: [PATCH] MST creation and updated graph vis

---
 main.py                          | 15 +++++++++++----
 src/deep_image_matching/graph.py |  1 +
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index c4d0dd4..dc80809 100644
--- a/main.py
+++ b/main.py
@@ -8,6 +8,7 @@
 from deep_image_matching.utils.loftr_roma_to_multiview import LoftrRomaToMultiview
 import yaml
 
+
 start_time = time.time()
 
 logger = dim.setup_logger("dim")
@@ -37,6 +38,7 @@
 )
 
 import shutil
+
 shutil.copyfile(database_path, output_dir / "debug.db")
 
 if matcher.matching in ["loftr", "se2loftr", "roma", "srif"]:
@@ -45,8 +47,9 @@
     LoftrRomaToMultiview(
         input_dir=feature_path.parent,
         output_dir=feature_path.parent,
-        image_dir=imgs_dir, 
-        img_ext=image_format)
+        image_dir=imgs_dir,
+        img_ext=image_format,
+    )
 
 # Visualize view graph
 if config.general["graph"]:
@@ -107,7 +110,11 @@
         #     },
         # )
         reconst_opts = {}
-        refine_intrinsics = config.general["refine_intrinsics"] if "refine_intrinsics" in config.general else True
+        refine_intrinsics = (
+            config.general["refine_intrinsics"]
+            if "refine_intrinsics" in config.general
+            else True
+        )
 
         # Run reconstruction
         model = reconstruction.pycolmap_reconstruction(
@@ -122,4 +129,4 @@
 end_time = time.time()
 
 total_time = end_time - start_time
-print(f"Total processing time: {total_time:.2f} seconds")
\ No newline at end of file
+print(f"Total processing time: {total_time:.2f} seconds")
diff --git a/src/deep_image_matching/graph.py b/src/deep_image_matching/graph.py
index d69c81c..3ed3230 100644
--- a/src/deep_image_matching/graph.py
+++ b/src/deep_image_matching/graph.py
@@ -21,6 +21,7 @@ def save_output_graph(G, name):
 
     # HTML template for view graph details panel
     nt.set_template(os.path.join(TEMPLATE_DIR, "template.html").replace("\\", "/"))
+    print(os.path.join(TEMPLATE_DIR, "template.html").replace("\\", "/"))
     nt.from_nx(G)
     nt.toggle_physics(False)