From f0f07193a872bdf9902dc56cbe697c789c882e4e Mon Sep 17 00:00:00 2001 From: "bin.xue" Date: Mon, 6 Feb 2023 02:07:09 +0000 Subject: [PATCH] [to #47522264] kws pipeline returns Chinese charactor by configuration Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11534109 --- modelscope/pipelines/audio/kws_farfield_pipeline.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/modelscope/pipelines/audio/kws_farfield_pipeline.py b/modelscope/pipelines/audio/kws_farfield_pipeline.py index e2f618fa..5bfc31e9 100644 --- a/modelscope/pipelines/audio/kws_farfield_pipeline.py +++ b/modelscope/pipelines/audio/kws_farfield_pipeline.py @@ -40,6 +40,10 @@ class KWSFarfieldPipeline(Pipeline): self.model.eval() frame_size = self.INPUT_CHANNELS * self.SAMPLE_WIDTH self._nframe = self.model.size_in // frame_size + if 'keyword_map' in kwargs: + self._keyword_map = kwargs['keyword_map'] + else: + self._keyword_map = {} def preprocess(self, inputs: Input, **preprocess_params) -> Dict[str, Any]: if isinstance(inputs, bytes): @@ -85,6 +89,10 @@ class KWSFarfieldPipeline(Pipeline): fout.writeframes(result['pcm']) if 'kws' in result: result['kws']['offset'] += start_index / self.SAMPLE_RATE + result['kws']['type'] = 'wakeup' + keyword = result['kws']['keyword'] + if keyword in self._keyword_map: + result['kws']['keyword'] = self._keyword_map[keyword] kws_list.append(result['kws']) def postprocess(self, inputs: Dict[str, Any], **kwargs) -> Dict[str, Any]: