Fix IndexError if ther is no mask filenames (#1095)

nerfstudio-project · Dec 6, 2022 · e605b33 · e605b33
1 parent 006703b
commit e605b33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py
@@ -196,7 +196,7 @@ def _generate_dataparser_outputs(self, split="train"):
 
         # Choose image_filenames and poses based on split, but after auto orient and scaling the poses.
         image_filenames = [image_filenames[i] for i in indices]
-        mask_filenames = [mask_filenames[i] for i in indices]
+        mask_filenames = [mask_filenames[i] for i in indices] if len(mask_filenames) > 0 else []
         poses = poses[indices]
 
         # in x,y,z order