From 4870f0f8a5f3ae9072a56b6f320ba7ebcfaf4373 Mon Sep 17 00:00:00 2001
From: Binbin Gu <gubinbin@outlook.com>
Date: 星期五, 02 六月 2023 11:51:02 +0800
Subject: [PATCH] Update cardinal.py (#562)
---
funasr/bin/diar_infer.py | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/diar_infer.py b/funasr/bin/diar_infer.py
index f2dcb1e..4460e3d 100755
--- a/funasr/bin/diar_infer.py
+++ b/funasr/bin/diar_infer.py
@@ -235,8 +235,11 @@
new_seq.append(x)
else:
idx_list = np.where(seq < 2 ** vec_dim)[0]
- idx = np.abs(idx_list - i).argmin()
- new_seq.append(seq[idx_list[idx]])
+ if len(idx_list) > 0:
+ idx = np.abs(idx_list - i).argmin()
+ new_seq.append(seq[idx_list[idx]])
+ else:
+ new_seq.append(0)
return np.row_stack([int2vec(x, vec_dim) for x in new_seq])
def post_processing(self, raw_logits: torch.Tensor, spk_num: int, output_format: str = "speaker_turn"):
--
Gitblit v1.9.1