From 4145f010d2269c996f68163f9883d996b71547d3 Mon Sep 17 00:00:00 2001
From: Pythonix Huang <370048201@qq.com>
Date: Mon, 27 Mar 2023 11:33:54 +0800
Subject: [PATCH] Update pixsfm with better param (#1661)

Previous setting will easily go to oom
---
 nerfstudio/process_data/hloc_utils.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/nerfstudio/process_data/hloc_utils.py b/nerfstudio/process_data/hloc_utils.py
index 6ec71edd0..f04f6c32c 100644
--- a/nerfstudio/process_data/hloc_utils.py
+++ b/nerfstudio/process_data/hloc_utils.py
@@ -117,7 +117,13 @@ def run_hloc(
         camera_model=camera_model.value
     )
     if refine_pixsfm:
-        sfm = PixSfM({"dense_features": {"max_edge": 1024}})
+        sfm = PixSfM(
+            conf={
+                "dense_features": {"use_cache": True},
+                "KA": {"dense_features": {"use_cache": True}, "max_kps_per_problem": 1000},
+                "BA": {"strategy": "costmaps"},
+            }
+        )
         refined, _ = sfm.reconstruction(
             sfm_dir,
             image_dir,