Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #47522264] kws pipeline returns Chinese charactor by configuration

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11534109
This commit is contained in:
bin.xue
2023-02-06 02:07:09 +00:00
committed by wenmeng.zwm
parent 0d3976920d
commit f0f07193a8
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modelscope/pipelines/audio/kws_farfield_pipeline.py
View File

@@ -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]: