Skip to content

Commit

Permalink
fixing indentation mistake
Browse files Browse the repository at this point in the history
  • Loading branch information
achaiah authored and achaiah committed May 10, 2019
1 parent 432b8ad commit da4c874
Showing 1 changed file with 45 additions and 43 deletions.
88 changes: 45 additions & 43 deletions pywick/datasets/MultiFolderDataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,9 @@ class MultiFolderDataset(FolderDataset):
For semantic segmentation this is required so the default is a binary mask. However, if you want to turn off
this feature then specify target_index_map=None
"""
def __init__(self,


def __init__(self,
roots,
class_mode='label',
class_to_idx=None,
Expand All @@ -86,45 +88,45 @@ def __init__(self,
exclusion_file=None,
target_index_map=None):

# call the super constructor first, then set our own parameters
# super().__init__()
self.num_inputs = 1 # these are hardcoded for the fit module to work
self.num_targets = 1 # these are hardcoded for the fit module to work

if default_loader == 'npy':
default_loader = npy_loader
elif default_loader == 'pil':
default_loader = pil_loader
self.default_loader = default_loader

# separate loading for targets (e.g. for black/white masks)
self.target_loader = target_loader

if class_to_idx:
self.classes = class_to_idx.keys()
self.class_to_idx = class_to_idx
else:
self.classes, self.class_to_idx = _find_classes(roots)

data_list = list()
for root in roots:
datai, _ = _finds_inputs_and_targets(root, class_mode=class_mode, class_to_idx=self.class_to_idx, input_regex=input_regex,
rel_target_root=rel_target_root, target_prefix=target_prefix, target_postfix=target_postfix,
target_extension=target_extension, exclusion_file=exclusion_file)
data_list.append(datai)

self.data = list(itertools.chain.from_iterable(data_list))

if len(self.data) == 0:
raise (RuntimeError('Found 0 data items in subfolders of: {}'.format(roots)))
else:
print('Found %i data items' % len(self.data))

self.roots = [os.path.expanduser(x) for x in roots]
self.transform = transform
self.target_transform = target_transform
self.co_transform = co_transform
self.apply_co_transform_first = apply_co_transform_first
self.target_index_map = target_index_map

self.class_mode = class_mode
# call the super constructor first, then set our own parameters
# super().__init__()
self.num_inputs = 1 # these are hardcoded for the fit module to work
self.num_targets = 1 # these are hardcoded for the fit module to work

if default_loader == 'npy':
default_loader = npy_loader
elif default_loader == 'pil':
default_loader = pil_loader
self.default_loader = default_loader

# separate loading for targets (e.g. for black/white masks)
self.target_loader = target_loader

if class_to_idx:
self.classes = class_to_idx.keys()
self.class_to_idx = class_to_idx
else:
self.classes, self.class_to_idx = _find_classes(roots)

data_list = list()
for root in roots:
datai, _ = _finds_inputs_and_targets(root, class_mode=class_mode, class_to_idx=self.class_to_idx, input_regex=input_regex,
rel_target_root=rel_target_root, target_prefix=target_prefix, target_postfix=target_postfix,
target_extension=target_extension, exclusion_file=exclusion_file)
data_list.append(datai)

self.data = list(itertools.chain.from_iterable(data_list))

if len(self.data) == 0:
raise (RuntimeError('Found 0 data items in subfolders of: {}'.format(roots)))
else:
print('Found %i data items' % len(self.data))

self.roots = [os.path.expanduser(x) for x in roots]
self.transform = transform
self.target_transform = target_transform
self.co_transform = co_transform
self.apply_co_transform_first = apply_co_transform_first
self.target_index_map = target_index_map

self.class_mode = class_mode

0 comments on commit da4c874

Please sign in to comment.