Commit
·
19e48e6
1
Parent(s):
378e4ce
Update README.md
Browse files
README.md
CHANGED
@@ -20,11 +20,11 @@ metrics:
|
|
20 |
|
21 |
This repository provides all the necessary tools to perform source separation with a [SepFormer](https://arxiv.org/abs/2010.13154v2)
|
22 |
model, implemented with SpeechBrain, and pretrained on WSJ0-2Mix dataset. For a better experience we encourage you to learn more about
|
23 |
-
[SpeechBrain](https://speechbrain.github.io). The given model performance is 22.
|
24 |
|
25 |
| Release | Test-Set SI-SNR | Test-Set SDR |
|
26 |
|:-------------:|:--------------:|:--------------:|
|
27 |
-
| 09-03-21 | 22.
|
28 |
|
29 |
|
30 |
## Install SpeechBrain
|
@@ -47,7 +47,7 @@ import torchaudio
|
|
47 |
|
48 |
model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix")
|
49 |
|
50 |
-
mix, fs = torchaudio.load("test_mixture.wav")
|
51 |
|
52 |
est_sources = model.separate(mix)
|
53 |
est_sources = est_sources / est_sources.max(dim=1, keepdim=True)[0]
|
|
|
20 |
|
21 |
This repository provides all the necessary tools to perform source separation with a [SepFormer](https://arxiv.org/abs/2010.13154v2)
|
22 |
model, implemented with SpeechBrain, and pretrained on WSJ0-2Mix dataset. For a better experience we encourage you to learn more about
|
23 |
+
[SpeechBrain](https://speechbrain.github.io). The given model performance is 22.4 dB on the test set of WSJ0-2Mix dataset.
|
24 |
|
25 |
| Release | Test-Set SI-SNR | Test-Set SDR |
|
26 |
|:-------------:|:--------------:|:--------------:|
|
27 |
+
| 09-03-21 | 22.4dB | 22.6dB |
|
28 |
|
29 |
|
30 |
## Install SpeechBrain
|
|
|
47 |
|
48 |
model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix")
|
49 |
|
50 |
+
mix, fs = torchaudio.load("samples/audio_samples/test_mixture.wav")
|
51 |
|
52 |
est_sources = model.separate(mix)
|
53 |
est_sources = est_sources / est_sources.max(dim=1, keepdim=True)[0]
|