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

transition

Browse Source
This commit is contained in:
Si11ium
2021-02-01 10:18:30 +01:00
parent 68431b848e
commit f1f327ef17
32 changed files with 22392 additions and 343 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
util/module_mixins.py
View File

@@ -162,7 +162,7 @@ class DatasetMixin:
hop_length=self.params.hop_length)
# Utility
normalize = NormalizeLocal()
utility_transforms = Compose([NormalizeLocal(), ToTensor()])
# Data Augmentations
mel_augmentations = Compose([
@@ -172,7 +172,7 @@ class DatasetMixin:
ShiftTime(0.4),
MaskAug(0.2),
], p=0.6),
normalize])
utility_transforms])
# Datasets
Dataset = namedtuple('Datasets', 'train_dataset val_dataset test_dataset')
@@ -187,13 +187,13 @@ class DatasetMixin:
fold=9,
reset=self.params.reset,
mel_kwargs=mel_kwargs,
mel_augmentations=normalize),
mel_augmentations=utility_transforms),
test_dataset=self.dataset_class(data_root=self.params.root, # TEST DATASET
setting=V.DATA_OPTION_test,
fold=10,
reset=self.params.reset,
mel_kwargs=mel_kwargs,
mel_augmentations=normalize),
mel_augmentations=utility_transforms),
)
if dataset.train_dataset.task_type == V.TASK_OPTION_binary: