Spaces:
Running
Running
update
Browse files
examples/spectrum_dfnet_aishell/step_3_evaluation.py
CHANGED
@@ -105,6 +105,9 @@ def enhance(mix_spec_complex: torch.Tensor,
|
|
105 |
mask_speech = speech_irm_prediction
|
106 |
mask_noise = 1.0 - speech_irm_prediction
|
107 |
|
|
|
|
|
|
|
108 |
# speech_spec = mix_spec_complex * mask_speech
|
109 |
noise_spec = mix_spec_complex * mask_noise
|
110 |
|
|
|
105 |
mask_speech = speech_irm_prediction
|
106 |
mask_noise = 1.0 - speech_irm_prediction
|
107 |
|
108 |
+
print(f"mix_spec_complex: {mix_spec_complex.shape}")
|
109 |
+
print(f"mask_noise: {mask_noise.shape}")
|
110 |
+
|
111 |
# speech_spec = mix_spec_complex * mask_speech
|
112 |
noise_spec = mix_spec_complex * mask_noise
|
113 |
|