-
Notifications
You must be signed in to change notification settings - Fork 80
Description
Bug description
In connectomics/data/dataset/build.py, line 351 the input loading works the following way:
volume, label, valid_mask = _get_input(cfg, mode, rank, dir_name_init, img_name_init, min_size=sample_volume_size)
However, if we look at the function _get_input definition:
def _get_input(cfg, mode='train', rank=None, preload_data=None, dir_name_init: Optional[list] = None, img_name_init: Optional[list] = None, min_size: Optional[tuple] = None, image_only_test: bool = True):
Which means that in line 351, dir_name_init will be used as preload_data in the _get_input function
Bug fix proposition
Changing line 351 by adding a keyword for dir_name_init:
volume, label, valid_mask = _get_input(cfg, mode, rank, dir_name_init=dir_name_init, img_name_init=img_name_init, min_size=sample_volume_size)