Spaces:
Sleeping
Sleeping
Upload f5_tts/eval/README.md with huggingface_hub
Browse files- f5_tts/eval/README.md +49 -0
f5_tts/eval/README.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
# Evaluation
|
| 3 |
+
|
| 4 |
+
Install packages for evaluation:
|
| 5 |
+
|
| 6 |
+
```bash
|
| 7 |
+
pip install -e .[eval]
|
| 8 |
+
```
|
| 9 |
+
|
| 10 |
+
## Generating Samples for Evaluation
|
| 11 |
+
|
| 12 |
+
### Prepare Test Datasets
|
| 13 |
+
|
| 14 |
+
1. *Seed-TTS testset*: Download from [seed-tts-eval](https://github.com/BytedanceSpeech/seed-tts-eval).
|
| 15 |
+
2. *LibriSpeech test-clean*: Download from [OpenSLR](http://www.openslr.org/12/).
|
| 16 |
+
3. Unzip the downloaded datasets and place them in the `data/` directory.
|
| 17 |
+
4. Update the path for *LibriSpeech test-clean* data in `src/f5_tts/eval/eval_infer_batch.py`
|
| 18 |
+
5. Our filtered LibriSpeech-PC 4-10s subset: `data/librispeech_pc_test_clean_cross_sentence.lst`
|
| 19 |
+
|
| 20 |
+
### Batch Inference for Test Set
|
| 21 |
+
|
| 22 |
+
To run batch inference for evaluations, execute the following commands:
|
| 23 |
+
|
| 24 |
+
```bash
|
| 25 |
+
# batch inference for evaluations
|
| 26 |
+
accelerate config # if not set before
|
| 27 |
+
bash src/f5_tts/eval/eval_infer_batch.sh
|
| 28 |
+
```
|
| 29 |
+
|
| 30 |
+
## Objective Evaluation on Generated Results
|
| 31 |
+
|
| 32 |
+
### Download Evaluation Model Checkpoints
|
| 33 |
+
|
| 34 |
+
1. Chinese ASR Model: [Paraformer-zh](https://huggingface.co/funasr/paraformer-zh)
|
| 35 |
+
2. English ASR Model: [Faster-Whisper](https://huggingface.co/Systran/faster-whisper-large-v3)
|
| 36 |
+
3. WavLM Model: Download from [Google Drive](https://drive.google.com/file/d/1-aE1NfzpRCLxA4GUxX9ITI3F9LlbtEGP/view).
|
| 37 |
+
|
| 38 |
+
Then update in the following scripts with the paths you put evaluation model ckpts to.
|
| 39 |
+
|
| 40 |
+
### Objective Evaluation
|
| 41 |
+
|
| 42 |
+
Update the path with your batch-inferenced results, and carry out WER / SIM evaluations:
|
| 43 |
+
```bash
|
| 44 |
+
# Evaluation for Seed-TTS test set
|
| 45 |
+
python src/f5_tts/eval/eval_seedtts_testset.py --gen_wav_dir <GEN_WAVE_DIR>
|
| 46 |
+
|
| 47 |
+
# Evaluation for LibriSpeech-PC test-clean (cross-sentence)
|
| 48 |
+
python src/f5_tts/eval/eval_librispeech_test_clean.py --gen_wav_dir <GEN_WAVE_DIR> --librispeech_test_clean_path <TEST_CLEAN_PATH>
|
| 49 |
+
```
|