steffen/ml_lib
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:23:22 +01:00
parent f6156c6cde
commit 010176e80b
18 changed files with 133 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
audio_toolset/audio_to_mel_dataset.py
View File

@ -20,7 +20,7 @@ class _AudioToMelDataset(Dataset, ABC):
def sampling_rate(self):
raise NotImplementedError
def __init__(self, audio_file_path, label, sample_segment_len=1, sample_hop_len=1, reset=False,
def __init__(self, audio_file_path, label, sample_segment_len=0, sample_hop_len=0, reset=False,
audio_augmentations=None, mel_augmentations=None, mel_kwargs=None, **kwargs):
self.ignored_kwargs = kwargs
self.mel_kwargs = mel_kwargs
@ -46,7 +46,7 @@ class _AudioToMelDataset(Dataset, ABC):
return self.dataset[item]
except FileNotFoundError:
assert self._build_mel()
return self.dataset[item]
return self.dataset[item]
def __len__(self):
return len(self.dataset)
@ -79,7 +79,6 @@ class LibrosaAudioToMelDataset(_AudioToMelDataset):
MelToImage()
])
def _build_mel(self):
if self.reset:
self.mel_file_path.unlink(missing_ok=True)