Skip to content
Snippets Groups Projects
Commit 03a1c8a3 authored by jrichter's avatar jrichter
Browse files

adapt data_module for the ears-wham dataset

parent eeb0386d
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 12 additions and 4 deletions
sgmse/data_module.py
+ 12
− 4
View file @ 03a1c8a3
...@@ -26,11 +26,19 @@ class Specs(Dataset): ...@@ -26,11 +26,19 @@ class Specs(Dataset):
# Read file paths according to file naming format. # Read file paths according to file naming format.
if format == "default": if format == "default":
self.clean_files = sorted(glob(join(data_dir, subset) + '/clean/*.wav')) self.clean_files = []
self.noisy_files = sorted(glob(join(data_dir, subset) + '/noisy/*.wav')) self.clean_files += sorted(glob(join(data_dir, subset, "clean", "*.wav")))
self.clean_files += sorted(glob(join(data_dir, subset, "clean", "**", "*.wav")))
self.noisy_files = []
self.noisy_files += sorted(glob(join(data_dir, subset, "noisy", "*.wav")))
self.noisy_files += sorted(glob(join(data_dir, subset, "noisy", "**", "*.wav")))
elif format == "reverb": elif format == "reverb":
self.clean_files = sorted(glob(join(data_dir, subset) + '/anechoic/*.wav')) self.clean_files = []
self.noisy_files = sorted(glob(join(data_dir, subset) + '/reverb/*.wav')) self.clean_files += sorted(glob(join(data_dir, subset, "anechoic", "*.wav")))
self.clean_files += sorted(glob(join(data_dir, subset, "anechoic", "**", "*.wav")))
self.noisy_files = []
self.noisy_files += sorted(glob(join(data_dir, subset, "reverb", "*.wav")))
self.noisy_files += sorted(glob(join(data_dir, subset, "reverb", "**", "*.wav")))
else: else:
# Feel free to add your own directory format # Feel free to add your own directory format
raise NotImplementedError(f"Directory format {format} unknown!") raise NotImplementedError(f"Directory format {format} unknown!")
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment