Merge pull request #225 from alibaba-damo-academy/dev_zly

update infer_online.py
This commit is contained in:
zhifu gao 2023-03-14 14:43:32 +08:00 committed by GitHub
commit 681868deae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 9 additions and 7 deletions

View File

@ -7,7 +7,7 @@ if __name__ == '__main__':
inference_pipline = pipeline(
task=Tasks.voice_activity_detection,
model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
model_revision=None,
model_revision='v1.2.0',
output_dir=output_dir,
batch_size=1,
)

View File

@ -8,9 +8,10 @@ if __name__ == '__main__':
inference_pipline = pipeline(
task=Tasks.voice_activity_detection,
model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
model_revision='v1.1.9',
output_dir=None,
model_revision='v1.2.0',
output_dir=output_dir,
batch_size=1,
mode='online',
)
speech, sample_rate = soundfile.read("./vad_example_16k.wav")
speech_length = speech.shape[0]

View File

@ -7,8 +7,8 @@ if __name__ == '__main__':
inference_pipline = pipeline(
task=Tasks.voice_activity_detection,
model="damo/speech_fsmn_vad_zh-cn-8k-common",
model_revision=None,
output_dir='./output_dir',
model_revision='v1.2.0',
output_dir=output_dir,
batch_size=1,
)
segments_result = inference_pipline(audio_in=audio_in)

View File

@ -8,9 +8,10 @@ if __name__ == '__main__':
inference_pipline = pipeline(
task=Tasks.voice_activity_detection,
model="damo/speech_fsmn_vad_zh-cn-8k-common",
model_revision='v1.1.9',
output_dir='./output_dir',
model_revision='v1.2.0',
output_dir=output_dir,
batch_size=1,
mode='online',
)
speech, sample_rate = soundfile.read("./vad_example_8k.wav")
speech_length = speech.shape[0]