|
|
@@ -18,15 +18,18 @@ source sgmse_env/bin/activate # 在 Windows 上使用 sgmse_env\Scripts\activat
|
|
|
|
|
|
pip install torch librosa soundfile tqdm pesq pandas pytorch_lightning wandb torchaudio pystoi
|
|
|
|
|
|
-
|
|
|
+# 划分数据
|
|
|
python preprocessing/create_wsj0_chime3.py data/wsj0 data/CHiME3/data/audio/16kHz target
|
|
|
|
|
|
python preprocessing/create_wsj0_qut.py wsj0 qut target
|
|
|
|
|
|
+# 模型训练
|
|
|
python train.py --base_dir target
|
|
|
|
|
|
+# 模型评估
|
|
|
python enhancement.py --test_dir test_data --enhanced_dir enhanced_data --ckpt model_checkpoint
|
|
|
|
|
|
+# 计算评估指标：
|
|
|
python calc_metrics.py --clean_dir clean_data --noisy_dir noisy_data --enhanced_dir enhanced_data
|
|
|
|
|
|
|
