Hugging Face's logo

Spaces:
qgyd2021
/
nx_denoise
Running

App Files Community
HoneyTian commited on
Commit
b8f2975
·
1 Parent(s): 0171728

update

Browse files
Files changed (1) hide show
  1. examples/spectrum_dfnet_aishell/step_3_evaluation.py +3 -2
examples/spectrum_dfnet_aishell/step_3_evaluation.py CHANGED
@@ -108,9 +108,10 @@ def enhance(mix_spec_complex: torch.Tensor,
108
  speech_spec = mix_spec_complex * mask_speech
109
  noise_spec = mix_spec_complex * mask_noise
110
 
111
- print(f"speech_spec_prediction: {speech_spec_prediction.shape}")
112
- print(f"noise_spec: {noise_spec.shape}")
113
 
 
114
  speech_wave = istft.forward(speech_spec_prediction)
115
  # speech_wave = istft.forward(speech_spec)
116
  noise_wave = istft.forward(noise_spec)
 
108
  speech_spec = mix_spec_complex * mask_speech
109
  noise_spec = mix_spec_complex * mask_noise
110
 
111
+ # print(f"speech_spec_prediction: {speech_spec_prediction.shape}")
112
+ # print(f"noise_spec: {noise_spec.shape}")
113
 
114
+ speech_spec_prediction = torch.view_as_complex(speech_spec_prediction)
115
  speech_wave = istft.forward(speech_spec_prediction)
116
  # speech_wave = istft.forward(speech_spec)
117
  noise_wave = istft.forward(noise_spec)