From e3443961531c809e951b99eaa3d2d1f73e46a45f Mon Sep 17 00:00:00 2001 From: evonneng Date: Thu, 8 Sep 2022 13:28:44 -0700 Subject: [PATCH] added kwargs instaed of data (#373) --- nerfactory/datamanagers/dataparsers/friends_parser.py | 4 ++-- nerfactory/datamanagers/datasets.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/nerfactory/datamanagers/dataparsers/friends_parser.py b/nerfactory/datamanagers/dataparsers/friends_parser.py index 3bfbf0e7f..2dbdf782f 100644 --- a/nerfactory/datamanagers/dataparsers/friends_parser.py +++ b/nerfactory/datamanagers/dataparsers/friends_parser.py @@ -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 diff --git a/nerfactory/datamanagers/datasets.py b/nerfactory/datamanagers/datasets.py index 13b430320..50744b7d9 100644 --- a/nerfactory/datamanagers/datasets.py +++ b/nerfactory/datamanagers/datasets.py @@ -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):