Commit
·
85e2e23
1
Parent(s):
93348e5
Add Pretrained example
Browse files
README.md
CHANGED
|
@@ -50,6 +50,20 @@ pip install speechbrain
|
|
| 50 |
Please notice that we encourage you to read our tutorials and learn more about
|
| 51 |
[SpeechBrain](https://speechbrain.github.io).
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
## Referencing Mimic Loss
|
| 54 |
|
| 55 |
If you find mimic loss useful, please cite:
|
|
@@ -76,4 +90,4 @@ publisher = {GitHub},
|
|
| 76 |
journal = {GitHub repository},
|
| 77 |
howpublished = {\url{https://github.com/speechbrain/speechbrain}},
|
| 78 |
}
|
| 79 |
-
```
|
|
|
|
| 50 |
Please notice that we encourage you to read our tutorials and learn more about
|
| 51 |
[SpeechBrain](https://speechbrain.github.io).
|
| 52 |
|
| 53 |
+
## Pretrained Usage
|
| 54 |
+
|
| 55 |
+
To use the mimic-loss-trained model for enhancement, use the following simple code:
|
| 56 |
+
|
| 57 |
+
```python
|
| 58 |
+
from speechbrain.pretrained import SpectralMaskEnhancement
|
| 59 |
+
|
| 60 |
+
enhance_model = SpectralMaskEnhancement.from_hparams(
|
| 61 |
+
source="speechbrain/mtl-mimic-voicebank",
|
| 62 |
+
savedir="pretrained_models/mtl-mimic-voicebank",
|
| 63 |
+
)
|
| 64 |
+
enhance_model.enhance_file("/path/to/file.wav")
|
| 65 |
+
```
|
| 66 |
+
|
| 67 |
## Referencing Mimic Loss
|
| 68 |
|
| 69 |
If you find mimic loss useful, please cite:
|
|
|
|
| 90 |
journal = {GitHub repository},
|
| 91 |
howpublished = {\url{https://github.com/speechbrain/speechbrain}},
|
| 92 |
}
|
| 93 |
+
```
|