tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

Fix mask filenames misaligned after splitting set into train and eval (#1093)

Browse Source 
* 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 <n.zunhammer@tum.de>
This commit is contained in:
Nicolas Zunhammer 2022-12-06 02:04:40 +01:00 committed by GitHub
parent f3442c8b7b
commit 6896028fc5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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