[egs] Fix wavfiles saving in eval.py for enh tasks (estimates)

someoneformulated · Jun 4, 2020 · ce5c345 · ce5c345
1 parent aa4b0aa
commit ce5c345
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/egs/kinect-wsj/DeepClustering/eval.py b/egs/kinect-wsj/DeepClustering/eval.py
@@ -65,7 +65,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'],
                                   metrics_list=compute_metrics)

diff --git a/egs/librimix/ConvTasNet/eval.py b/egs/librimix/ConvTasNet/eval.py
@@ -70,7 +70,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix.cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         # For each utterance, we get a dictionary with the mixture path,
         # the input and output metrics
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,

diff --git a/egs/wham/ConvTasNet/eval.py b/egs/wham/ConvTasNet/eval.py
@@ -62,7 +62,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'])
         utt_metrics['mix_path'] = test_set.mix[idx][0]

diff --git a/egs/wham/DPRNN/eval.py b/egs/wham/DPRNN/eval.py
@@ -62,7 +62,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'])
         utt_metrics['mix_path'] = test_set.mix[idx][0]

diff --git a/egs/wham/DynamicMixing/eval.py b/egs/wham/DynamicMixing/eval.py
@@ -60,7 +60,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'])
         utt_metrics['mix_path'] = test_set.mix[idx][0]

diff --git a/egs/wham/TwoStep/eval.py b/egs/wham/TwoStep/eval.py
@@ -64,7 +64,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'])
         utt_metrics['mix_path'] = test_set.mix[idx][0]

diff --git a/egs/whamr/TasNet/eval.py b/egs/whamr/TasNet/eval.py
@@ -63,7 +63,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'])
         utt_metrics['mix_path'] = test_set.mix[idx][0]

diff --git a/egs/wsj0-mix/DeepClustering/eval.py b/egs/wsj0-mix/DeepClustering/eval.py
@@ -62,7 +62,7 @@ def main(conf):
                                             return_est=True)
         mix_np = mix[None].cpu().data.numpy()
         sources_np = sources.cpu().data.numpy()
-        est_sources_np = reordered_sources.squeeze().cpu().data.numpy()
+        est_sources_np = reordered_sources.squeeze(0).cpu().data.numpy()
         utt_metrics = get_metrics(mix_np, sources_np, est_sources_np,
                                   sample_rate=conf['sample_rate'],
                                   metrics_list=compute_metrics)