Add model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
license: cc-by-nc-4.0
|
| 4 |
+
pipeline_tag: automatic-speech-recognition
|
| 5 |
+
tags:
|
| 6 |
+
- lip-reading
|
| 7 |
+
- visual-speech-recognition
|
| 8 |
+
- auto-avsr
|
| 9 |
+
- lrs3
|
| 10 |
+
- audio-visual
|
| 11 |
+
datasets:
|
| 12 |
+
- Ainncy/LRS3
|
| 13 |
+
metrics:
|
| 14 |
+
- wer
|
| 15 |
+
model-index:
|
| 16 |
+
- name: LRS3 Visual-Only Lip Reader
|
| 17 |
+
results: []
|
| 18 |
+
---
|
| 19 |
+
|
| 20 |
+
# LRS3 Visual-Only Lip Reading Model
|
| 21 |
+
|
| 22 |
+
**Visual-only English lip-reading model** based on the [Auto-AVSR](https://github.com/mpc001/auto_avsr) architecture
|
| 23 |
+
(commit `182b628`), fine-tuned on the Ainncy/LRS3 trainval split.
|
| 24 |
+
|
| 25 |
+
## Model Description
|
| 26 |
+
|
| 27 |
+
- **Architecture**: End-to-end (E2E) visual-only speech recognition, 250M parameters
|
| 28 |
+
- **Modality**: Video only (mouth ROI crops, no audio)
|
| 29 |
+
- **Frontend**: MediaPipe face detection + mouth ROI cropping at 16-second segments
|
| 30 |
+
- **Training**: 20 epochs on a single NVIDIA A100-SXM4-80GB, learning rate 0.001
|
| 31 |
+
- **Framework**: PyTorch 2.5.1 + PyTorch Lightning 2.5.0
|
| 32 |
+
- **Checkpoint**: `model_avg_10.pth` — average of the last 10 epoch checkpoints
|
| 33 |
+
|
| 34 |
+
## Dataset
|
| 35 |
+
|
| 36 |
+
Trained on the **supervised trainval split** from [Ainncy/LRS3](https://huggingface.co/datasets/Ainncy/LRS3):
|
| 37 |
+
|
| 38 |
+
- **Training samples**: ~98% of trainval (lrs3_train_fit.csv)
|
| 39 |
+
- **Validation samples**: ~2% of trainval (lrs3_train_val.csv)
|
| 40 |
+
- **Note**: The official LRS3 test set is **not** distributed by Ainncy/LRS3.
|
| 41 |
+
Validation metrics are computed on a deterministic held-out subset of trainval
|
| 42 |
+
and should **not** be compared to published LRS3 test-set results.
|
| 43 |
+
|
| 44 |
+
## Intended Use
|
| 45 |
+
|
| 46 |
+
This model is intended for **research purposes only** in visual speech recognition
|
| 47 |
+
(lip reading). It does **not** use audio input and is designed to explore the
|
| 48 |
+
limits of visual-only speech recognition.
|
| 49 |
+
|
| 50 |
+
## Limitations
|
| 51 |
+
|
| 52 |
+
- **Visual-only**: Performance is inherently lower than audio-visual or audio-only models.
|
| 53 |
+
- **English only**: Trained solely on English speech from LRS3 (TED/TEDx talks).
|
| 54 |
+
- **Controlled environment**: LRS3 consists of frontal-facing speakers with relatively
|
| 55 |
+
stable lighting. Real-world performance will be significantly lower.
|
| 56 |
+
- **No speaker diarization**: Cannot distinguish between multiple speakers.
|
| 57 |
+
- **Vocabulary constrained**: Limited to vocabulary present in the training data.
|
| 58 |
+
- **Privacy & ethics**: Lip-reading technology raises privacy concerns. This model
|
| 59 |
+
should not be used for surveillance or non-consensual speech recognition.
|
| 60 |
+
|
| 61 |
+
## Training Setup
|
| 62 |
+
|
| 63 |
+
Preprocessing was performed on Modal cloud GPUs using the pipeline in the
|
| 64 |
+
[accompanying repository](https://github.com/yoannjardin78/lrs3-lipreader):
|
| 65 |
+
|
| 66 |
+
1. Mouth ROI detection with MediaPipe (CPU, 8 parallel shards)
|
| 67 |
+
2. Video cropping to 16-second segments
|
| 68 |
+
3. Training on 1× A100-80GB for 20 epochs (~4h40)
|
| 69 |
+
|
| 70 |
+
The raw MP4 video files, crops, and transcriptions are **not** included in this
|
| 71 |
+
repository and must not be redistributed without explicit authorization from the
|
| 72 |
+
dataset copyright holders.
|
| 73 |
+
|
| 74 |
+
## Citation
|
| 75 |
+
|
| 76 |
+
```bibtex
|
| 77 |
+
@article{ma2023auto,
|
| 78 |
+
title={Auto-AVSR: Audio-Visual Speech Recognition with Automatic Labels},
|
| 79 |
+
author={Ma, Pingchuan and Haliassos, Alexandros and Fernandez-Lopez, Adriana
|
| 80 |
+
and Chen, Honglie and Petridis, Stavros and Pantic, Maja},
|
| 81 |
+
journal={arXiv preprint arXiv:2303.08807},
|
| 82 |
+
year={2023}
|
| 83 |
+
}
|
| 84 |
+
```
|
| 85 |
+
|
| 86 |
+
## License
|
| 87 |
+
|
| 88 |
+
This model is released under the Creative Commons Attribution-NonCommercial 4.0
|
| 89 |
+
International License (CC BY-NC 4.0), consistent with the LRS3 dataset terms.
|
| 90 |
+
Commercial use is prohibited.
|