Upload EBENGenerator
Browse files- README.md +50 -0
- config.json +5 -0
- model.safetensors +3 -0
README.md
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: fr
|
| 3 |
+
license: mit
|
| 4 |
+
library_name: transformers
|
| 5 |
+
tags:
|
| 6 |
+
- audio
|
| 7 |
+
- audio-to-audio
|
| 8 |
+
- speech
|
| 9 |
+
datasets:
|
| 10 |
+
- Cnam-LMSSC/vibravox
|
| 11 |
+
---
|
| 12 |
+
# Model Card
|
| 13 |
+
|
| 14 |
+
- **Developed by:** [Cnam-LMSSC](https://huggingface.co/Cnam-LMSSC)
|
| 15 |
+
- **Model type:** [EBEN](https://github.com/jhauret/vibravox/blob/main/vibravox/torch_modules/dnn/eben_generator.py) (see [publication](https://ieeexplore.ieee.org/document/10244161))
|
| 16 |
+
- **Language:** French
|
| 17 |
+
- **License:** MIT
|
| 18 |
+
- **Finetuned dataset:** `speech_clean` subset of [Cnam-LMSSC/vibravox](https://huggingface.co/datasets/Cnam-LMSSC/vibravox)
|
| 19 |
+
- **Samplerate for usage:** 16kHz
|
| 20 |
+
|
| 21 |
+
## Overview
|
| 22 |
+
|
| 23 |
+
This bandwidth extension model is trained on one specific body conduction sensor data from the [Vibravox dataset](https://huggingface.co/datasets/Cnam-LMSSC/vibravox).
|
| 24 |
+
The model is designed to to enhance the audio quality of body-conducted captured speech, by denoising and regenerating mid and high frequencies from low frequency content only.
|
| 25 |
+
|
| 26 |
+
## Disclaimer
|
| 27 |
+
This model has been trained for **specific non-conventional speech sensors** and is intended to be used with **in-domain data**.
|
| 28 |
+
Please be advised that using these models outside their intended sensor data may result in suboptimal performance.
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
## Training procedure
|
| 32 |
+
|
| 33 |
+
Detailed instructions for reproducing the experiments are available on the [jhauret/vibravox](https://github.com/jhauret/vibravox) Github repository.
|
| 34 |
+
|
| 35 |
+
## Inference script :
|
| 36 |
+
|
| 37 |
+
```python
|
| 38 |
+
import torch, torchaudio
|
| 39 |
+
from vibravox import EBENGenerator
|
| 40 |
+
from datasets import load_dataset
|
| 41 |
+
|
| 42 |
+
audio_16kHz, _ = torch.load("path_to_audio")
|
| 43 |
+
|
| 44 |
+
cut_audio_16kHz = model.cut_to_valid_length(audio_16kHz)
|
| 45 |
+
enhanced_audio_16kHz = model(cut_audio_16kHz)
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
## Link to other BWE models trained on other body conducted sensors :
|
| 49 |
+
|
| 50 |
+
An entry point to all **audio bandwidth extension** (BWE) models trained on different sensor data from the trained on different sensor data from the [Vibravox dataset](https://huggingface.co/datasets/Cnam-LMSSC/vibravox) is available at [https://huggingface.co/Cnam-LMSSC/vibravox_EBEN_bwe_models](https://huggingface.co/Cnam-LMSSC/vibravox_EBEN_bwe_models).
|
config.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"m": 4,
|
| 3 |
+
"n": 32,
|
| 4 |
+
"p": 4
|
| 5 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cd11cc4c34c2685caad71c8bef2161f33b2714bd2c2fc6d148e80f01838758f2
|
| 3 |
+
size 7798600
|