From 62f3e6dc863b115cd2546f055b82237b2b92fefb Mon Sep 17 00:00:00 2001 From: Justin Kerr Date: Wed, 22 May 2024 12:55:18 -0700 Subject: [PATCH] polycam bug --- nerfstudio/scripts/process_data.py | 3 +++ 1 file changed, 3 insertions(+) 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"