solved bug exporing into colmap

3DOM-FBK · Sep 26, 2024 · 31573ef · 31573ef
1 parent 2d91358
commit 31573ef
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/deep_image_matching/utils/loftr_roma_to_multiview.py b/src/deep_image_matching/utils/loftr_roma_to_multiview.py
@@ -122,7 +122,7 @@ def get_unique_idxs(A, dim=1):
 
 def import_into_colmap(img_dir,
                        feature_dir ='.featureout',
-                       database_path = 'colmap.db'
+                       database_path = 'colmap.db',
                        ):
     db = COLMAPDatabase.connect(database_path)
     #db.create_tables()
@@ -215,15 +215,16 @@ def LoftrRomaToMultiview(
             for k2, match in gr.items():
                 group[k2] = match
 
-    try:
-        os.remove(f"{output_dir}/database.db")
-    except:
-        pass
+    #try:
+    #    os.remove(f"{output_dir}/database.db")
+    #except:
+    #    pass
 
     import_into_colmap(
         image_dir, 
         feature_dir=f"{output_dir}", 
         database_path=f"{output_dir}/database.db")
+
 
 
 if __name__ == '__main__':