Fixing the image_dir if using skip_image_processing (nerfstudio-proje…

…ct#2111) Co-authored-by: khoa <[email protected]> Co-authored-by: Matthew Tancik <[email protected]>
YueYin27 · Jun 21, 2023 · b747db6 · b747db6
1 parent ccb8a25
commit b747db6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py b/nerfstudio/process_data/colmap_converter_to_nerfstudio_dataset.py
@@ -189,9 +189,15 @@ def _run_colmap(self, mask_path: Optional[Path] = None):
         if self.refine_pixsfm:
             assert sfm_tool == "hloc", "refine_pixsfm only works with sfm_tool hloc"
 
+        # set the image_dir if didn't copy
+        if self.skip_image_processing:
+            image_dir = self.data
+        else:
+            image_dir = self.image_dir
+
         if sfm_tool == "colmap":
             colmap_utils.run_colmap(
-                image_dir=self.image_dir,
+                image_dir=image_dir,
                 colmap_dir=self.absolute_colmap_path,
                 camera_model=CAMERA_MODELS[self.camera_type],
                 camera_mask_path=mask_path,
@@ -208,7 +214,7 @@ def _run_colmap(self, mask_path: Optional[Path] = None):
             assert matcher_type is not None
             assert matcher_type != "NN"  # Only used for colmap.
             hloc_utils.run_hloc(
-                image_dir=self.image_dir,
+                image_dir=image_dir,
                 colmap_dir=self.absolute_colmap_path,
                 camera_model=CAMERA_MODELS[self.camera_type],
                 verbose=self.verbose,