From 31b4a9837a5836b809de276c1bc781dfa5b3189f Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 23 二月 2023 23:03:40 +0800
Subject: [PATCH] sond pipeline

---
 egs/mars/sd/scripts/simu_chunk_with_labels.py                        |    2 ++
 egs/mars/sd/scripts/real_meeting_process/dump_real_meeting_chunks.py |    1 +
 2 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/egs/mars/sd/scripts/real_meeting_process/dump_real_meeting_chunks.py b/egs/mars/sd/scripts/real_meeting_process/dump_real_meeting_chunks.py
index c6ff1f5..a2bcd39 100644
--- a/egs/mars/sd/scripts/real_meeting_process/dump_real_meeting_chunks.py
+++ b/egs/mars/sd/scripts/real_meeting_process/dump_real_meeting_chunks.py
@@ -26,6 +26,7 @@
         short_spk_list.append(spk)
 
     ivc_list = [kaldiio.load_mat(utt2ivc[utt]) for utt in utt_list]
+    ivc_list = [x/np.linalg.norm(x, axis=-1) for x in ivc_list]
     ivc = np.concatenate(ivc_list, axis=0)
     ivc = np.mean(ivc, axis=0, keepdims=False)
     return ivc
diff --git a/egs/mars/sd/scripts/simu_chunk_with_labels.py b/egs/mars/sd/scripts/simu_chunk_with_labels.py
index 3f9fe1e..226784b 100644
--- a/egs/mars/sd/scripts/simu_chunk_with_labels.py
+++ b/egs/mars/sd/scripts/simu_chunk_with_labels.py
@@ -103,6 +103,8 @@
         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)]
+    # TODO: rerun the simulation
+    xvec_list = [x / np.linalg.norm(x, axis=-1) for x in xvec_list]
     xvec = np.mean(np.concatenate(xvec_list, axis=0), axis=0)
 
     return xvec

--
Gitblit v1.9.1