From 15c4709beb4b588db2135fc1133cd6955b5ef819 Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期一, 11 三月 2024 22:04:03 +0800 Subject: [PATCH] onnx (#1473) --- runtime/python/onnxruntime/funasr_onnx/vad_bin.py | 16 ++++++++-------- 1 files changed, 8 insertions(+), 8 deletions(-) diff --git a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py index 384f377..6b3a1bc 100644 --- a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py @@ -63,8 +63,8 @@ model = AutoModel(model=model_dir) model_dir = model.export(type="onnx", quantize=quantize) - config_file = os.path.join(model_dir, 'vad.yaml') - cmvn_file = os.path.join(model_dir, 'vad.mvn') + config_file = os.path.join(model_dir, 'config.yaml') + cmvn_file = os.path.join(model_dir, 'am.mvn') config = read_yaml(config_file) self.frontend = WavFrontend( @@ -73,8 +73,8 @@ ) self.ort_infer = OrtInferSession(model_file, device_id, intra_op_num_threads=intra_op_num_threads) self.batch_size = batch_size - self.vad_scorer = E2EVadModel(config["vad_post_conf"]) - self.max_end_sil = max_end_sil if max_end_sil is not None else config["vad_post_conf"]["max_end_silence_time"] + self.vad_scorer = E2EVadModel(config["model_conf"]) + self.max_end_sil = max_end_sil if max_end_sil is not None else config["model_conf"]["max_end_silence_time"] self.encoder_conf = config["encoder_conf"] def prepare_cache(self, in_cache: list = []): @@ -228,8 +228,8 @@ model = AutoModel(model=model_dir) model_dir = model.export(type="onnx", quantize=quantize) - config_file = os.path.join(model_dir, 'vad.yaml') - cmvn_file = os.path.join(model_dir, 'vad.mvn') + config_file = os.path.join(model_dir, 'config.yaml') + cmvn_file = os.path.join(model_dir, 'am.mvn') config = read_yaml(config_file) self.frontend = WavFrontendOnline( @@ -238,8 +238,8 @@ ) self.ort_infer = OrtInferSession(model_file, device_id, intra_op_num_threads=intra_op_num_threads) self.batch_size = batch_size - self.vad_scorer = E2EVadModel(config["vad_post_conf"]) - self.max_end_sil = max_end_sil if max_end_sil is not None else config["vad_post_conf"]["max_end_silence_time"] + self.vad_scorer = E2EVadModel(config["model_conf"]) + self.max_end_sil = max_end_sil if max_end_sil is not None else config["model_conf"]["max_end_silence_time"] self.encoder_conf = config["encoder_conf"] def prepare_cache(self, in_cache: list = []): -- Gitblit v1.9.1