unlinken of already processed training files (seed dependent audio augmentation)

This commit is contained in:
Si11ium 2020-05-15 11:13:14 +02:00
parent 9ae9143544
commit 4a6b32d1dd

View File

@ -58,7 +58,11 @@ class BinaryMasksDataset(Dataset):
# Delete File if one exists. # Delete File if one exists.
if not self.use_preprocessed: if not self.use_preprocessed:
for key in labeldict.keys(): for key in labeldict.keys():
(self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink(missing_ok=True) try:
(self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink()
except FileNotFoundError:
pass
return labeldict return labeldict
def __len__(self): def __len__(self):