From de034ca5d232178f97a0836e05a7b99e36f67050 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期三, 15 二月 2023 20:10:04 +0800
Subject: [PATCH] add scripts for simu data

---
 egs/mars/sd/scripts/dump_rttm_to_labels.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py
index 6b98d17..d3a58a2 100644
--- a/egs/mars/sd/scripts/dump_rttm_to_labels.py
+++ b/egs/mars/sd/scripts/dump_rttm_to_labels.py
@@ -75,7 +75,7 @@
                 sr=None, frame_shift=0.01):
     frame_shift = int(frame_shift * sr)
     num_frame = int((float(length) + (float(frame_shift) / 2)) / frame_shift)
-    multi_label = np.zeros([n_spk, num_frame], dtype=np.int32)
+    multi_label = np.zeros([n_spk, num_frame], dtype=np.float32)
     for _, st, dur, spk in spk_turns:
         idx = spk_list.index(spk)
 

--
Gitblit v1.9.1