From 7e23023b31e9bfddde1eeaef0e845b57accab26b Mon Sep 17 00:00:00 2001 From: Mengkun She Date: Mon, 27 May 2024 15:33:29 +0200 Subject: [PATCH] Fix: when --use-sfm-depth is enabled and the model is given by --colmap-model-path. It searches the wrong reconstruction directory --- .../process_data/colmap_converter_to_nerfstudio_dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py index b172433297..4429730368 100644 --- 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,