diff --git a/nerfstudio/scripts/process_data.py b/nerfstudio/scripts/process_data.py index b2c2fa13f..7a0968af5 100644 --- a/nerfstudio/scripts/process_data.py +++ b/nerfstudio/scripts/process_data.py @@ -152,6 +152,9 @@ class ProcessPolycam(BaseConverterToNerfstudioDataset): zip_ref.extractall(self.output_dir) extracted_folder = zip_ref.namelist()[0].split("/")[0] self.data = self.output_dir / extracted_folder + if not (self.data / "keyframes").exists(): + # new versions of polycam data have a different structure, strip the last dir off + self.data = self.output_dir if (self.data / "keyframes" / "corrected_images").exists() and not self.use_uncorrected_images: polycam_image_dir = self.data / "keyframes" / "corrected_images"