Andreas Nautsch commited on
Commit
80af4b0
1 Parent(s): 5c3c8db

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +92 -0
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: "de"
3
+ thumbnail:
4
+ tags:
5
+ - automatic-speech-recognition
6
+ - CTC
7
+ - Attention
8
+ - pytorch
9
+ - speechbrain
10
+ license: "apache-2.0"
11
+ datasets:
12
+ - common_voice
13
+ metrics:
14
+ - wer
15
+ - cer
16
+ ---
17
+ <iframe src="https://ghbtns.com/github-btn.html?user=speechbrain&repo=speechbrain&type=star&count=true&size=large&v=2" frameborder="0" scrolling="0" width="170" height="30" title="GitHub"></iframe>
18
+ <br/><br/>
19
+ # CRDNN with CTC/Attention trained on CommonVoice 7.0 German (No LM)
20
+ This repository provides all the necessary tools to perform automatic speech
21
+ recognition from an end-to-end system pretrained on CommonVoice (German Language) within
22
+ SpeechBrain. For a better experience, we encourage you to learn more about
23
+ [SpeechBrain](https://speechbrain.github.io).
24
+ The performance of the model is the following:
25
+ | Release | Test CER | Test WER | GPUs |
26
+ |:-------------:|:--------------:|:--------------:| :--------:|
27
+ | 28-10-21 | 4.93 | 15.37 | 1xV100 16GB |
28
+ ## Credits
29
+ The model is provided by [vitas.ai](vitas.ai).
30
+ ## Pipeline description
31
+ This ASR system is composed of 2 different but linked blocks:
32
+ - Tokenizer (unigram) that transforms words into subword units and trained with
33
+ the train transcriptions (train.tsv) of CommonVoice (DE).
34
+ - Acoustic model (CRDNN + CTC/Attention). The CRDNN architecture is made of
35
+ N blocks of convolutional neural networks with normalization and pooling on the
36
+ frequency domain. Then, a bidirectional LSTM is connected to a final DNN to obtain
37
+ the final acoustic representation that is given to the CTC and attention decoders.
38
+ ## Install SpeechBrain
39
+ First of all, please install SpeechBrain with the following command:
40
+ ```
41
+ pip install speechbrain
42
+ ```
43
+ Please notice that we encourage you to read our tutorials and learn more about
44
+ [SpeechBrain](https://speechbrain.github.io).
45
+ ### Transcribing your own audio files (in German)
46
+ ```python
47
+ from speechbrain.pretrained import EncoderDecoderASR
48
+ asr_model = EncoderDecoderASR.from_hparams(source="speechbrain/asr-crdnn-commonvoice-de", savedir="pretrained_models/asr-crdnn-commonvoice-de")
49
+ asr_model.transcribe_file("speechbrain/asr-crdnn-commonvoice-de/example-de.wav")
50
+ ```
51
+ ### Inference on GPU
52
+ To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
53
+ ## Parallel Inference on a Batch
54
+ Please, [see this Colab notebook](https://colab.research.google.com/drive/1hX5ZI9S4jHIjahFCZnhwwQmFoGAi3tmu?usp=sharing) to figure out how to transcribe in parallel a batch of input sentences using a pre-trained model.
55
+ ### Training
56
+ The model was trained with SpeechBrain (986a2175).
57
+ To train it from scratch follows these steps:
58
+ 1. Clone SpeechBrain:
59
+ ```bash
60
+ git clone https://github.com/speechbrain/speechbrain/
61
+ ```
62
+ 2. Install it:
63
+ ```
64
+ cd speechbrain
65
+ pip install -r requirements.txt
66
+ pip install -e .
67
+ ```
68
+ 3. Run Training:
69
+ ```
70
+ cd recipes/CommonVoice/ASR/seq2seq
71
+ python train.py hparams/train_de.yaml --data_folder=your_data_folder
72
+ ```
73
+ You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/13i7rdgVX7-qZ94Rtj6OdUgU-S6BbKKvw?usp=sharing)
74
+ ### Limitations
75
+ The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
76
+ # **About SpeechBrain**
77
+ - Website: https://speechbrain.github.io/
78
+ - Code: https://github.com/speechbrain/speechbrain/
79
+ - HuggingFace: https://huggingface.co/speechbrain/
80
+ # **Citing SpeechBrain**
81
+ Please, cite SpeechBrain if you use it for your research or business.
82
+ ```bibtex
83
+ @misc{speechbrain,
84
+ title={{SpeechBrain}: A General-Purpose Speech Toolkit},
85
+ author={Mirco Ravanelli and Titouan Parcollet and Peter Plantinga and Aku Rouhe and Samuele Cornell and Loren Lugosch and Cem Subakan and Nauman Dawalatabad and Abdelwahab Heba and Jianyuan Zhong and Ju-Chieh Chou and Sung-Lin Yeh and Szu-Wei Fu and Chien-Feng Liao and Elena Rastorgueva and François Grondin and William Aris and Hwidong Na and Yan Gao and Renato De Mori and Yoshua Bengio},
86
+ year={2021},
87
+ eprint={2106.04624},
88
+ archivePrefix={arXiv},
89
+ primaryClass={eess.AS},
90
+ note={arXiv:2106.04624}
91
+ }
92
+ ```