Commit
•
ecd33f6
1
Parent(s):
adbfcb0
Update README.md
Browse files
README.md
CHANGED
|
@@ -38,16 +38,12 @@ Please notice that we encourage you to read our tutorials and learn more about [
|
|
| 38 |
### Perform source separation on your own audio file
|
| 39 |
|
| 40 |
```python
|
| 41 |
-
|
| 42 |
-
from speechbrain.pretrained import separator
|
| 43 |
import torchaudio
|
| 44 |
|
| 45 |
-
model = separator.from_hparams(source="speechbrain/sepformer-wham")
|
| 46 |
-
|
| 47 |
-
mix, fs = torchaudio.load("yourspeechbrainpath/samples/audio_samples/test_mixture.wav")
|
| 48 |
|
| 49 |
-
est_sources = model.
|
| 50 |
-
est_sources = est_sources / est_sources.max(dim=1, keepdim=True)[0]
|
| 51 |
|
| 52 |
torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
|
| 53 |
torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
|
|
|
|
| 38 |
### Perform source separation on your own audio file
|
| 39 |
|
| 40 |
```python
|
| 41 |
+
from speechbrain.pretrained import SepformerSeparation as separator
|
|
|
|
| 42 |
import torchaudio
|
| 43 |
|
| 44 |
+
model = separator.from_hparams(source="speechbrain/sepformer-wham", savedir='sepformer-wham')
|
|
|
|
|
|
|
| 45 |
|
| 46 |
+
est_sources = model.separate_file('your_speechbrain_path/samples/audio_samples/test_mixture.wav')
|
|
|
|
| 47 |
|
| 48 |
torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
|
| 49 |
torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
|