From 56a32ce5e3b4119ace5a6b972c9bef40f4cfc085 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= Date: Wed, 15 Feb 2023 20:08:38 +0800 Subject: [PATCH] add scripts for simu data --- egs/mars/sd/scripts/dump_rttm_to_labels.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py index f254e8502..6b98d17bf 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 @@ def calc_labels(spk_turns, spk_list, length, n_spk, remove_sil=False, max_overla 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=int) + multi_label = np.zeros([n_spk, num_frame], dtype=np.int32) for _, st, dur, spk in spk_turns: idx = spk_list.index(spk)