From 6896028fc503b00947d04b7ed142a8dbd685749d Mon Sep 17 00:00:00 2001 From: Nicolas Zunhammer Date: Tue, 6 Dec 2022 02:04:40 +0100 Subject: [PATCH] Fix mask filenames misaligned after splitting set into train and eval (#1093) * Added Dockerfile to build docker image including all dependencies and nerfstudio, added description on how to user docker. * Move docker description to Installation.md. * Fix mask_filename indices after splitting train set. Co-authored-by: Nicolas Zunhammer --- nerfstudio/data/dataparsers/nerfstudio_dataparser.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index e2b86bd6d..6dd41738b 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -196,6 +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] poses = poses[indices] # in x,y,z order