From 9ea067966ddd399751097fb7b33c3a412fde9fbf Mon Sep 17 00:00:00 2001
From: jrichter <jrichter@exchange.informatik.uni-hamburg.de>
Date: Wed, 27 Sep 2023 10:18:40 +0200
Subject: [PATCH] add file naming format for WSJ0_REVERB
---
sgmse/data_module.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/sgmse/data_module.py b/sgmse/data_module.py
index e1c1c6c..0eb4535 100644
--- a/sgmse/data_module.py
+++ b/sgmse/data_module.py
@@ -28,6 +28,9 @@ class Specs(Dataset):
if format == "default":
self.clean_files = sorted(glob(join(data_dir, subset) + '/clean/*.wav'))
self.noisy_files = sorted(glob(join(data_dir, subset) + '/noisy/*.wav'))
+ elif format == "reverb":
+ self.clean_files = sorted(glob(join(data_dir, subset) + '/anechoic/*.wav'))
+ self.noisy_files = sorted(glob(join(data_dir, subset) + '/reverb/*.wav'))
else:
# Feel free to add your own directory format
raise NotImplementedError(f"Directory format {format} unknown!")
@@ -93,7 +96,7 @@ class SpecsDataModule(pl.LightningDataModule):
@staticmethod
def add_argparse_args(parser):
parser.add_argument("--base_dir", type=str, required=True, help="The base directory of the dataset. Should contain `train`, `valid` and `test` subdirectories, each of which contain `clean` and `noisy` subdirectories.")
- parser.add_argument("--format", type=str, choices=("default", "dns"), default="default", help="Read file paths according to file naming format.")
+ parser.add_argument("--format", type=str, choices=("default", "reverb"), default="default", help="Read file paths according to file naming format.")
parser.add_argument("--batch_size", type=int, default=8, help="The batch size. 8 by default.")
parser.add_argument("--n_fft", type=int, default=510, help="Number of FFT bins. 510 by default.") # to assure 256 freq bins
parser.add_argument("--hop_length", type=int, default=128, help="Window hop length. 128 by default.")
--
GitLab