def silero_lang_detector(**kwargs):
"""Silero Language Classifier
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/number_detector.jit')
utils = (get_language,
read_audio)
return model, utils
def silero_vad(**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.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
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/number_detector.jit')
utils = (get_number_ts, save_audio, read_audio, collect_chunks,
drop_chunks)
return model, utils
def silero_vad(onnx=False):
"""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()
if onnx:
model = OnnxWrapper(
f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.onnx')
else:
model = init_jit_model(
model_path=
f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.jit')
utils = (get_speech_timestamps, save_audio, read_audio, VADIterator,
collect_chunks)
return model, utils
def silero_lang_detector_116(**kwargs):
"""Silero Language Classifier (116 languages)
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/lang_classifier_116.jit')
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_116.json',
'r') as f:
lang_dict = json.load(f)
with open(
f'{hub_dir}/snakers4_silero-vad_master/files/lang_group_dict_116.json',
'r') as f:
lang_group_dict = json.load(f)
utils = (get_language_and_group, read_audio)
return model, lang_dict, lang_group_dict, utils
