mirror of
https://github.com/snakers4/silero-vad.git
synced 2026-02-05 01:49:22 +08:00
Compare commits
1 Commits
v6.1
...
v3.1stable
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f9876dd5f9 |
13
hubconf.py
13
hubconf.py
@@ -1,5 +1,6 @@
|
|||||||
dependencies = ['torch', 'torchaudio']
|
dependencies = ['torch', 'torchaudio']
|
||||||
import torch
|
import torch
|
||||||
|
import os
|
||||||
import json
|
import json
|
||||||
from utils_vad import (init_jit_model,
|
from utils_vad import (init_jit_model,
|
||||||
get_speech_timestamps,
|
get_speech_timestamps,
|
||||||
@@ -20,11 +21,11 @@ def silero_vad(onnx=False):
|
|||||||
Returns a model with a set of utils
|
Returns a model with a set of utils
|
||||||
Please see https://github.com/snakers4/silero-vad for usage examples
|
Please see https://github.com/snakers4/silero-vad for usage examples
|
||||||
"""
|
"""
|
||||||
hub_dir = torch.hub.get_dir()
|
model_dir = os.path.join(os.path.dirname(__file__), 'files')
|
||||||
if onnx:
|
if onnx:
|
||||||
model = OnnxWrapper(f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.onnx')
|
model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx'))
|
||||||
else:
|
else:
|
||||||
model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.jit')
|
model = init_jit_model(os.path.join(model_dir, 'silero_vad.jit'))
|
||||||
utils = (get_speech_timestamps,
|
utils = (get_speech_timestamps,
|
||||||
save_audio,
|
save_audio,
|
||||||
read_audio,
|
read_audio,
|
||||||
@@ -75,17 +76,17 @@ def silero_lang_detector_95(onnx=False):
|
|||||||
Please see https://github.com/snakers4/silero-vad for usage examples
|
Please see https://github.com/snakers4/silero-vad for usage examples
|
||||||
"""
|
"""
|
||||||
|
|
||||||
hub_dir = torch.hub.get_dir()
|
|
||||||
if onnx:
|
if onnx:
|
||||||
url = 'https://models.silero.ai/vad_models/lang_classifier_95.onnx'
|
url = 'https://models.silero.ai/vad_models/lang_classifier_95.onnx'
|
||||||
else:
|
else:
|
||||||
url = 'https://models.silero.ai/vad_models/lang_classifier_95.jit'
|
url = 'https://models.silero.ai/vad_models/lang_classifier_95.jit'
|
||||||
model = Validator(url)
|
model = Validator(url)
|
||||||
|
|
||||||
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_95.json', 'r') as f:
|
model_dir = os.path.join(os.path.dirname(__file__), 'files')
|
||||||
|
with open(os.path.join(model_dir, 'lang_dict_95.json'), 'r') as f:
|
||||||
lang_dict = json.load(f)
|
lang_dict = json.load(f)
|
||||||
|
|
||||||
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_group_dict_95.json', 'r') as f:
|
with open(os.path.join(model_dir, 'lang_group_dict_95.json'), 'r') as f:
|
||||||
lang_group_dict = json.load(f)
|
lang_group_dict = json.load(f)
|
||||||
|
|
||||||
utils = (get_language_and_group, read_audio)
|
utils = (get_language_and_group, read_audio)
|
||||||
|
|||||||
Reference in New Issue
Block a user