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

fingerprinted now should work correctly

Browse Source
This commit is contained in:
Si11ium
2020-05-20 13:29:17 +02:00
parent 7dd10d9a14
commit e021e2209b
6 changed files with 28 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
util/module_mixins.py
View File

@ -122,7 +122,8 @@ class BinaryMaskDatasetMixin:
mel_transforms = Compose([
# Audio to Mel Transformations
AudioToMel(sr=self.params.sr, n_mels=self.params.n_mels, n_fft=self.params.n_fft,
hop_length=self.params.hop_length), MelToImage()])
hop_length=self.params.hop_length),
MelToImage()])
# Data Augmentations
aug_transforms = Compose([
RandomApply([
@ -132,7 +133,8 @@ class BinaryMaskDatasetMixin:
MaskAug(self.params.mask_ratio),
], p=0.6),
# Utility
NormalizeLocal(), ToTensor()
NormalizeLocal(),
ToTensor()
])
val_transforms = Compose([NormalizeLocal(), ToTensor()])
@ -143,7 +145,7 @@ class BinaryMaskDatasetMixin:
# TRAIN DATASET
train_dataset=BinaryMasksDataset(self.params.root, setting=V.DATA_OPTIONS.train,
use_preprocessed=self.params.use_preprocessed,
mixup=self.params.mixup, stretch_dataset=self.params.stretch,
stretch_dataset=self.params.stretch,
mel_transforms=mel_transforms_train, transforms=aug_transforms),
# VALIDATION DATASET
val_train_dataset=BinaryMasksDataset(self.params.root, setting=V.DATA_OPTIONS.train,