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

LinearModule

Browse Source
This commit is contained in:
Si11ium
2020-05-09 21:56:58 +02:00
parent 5e6b0e598f
commit 3fbc98dfa3
7 changed files with 145 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
main_inference.py
View File

@ -23,7 +23,10 @@ from datasets.binar_masks import BinaryMasksDataset
def prepare_dataloader(config_obj):
mel_transforms = Compose([AudioToMel(n_mels=config_obj.data.n_mels), MelToImage()])
mel_transforms = Compose([
# Audio to Mel Transformations
AudioToMel(sr=config_obj.data.sr, n_mels=config_obj.data.n_mels, n_fft=config_obj.data.n_fft,
hop_length=config_obj.data.hop_length), MelToImage()])
transforms = Compose([NormalizeLocal(), ToTensor()])
dataset: Dataset = BinaryMasksDataset(config_obj.data.root, setting='test',