From 0c3c9be2c4c1c4e4da4628c3987708c9a0763391 Mon Sep 17 00:00:00 2001
From: will_wang <53147925+willnufe@users.noreply.github.com>
Date: 星期三, 04 十二月 2024 17:47:31 +0800
Subject: [PATCH] paraformer onnx fp16导出方案 (#2264)
---
funasr/utils/load_utils.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index aabf5d9..9613d1b 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -138,7 +138,7 @@
def load_bytes(input):
- input = validate_frame_rate(input)
+ # input = validate_frame_rate(input)
middle_data = np.frombuffer(input, dtype=np.int16)
middle_data = np.asarray(middle_data)
if middle_data.dtype.kind not in "iu":
@@ -153,11 +153,12 @@
array = np.frombuffer((middle_data.astype(dtype) - offset) / abs_max, dtype=np.float32)
return array
+
def validate_frame_rate(
input,
fs: int = 16000,
):
-
+
# 灏嗘枃浠惰鍙栦负瀛楄妭娴�
byte_data = BytesIO(input)
@@ -175,7 +176,7 @@
# 鑾峰彇閲嶆柊閲囨牱鍚庣殑瀛楄妭娴佹暟鎹�
input = output.read()
-
+
return input
--
Gitblit v1.9.1