Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] Fix remove useless Tasks.key_word_spotting

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9303027
This commit is contained in:
shichen.fsc
2022-07-07 17:02:01 +08:00
committed by yingda.chen
parent 9f1ad5da80
commit 39e8e97033
4 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/models/audio/kws/generic_key_word_spotting.py
View File

@@ -9,7 +9,8 @@ from modelscope.utils.constant import Tasks
__all__ = ['GenericKeyWordSpotting']
@MODELS.register_module(Tasks.key_word_spotting, module_name=Models.kws_kwsbp)
@MODELS.register_module(
Tasks.auto_speech_recognition, module_name=Models.kws_kwsbp)
class GenericKeyWordSpotting(Model):
def __init__(self, model_dir: str, *args, **kwargs):

2
modelscope/pipelines/audio/kws_kwsbp_pipeline.py
View File

@@ -18,7 +18,7 @@ __all__ = ['KeyWordSpottingKwsbpPipeline']
@PIPELINES.register_module(
Tasks.key_word_spotting, module_name=Pipelines.kws_kwsbp)
Tasks.auto_speech_recognition, module_name=Pipelines.kws_kwsbp)
class KeyWordSpottingKwsbpPipeline(Pipeline):
"""KWS Pipeline - key word spotting decoding
"""

1
modelscope/utils/constant.py
View File

@@ -60,7 +60,6 @@ class Tasks(object):
auto_speech_recognition = 'auto-speech-recognition'
text_to_speech = 'text-to-speech'
speech_signal_process = 'speech-signal-process'
key_word_spotting = 'key-word-spotting'
# multi-modal tasks
image_captioning = 'image-captioning'

15
tests/pipelines/test_key_word_spotting.py
View File

@@ -6,11 +6,8 @@ import unittest
import requests
from modelscope.metainfo import Pipelines, Preprocessors
from modelscope.models import Model
from modelscope.pipelines import pipeline
from modelscope.preprocessors import build_preprocessor
from modelscope.utils.constant import Fields, InputFields, Tasks
from modelscope.utils.constant import Tasks
from modelscope.utils.test_utils import test_level
KWSBP_URL = 'https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/KWS/tools/kwsbp'
@@ -58,7 +55,7 @@ class KeyWordSpottingTest(unittest.TestCase):
f.write(r.content)
kwsbp_16k_pipline = pipeline(
task=Tasks.key_word_spotting, model=self.model_id)
task=Tasks.auto_speech_recognition, model=self.model_id)
self.assertTrue(kwsbp_16k_pipline is not None)
kws_result = kwsbp_16k_pipline(
@@ -107,7 +104,7 @@ class KeyWordSpottingTest(unittest.TestCase):
keywords = [{'keyword': '播放音乐'}]
kwsbp_16k_pipline = pipeline(
task=Tasks.key_word_spotting,
task=Tasks.auto_speech_recognition,
model=self.model_id,
keywords=keywords)
self.assertTrue(kwsbp_16k_pipline is not None)
@@ -169,7 +166,7 @@ class KeyWordSpottingTest(unittest.TestCase):
f.write(r.content)
kwsbp_16k_pipline = pipeline(
task=Tasks.key_word_spotting, model=self.model_id)
task=Tasks.auto_speech_recognition, model=self.model_id)
self.assertTrue(kwsbp_16k_pipline is not None)
kws_result = kwsbp_16k_pipline(
@@ -232,7 +229,7 @@ class KeyWordSpottingTest(unittest.TestCase):
f.write(r.content)
kwsbp_16k_pipline = pipeline(
task=Tasks.key_word_spotting, model=self.model_id)
task=Tasks.auto_speech_recognition, model=self.model_id)
self.assertTrue(kwsbp_16k_pipline is not None)
kws_result = kwsbp_16k_pipline(
@@ -318,7 +315,7 @@ class KeyWordSpottingTest(unittest.TestCase):
f.write(r.content)
kwsbp_16k_pipline = pipeline(
task=Tasks.key_word_spotting, model=self.model_id)
task=Tasks.auto_speech_recognition, model=self.model_id)
self.assertTrue(kwsbp_16k_pipline is not None)
kws_result = kwsbp_16k_pipline(