tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

reformat

Browse Source
This commit is contained in:
Aaryan Chandna 2025-06-15 19:10:32 -07:00
parent 4bc47e9c95
commit aeafdafa57
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,8 +31,8 @@ from nerfstudio.data.utils.dataparsers_utils import (
get_train_eval_split_all,
get_train_eval_split_filename,
get_train_eval_split_fraction,
get_train_eval_split_indices,
get_train_eval_split_interval,
get_train_eval_split_indices
)
from nerfstudio.utils.io import load_from_json
from nerfstudio.utils.rich_utils import CONSOLE

6
nerfstudio/data/utils/dataparsers_utils.py
View File

@ -100,7 +100,10 @@ def get_train_eval_split_all(image_filenames: List) -> Tuple[np.ndarray, np.ndar
i_eval = i_all
return i_train, i_eval
def get_train_eval_split_indices(image_filenames: List, eval_image_indices: Tuple[int, ...]) -> Tuple[np.ndarray, np.ndarray]:
def get_train_eval_split_indices(
image_filenames: List, eval_image_indices: Tuple[int, ...]
) -> Tuple[np.ndarray, np.ndarray]:
"""
Get the train/eval split based on specified indices in the config.
@ -116,4 +119,3 @@ def get_train_eval_split_indices(image_filenames: List, eval_image_indices: Tupl
train_indices = all_indices - eval_indices
return np.array(sorted(train_indices)), np.array(sorted(eval_indices))