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 <132313008+jb-ye@users.noreply.github.com>
This commit is contained in:
SerenitysmkandJ.Y authored and GitHub committed 2024-05-28 15:08:20 -04:00
1 parent 67d9fd1933
commit 12f2e681a4
1 file changed
+3 -1
@@ -162,7 +162,9 @@ class ColmapConverterToNerfstudioDataset(BaseConverterToNerfstudioDataset):
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,