From 9285e7d5998e2e7b6260dd0abd1868432f9e6f52 Mon Sep 17 00:00:00 2001 From: "shixian.shi" Date: Mon, 22 Jan 2024 16:22:49 +0800 Subject: [PATCH] update speaker infer --- funasr/auto/auto_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py index ca6189dd2..21d8e1dd1 100644 --- a/funasr/auto/auto_model.py +++ b/funasr/auto/auto_model.py @@ -6,6 +6,7 @@ import string import logging import os.path +import numpy as np from tqdm import tqdm from omegaconf import DictConfig, OmegaConf, ListConfig @@ -334,7 +335,7 @@ def inference_with_vad(self, input, input_len=None, **cfg): for _b in range(len(speech_j)): vad_segments = [[sorted_data[beg_idx:end_idx][_b][0][0]/1000.0, sorted_data[beg_idx:end_idx][_b][0][1]/1000.0, - speech_j[_b]]] + np.array(speech_j[_b])]] segments = sv_chunk(vad_segments) all_segments.extend(segments) speech_b = [i[2] for i in segments]