tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

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

Browse Source
This commit is contained in:
ksnzh 2022-12-06 14:10:55 +08:00 committed by GitHub
parent 006703bb64
commit e605b3344a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/data/dataparsers/nerfstudio_dataparser.py
View File

@ -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