From 41b1d35048603ba0a74e9dee5be1a1c8c76444d3 Mon Sep 17 00:00:00 2001
From: 凌匀 <ailsa.zly@alibaba-inc.com>
Date: 星期四, 16 二月 2023 17:02:56 +0800
Subject: [PATCH] update vad_inference.py
---
funasr/bin/vad_inference.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/vad_inference.py b/funasr/bin/vad_inference.py
index 0d96594..1cdb582 100644
--- a/funasr/bin/vad_inference.py
+++ b/funasr/bin/vad_inference.py
@@ -111,6 +111,7 @@
# b. Forward Encoder sreaming
segments = []
+ segments_tmp = []
step = 6000
t_offset = 0
for t_offset in range(0, feats_len, min(step, feats_len - t_offset)):
@@ -128,9 +129,8 @@
batch = to_device(batch, device=self.device)
segments_part = self.vad_model(**batch)
if segments_part:
- segments += segments_part
- #print(segments)
-
+ segments_tmp += segments_part[0]
+ segments.append(segments_tmp)
return segments
--
Gitblit v1.9.1