From dd4946a50db62a180ab11a5e371ea2ef44954c3b Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期三, 15 三月 2023 16:44:59 +0800
Subject: [PATCH] update
---
funasr/bin/eend_ola_inference.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/eend_ola_inference.py b/funasr/bin/eend_ola_inference.py
index 79e93a8..b35824a 100755
--- a/funasr/bin/eend_ola_inference.py
+++ b/funasr/bin/eend_ola_inference.py
@@ -237,7 +237,8 @@
results = speech2diar(**batch)
# post process
- a = medfilt(results[0], (11, 1))
+ a = results[0].cpu().numpy()
+ a = medfilt(a, (11, 1))
rst = []
for spkid, frames in enumerate(a.T):
frames = np.pad(frames, (1, 1), 'constant')
--
Gitblit v1.9.1