sp-uhh / sgmse

Score-based Generative Models (Diffusion Models) for Speech Enhancement and Dereverberation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Dereverberation not work

loboere opened this issue · comments

I run
!python enhancement.py --test_dir "/content/test" --enhanced_dir "/content/na" --ckpt "/content/drive/MyDrive/SGMSE+ Dereverberation Checkpoints/epoch=326-step=408750.ckpt"

but the results are not the same as in the demo page ,the sound comes out saturated

Hi, sorry about this. We had accidentally removed some options for the network that we thought were unused. It should be fixed on the most recent commit, c2641a9 . I checked both SE and Dereverberation again.

Note that changing the sampler settings slightly seems to work better for dereverberation, using --N 50 --snr 0.33 (I'll also add this info to the README).

Let me know if it works or if something's still broken for you.

Closing this now due to inactivity. Feel free to open another issue if something is still broken for you :)