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

BandwiseBinaryClassifier is no longer work in progress

Browse Source
This commit is contained in:
Si11ium
2020-05-05 10:58:36 +02:00
parent 451f78f820
commit c2860b0aed
2 changed files with 29 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
datasets/binar_masks.py
View File

@ -30,7 +30,7 @@ class BinaryMasksDataset(Dataset):
self._labels = self._build_labels()
self._wav_folder = self.data_root / 'wav'
self._wav_files = list(sorted(self._labels.keys()))
self._mel_folder = self.data_root / 'transformed'
self._transformed_folder = self.data_root / 'transformed'
def _build_labels(self):
with open(Path(self.data_root) / 'lab' / 'labels.csv', mode='r') as f:
@ -51,13 +51,13 @@ class BinaryMasksDataset(Dataset):
key = self._wav_files[item]
filename = key[:-4] + '.pik'
if not (self._mel_folder / filename).exists():
if not (self._transformed_folder / filename).exists():
raw_sample, sr = librosa.core.load(self._wav_folder / self._wav_files[item])
transformed_sample = self._transforms(raw_sample)
self._mel_folder.mkdir(exist_ok=True, parents=True)
with (self._mel_folder / filename).open(mode='wb') as f:
self._transformed_folder.mkdir(exist_ok=True, parents=True)
with (self._transformed_folder / filename).open(mode='wb') as f:
pickle.dump(transformed_sample, f, protocol=pickle.HIGHEST_PROTOCOL)
with (self._mel_folder / filename).open(mode='rb') as f:
with (self._transformed_folder / filename).open(mode='rb') as f:
sample = pickle.load(f, fix_imports=True)
label = torch.as_tensor(self._labels[key], dtype=torch.float)
return sample, label