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

added kwargs instaed of data (#373)

Browse Source
This commit is contained in:
evonneng 2022-09-08 13:28:44 -07:00 committed by GitHub
parent e319b8210a
commit e344396153
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nerfactory/datamanagers/dataparsers/friends_parser.py
View File

@ -32,7 +32,7 @@ from nerfactory.utils.io import get_absolute_path, load_from_json
CONSOLE = Console()
def get_semantics_and_masks(semantics: Semantics, image_idx: int):
def get_semantics_and_masks(image_idx: int, semantics: Semantics):
"""function to process additional semantics and mask information"""
# handle mask
person_index = semantics.thing_classes.index("person")
@ -160,6 +160,6 @@ class Friends(DataParser):
image_filenames=image_filenames,
cameras=cameras,
scene_bounds=scene_bounds,
additional_inputs={"semantics": {"data": semantics, "func": get_semantics_and_masks}},
additional_inputs={"semantics": {"func": get_semantics_and_masks, "kwargs": {"semantics": semantics}}},
)
return dataset_inputs

4
nerfactory/datamanagers/datasets.py
View File

@ -83,8 +83,8 @@ class InputDataset(Dataset):
for _, data_func_dict in self.dataset_inputs.additional_inputs.items():
assert "func" in data_func_dict, "Missing function to process data: specify `func` in `additional_inputs`"
func = data_func_dict["func"]
assert "data" in data_func_dict, "No data to process: specify `data` in `additional_inputs`"
data.update(func(data_func_dict["data"], image_idx))
assert "kwargs" in data_func_dict, "No data to process: specify `kwargs` in `additional_inputs`"
data.update(func(image_idx, **data_func_dict["kwargs"]))
return data
def __getitem__(self, image_idx):