MST creation and updated graph vis

Author: gperda

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")
+print(f"Total processing time: {total_time:.2f} seconds")

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)