BandwiseBinaryClassifier is work in progress; TODO: Shape Piping.

2020-05-04 18:45:13 +02:00
parent e4f6506a4b
commit 451f78f820
7 changed files with 190 additions and 42 deletions
--- a/datasets/binar_masks.py
+++ b/datasets/binar_masks.py
@ -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 / 'raw_mel'
+        self._mel_folder = self.data_root / 'transformed'

    def _build_labels(self):
        with open(Path(self.data_root) / 'lab' / 'labels.csv', mode='r') as f: