From 572aab822184ecdb9cbacc99aa664b0b17da4e3f Mon Sep 17 00:00:00 2001 From: adamnsandle Date: Mon, 22 Mar 2021 13:30:02 +0000 Subject: [PATCH] add VAD micro 8k to hubconf --- README.md | 2 ++ hubconf.py | 17 +++++++++++++++++ 2 files changed, 19 insertions(+) diff --git a/README.md b/README.md index 319fab8..8923597 100644 --- a/README.md +++ b/README.md @@ -61,6 +61,7 @@ The models are small enough to be included directly into this repository. Newer |--------------------------------|--------|---------------------|--------------------|----------------|---------|------|-------| | `'silero_vad'` | 1.1M | VAD | Yes | `ru`, `en`, `de`, `es` (*) | :heavy_check_mark: | :heavy_check_mark: | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/snakers4/silero-vad/blob/master/silero-vad.ipynb) | | `'silero_vad_micro'` | 10K | VAD | Yes | `ru`, `en`, `de`, `es` (*) | :heavy_check_mark: | :heavy_check_mark: | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/snakers4/silero-vad/blob/master/silero-vad.ipynb) | +| `'silero_vad_micro_8k'` | 10K | VAD | Yes | `ru`, `en`, `de`, `es` (*) | :heavy_check_mark: | :heavy_check_mark: | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/snakers4/silero-vad/blob/master/silero-vad.ipynb) | | `'silero_number_detector'` | 1.1M | Number Detector | No | `ru`, `en`, `de`, `es` | :heavy_check_mark: | :heavy_check_mark: | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/snakers4/silero-vad/blob/master/silero-vad.ipynb) | | `'silero_lang_detector'` | 1.1M | Language Classifier | No | `ru`, `en`, `de`, `es` | :heavy_check_mark: | :heavy_check_mark: | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/snakers4/silero-vad/blob/master/silero-vad.ipynb) | @@ -81,6 +82,7 @@ What models do: | `v1.2` | 2020-12-30 | Number Detector added | `v2` | 2021-01-11 | Add Language Classifier heads (en, ru, de, es) | | `v2.1` | 2021-02-11 | Add micro (10k params) VAD models | +| `v2.2` | 2021-03-22 | Add micro 8000 sample rate VAD models | ### PyTorch diff --git a/hubconf.py b/hubconf.py index 5a515ca..64a8c0a 100644 --- a/hubconf.py +++ b/hubconf.py @@ -46,6 +46,23 @@ def silero_vad_micro(**kwargs): return model, utils +def silero_vad_micro_8k(**kwargs): + """Silero Voice Activity Detector + Returns a model with a set of utils + Please see https://github.com/snakers4/silero-vad for usage examples + """ + hub_dir = torch.hub.get_dir() + model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/model_micro_8k.jit') + utils = (get_speech_ts, + save_audio, + read_audio, + state_generator, + single_audio_stream, + collect_chunks) + + return model, utils + + def silero_number_detector(**kwargs): """Silero Number Detector Returns a model with a set of utils