| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://modelscope.oss-cn-beijing.aliyuncs.com/test/audios/asr_example.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "每一天都要快乐喔" |
| | | |
| | | def test_paraformer(self): |
| | | inference_pipeline = pipeline( |
| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://modelscope.oss-cn-beijing.aliyuncs.com/test/audios/asr_example.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "每一天都要快乐喔" |
| | | |
| | | |
| | | class TestMfccaInferencePipelines(unittest.TestCase): |
| | |
| | | model='NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950', |
| | | model_revision='v3.0.0') |
| | | rec_result = inference_pipeline( |
| | | audio_in='16:32https://pre.modelscope.cn/api/v1/models/NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/repo?Revision=master&FilePath=example/asr_example_mc.wav') |
| | | audio_in='https://pre.modelscope.cn/api/v1/models/NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/repo?Revision=master&FilePath=example/asr_example_mc.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | |
| | | |
| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_hotword.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "国务院发展研究中心市场经济研究所副所长邓郁松认为" |
| | | |
| | | def test_paraformer_large_aishell1(self): |
| | | inference_pipeline = pipeline( |
| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "欢迎大家来体验达摩院推出的语音识别模型" |
| | | |
| | | def test_paraformer_large_aishell2(self): |
| | | inference_pipeline = pipeline( |
| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "欢迎大家来体验达摩院推出的语音识别模型" |
| | | |
| | | def test_paraformer_large_common(self): |
| | | inference_pipeline = pipeline( |
| | |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "欢迎大家来体验达摩院推出的语音识别模型" |
| | | |
| | | def test_paraformer_large_online_common(self): |
| | | inference_pipeline = pipeline( |
| | | task=Tasks.auto_speech_recognition, |
| | | model='damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online', |
| | | model_revision='v1.0.6', |
| | | update_model=False, |
| | | mode="paraformer_fake_streaming" |
| | | ) |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "欢迎大家来体验达摩院推出的语音识别模型" |
| | | |
| | | def test_paraformer_online_common(self): |
| | | inference_pipeline = pipeline( |
| | | task=Tasks.auto_speech_recognition, |
| | | model='damo/speech_paraformer_asr_nat-zh-cn-16k-common-vocab8404-online', |
| | | model_revision='v1.0.6', |
| | | update_model=False, |
| | | mode="paraformer_fake_streaming" |
| | | ) |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav') |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | assert rec_result["text"] == "欢迎大家来体验达摩院推出的语音识别模型" |
| | | |
| | | def test_paraformer_tiny_commandword(self): |
| | | inference_pipeline = pipeline( |
| | |
| | | param_dict={"decoding_model": "normal"}) |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | |
| | | def test_uniasr_2pass_zhcn_16k_common_vocab8358_offline(self): |
| | | inference_pipeline = pipeline( |
| | | task=Tasks.auto_speech_recognition, |
| | | model='damo/speech_UniASR_asr_2pass-zh-cn-16k-common-vocab8358-tensorflow1-offline') |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav', |
| | | param_dict={"decoding_model": "offline"}) |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | |
| | | def test_uniasr_2pass_zhcn_16k_common_vocab8358_online(self): |
| | | inference_pipeline = pipeline( |
| | | task=Tasks.auto_speech_recognition, |
| | | model='damo/speech_UniASR_asr_2pass-zh-cn-16k-common-vocab8358-tensorflow1-online') |
| | | rec_result = inference_pipeline( |
| | | audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav', |
| | | param_dict={"decoding_model": "normal"}) |
| | | logger.info("asr inference result: {0}".format(rec_result)) |
| | | |
| | | |
| | | |
| | | if __name__ == '__main__': |