mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 04:29:22 +01:00
[to #42322933] add new kws models in UT and update requirements/audio_kws.txt
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11830273 * [Add] add new models for kws UT * [Update] update requirements * pre-commit
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
kaldiio
|
||||
kwsbp>=0.0.2
|
||||
kwsbp>=0.0.6
|
||||
matplotlib
|
||||
numpy
|
||||
py_sound_connect>=0.1
|
||||
|
||||
@@ -180,6 +180,14 @@ class KeyWordSpottingTest(unittest.TestCase, DemoCompatibilityCheck):
|
||||
'model_id': 'damo/speech_charctc_kws_phone-xiaoyun',
|
||||
'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
|
||||
'keywords': '小云小云'
|
||||
}, {
|
||||
'model_id': 'damo/speech_charctc_kws_phone-speechcommands',
|
||||
'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
|
||||
'keywords': '小云小云'
|
||||
}, {
|
||||
'model_id': 'damo/speech_charctc_kws_phone-wenwen',
|
||||
'wav_path': 'data/test/audios/kws_xiaoyunxiaoyun.wav',
|
||||
'keywords': '小云小云'
|
||||
}]
|
||||
|
||||
def setUp(self) -> None:
|
||||
@@ -330,10 +338,11 @@ class KeyWordSpottingTest(unittest.TestCase, DemoCompatibilityCheck):
|
||||
wav_path = item['wav_path']
|
||||
keywords = item['keywords']
|
||||
|
||||
logger.info('run with model_id:' + model_id)
|
||||
logger.info('run with model_id:' + model_id + ' with keywords:'
|
||||
+ keywords)
|
||||
kws_result = self.run_pipeline(
|
||||
model_id=model_id, audio_in=wav_path, keywords=keywords)
|
||||
self.check_result('test_run_with_all_models', kws_result)
|
||||
logger.info(ColorCodes.YELLOW + str(kws_result) + ColorCodes.END)
|
||||
|
||||
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
|
||||
def test_demo_compatibility(self):
|
||||
|
||||
Reference in New Issue
Block a user