steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dataset stretch now optional

Browse Source
This commit is contained in:
Si11ium
2020-05-15 09:43:30 +02:00
parent f458660613
commit 09f9dd9131
4 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
datasets/binar_masks.py
View File

@ -19,7 +19,7 @@ class BinaryMasksDataset(Dataset):
def sample_shape(self):
return self[0][0].shape
def __init__(self, data_root, setting, mel_transforms, transforms=None, mixup=False, stretch_dataset=True):
def __init__(self, data_root, setting, mel_transforms, transforms=None, mixup=False, stretch_dataset=False):
self.stretch = stretch_dataset
assert isinstance(setting, str), f'Setting has to be a string, but was: {type(setting)}.'
assert setting in V.DATA_OPTIONS, f'Setting must match one of: {V.DATA_OPTIONS}.'