From 630509da0f16bcfd145a69bb3223bed92761dc61 Mon Sep 17 00:00:00 2001
From: speech_asr <wucong.lyb@alibaba-inc.com>
Date: 星期五, 15 十二月 2023 14:03:43 +0800
Subject: [PATCH] update funasr_wss_client.py
---
runtime/python/websocket/funasr_wss_client.py | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/runtime/python/websocket/funasr_wss_client.py b/runtime/python/websocket/funasr_wss_client.py
index 66b3ce0..bfe4b0d 100644
--- a/runtime/python/websocket/funasr_wss_client.py
+++ b/runtime/python/websocket/funasr_wss_client.py
@@ -41,6 +41,10 @@
type=str,
default=None,
help="audio_in")
+parser.add_argument("--audio_fs",
+ type=int,
+ default=16000,
+ help="audio_fs")
parser.add_argument("--send_without_sleep",
action="store_true",
default=True,
@@ -164,7 +168,7 @@
hotword_msg=json.dumps(fst_dict)
print (hotword_msg)
- sample_rate = 16000
+ sample_rate = args.audio_fs
wav_format = "pcm"
use_itn=True
if args.use_itn == 0:
@@ -194,8 +198,7 @@
with open(wav_path, "rb") as f:
audio_bytes = f.read()
- # stride = int(args.chunk_size/1000*16000*2)
- stride = int(60 * args.chunk_size[1] / args.chunk_interval / 1000 * 16000 * 2)
+ stride = int(60 * args.chunk_size[1] / args.chunk_interval / 1000 * sample_rate * 2)
chunk_num = (len(audio_bytes) - 1) // stride + 1
# print(stride)
--
Gitblit v1.9.1