diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index d92282755..2539a4aa0 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -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 diff --git a/nerfstudio/data/utils/dataparsers_utils.py b/nerfstudio/data/utils/dataparsers_utils.py index 12ac73ee0..7e8e7bfff 100644 --- a/nerfstudio/data/utils/dataparsers_utils.py +++ b/nerfstudio/data/utils/dataparsers_utils.py @@ -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)) -