From b340c0fadfe0d36d714b6fc592a8c5e308dea7fd Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期一, 20 二月 2023 11:47:10 +0800
Subject: [PATCH] simu data
---
egs/mars/sd/scripts/simu_chunk_with_labels.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/egs/mars/sd/scripts/simu_chunk_with_labels.py b/egs/mars/sd/scripts/simu_chunk_with_labels.py
index 96d3c0e..3f9fe1e 100644
--- a/egs/mars/sd/scripts/simu_chunk_with_labels.py
+++ b/egs/mars/sd/scripts/simu_chunk_with_labels.py
@@ -93,9 +93,9 @@
def calculate_embedding(spk, spk2utts, utt2xvec, embedding_dim, average_emb_num):
- # process for empty speaker
+ # process for dummy speaker
if spk == "None":
- return np.zeros((embedding_dim, ), dtype=np.float32)
+ return np.zeros((1, embedding_dim), dtype=np.float32)
# calculate averaged speaker embeddings
utt_list = spk2utts[spk]
@@ -103,7 +103,7 @@
xvec_list = [kaldiio.load_mat(utt2xvec[utt]) for utt in utt_list]
else:
xvec_list = [kaldiio.load_mat(utt2xvec[utt]) for utt in random.sample(utt_list, average_emb_num)]
- xvec = np.mean(np.hstack(xvec_list), axis=0)
+ xvec = np.mean(np.concatenate(xvec_list, axis=0), axis=0)
return xvec
--
Gitblit v1.9.1