Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Feb 14

Commit

6e4b372

1 Parent(s): d41970b

update

Files changed (1) hide show

examples/spectrum_dfnet_aishell/step_3_evaluation.py CHANGED Viewed

@@ -105,7 +105,7 @@ def enhance(mix_spec_complex: torch.Tensor,
     mask_speech = speech_irm_prediction
     mask_noise = 1.0 - speech_irm_prediction
-    speech_spec = mix_spec_complex * mask_speech
     noise_spec = mix_spec_complex * mask_noise
     speech_wave = istft.forward(speech_spec_prediction)

     mask_speech = speech_irm_prediction
     mask_noise = 1.0 - speech_irm_prediction
+    # speech_spec = mix_spec_complex * mask_speech
     noise_spec = mix_spec_complex * mask_noise
     speech_wave = istft.forward(speech_spec_prediction)