Fix: when --use-sfm-depth is enabled and the model is given by --colm… (

#3172) Fix: when --use-sfm-depth is enabled and the model is given by --colmap-model-path. It searches the wrong reconstruction directory Co-authored-by: J.Y <[email protected]>
nerfstudio-project · May 28, 2024 · 12f2e68 · 12f2e68
1 parent 67d9fd1
commit 12f2e68
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py
@@ -162,7 +162,9 @@ def _export_depth(self) -> Tuple[Optional[Dict[int, Path]], List[str]]:
             depth_dir = self.output_dir / "depth"
             depth_dir.mkdir(parents=True, exist_ok=True)
             image_id_to_depth_path = colmap_utils.create_sfm_depth(
-                recon_dir=self.output_dir / self.default_colmap_path(),
+                recon_dir=self.absolute_colmap_model_path
+                if self.skip_colmap
+                else self.output_dir / self.default_colmap_path(),
                 output_dir=depth_dir,
                 include_depth_debug=self.include_depth_debug,
                 input_images_dir=self.image_dir,