diff --git a/hpc_entrypoint.sh b/hpc_entrypoint.sh index 5300dbb..1d5c602 100644 --- a/hpc_entrypoint.sh +++ b/hpc_entrypoint.sh @@ -36,7 +36,7 @@ pwd #python transcriber/tasks/embeddings/timit.py --directory /scratch/$USER/TIMIT/data/lisa/data/timit/raw/TIMIT/TEST --output ./data/test python noisyspeech_synthesizer.py echo "files" -mkdir /scratch/c.sistc3/MS-SNSD/DNS10 -mv ./CleanSpeech_training/ /scratch/c.sistc3/MS-SNSD/DNS10 -mv ./NoisySpeech_training/ /scratch/c.sistc3/MS-SNSD/DNS10 +#mkdir /scratch/c.sistc3/MS-SNSD/DNS10 +mv ./CleanSpeech_testing/ /scratch/c.sistc3/MS-SNSD/DNS10 +mv ./NoisySpeech_testing/ /scratch/c.sistc3/MS-SNSD/DNS10 #python enhancer/cli/train.py diff --git a/noisyspeech_synthesizer.cfg b/noisyspeech_synthesizer.cfg index f382a09..3054509 100644 --- a/noisyspeech_synthesizer.cfg +++ b/noisyspeech_synthesizer.cfg @@ -19,11 +19,12 @@ sampling_rate: 16000 audioformat: *.wav audio_length: 10 silence_length: 0.2 -total_hours: 10 +total_hours: 0.5 snr_lower: 0 snr_upper: 40 total_snrlevels: 5 +naming: test -noise_dir: /scratch/c.sistc3/MS-SNSD/noise_train -speech_dir: /scratch/c.sistc3/MS-SNSD/clean_train +noise_dir: /scratch/c.sistc3/MS-SNSD/noise_test +speech_dir: /scratch/c.sistc3/MS-SNSD/clean_test noise_types_excluded: None diff --git a/noisyspeech_synthesizer.py b/noisyspeech_synthesizer.py index 00e9b62..167f93a 100644 --- a/noisyspeech_synthesizer.py +++ b/noisyspeech_synthesizer.py @@ -27,23 +27,23 @@ def main(cfg): noise_dir = cfg["noise_dir"] if not os.path.exists(noise_dir): assert False, "Noise data is required" - + name = cfg["naming"] fs = float(cfg["sampling_rate"]) audioformat = cfg["audioformat"] total_hours = float(cfg["total_hours"]) audio_length = float(cfg["audio_length"]) silence_length = float(cfg["silence_length"]) noisyspeech_dir = os.path.join( - os.path.dirname(__file__), "NoisySpeech_training" + os.path.dirname(__file__), f"NoisySpeech_{name}ing" ) if not os.path.exists(noisyspeech_dir): os.makedirs(noisyspeech_dir) clean_proc_dir = os.path.join( - os.path.dirname(__file__), "CleanSpeech_training" + os.path.dirname(__file__), f"CleanSpeech_{name}ing" ) if not os.path.exists(clean_proc_dir): os.makedirs(clean_proc_dir) - noise_proc_dir = os.path.join(os.path.dirname(__file__), "Noise_training") + noise_proc_dir = os.path.join(os.path.dirname(__file__), f"Noise_{name}ing") if not os.path.exists(noise_proc_dir): os.makedirs(noise_proc_dir)