From e605b3344ac623fd96e2458ef98f8fbc7a97d718 Mon Sep 17 00:00:00 2001 From: ksnzh Date: Tue, 6 Dec 2022 14:10:55 +0800 Subject: [PATCH] Fix IndexError if ther is no mask filenames (#1095) --- nerfstudio/data/dataparsers/nerfstudio_dataparser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index 6dd41738b..347a292a5 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -196,7 +196,7 @@ class Nerfstudio(DataParser): # 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