From 56a32ce5e3b4119ace5a6b972c9bef40f4cfc085 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期三, 15 二月 2023 20:08:38 +0800
Subject: [PATCH] add scripts for simu data
---
egs/mars/sd/scripts/dump_rttm_to_labels.py | 14 ++++++++++----
1 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py
index ce9e7e8..6b98d17 100644
--- a/egs/mars/sd/scripts/dump_rttm_to_labels.py
+++ b/egs/mars/sd/scripts/dump_rttm_to_labels.py
@@ -36,14 +36,20 @@
len(meeting_scp), len(meeting2rttm)))
common_keys = set(meeting_scp.keys()) & set(meeting2rttm.keys())
logging.warning("Keep {} records.".format(len(common_keys)))
+ new_meeting_scp = OrderedDict()
for key in meeting_scp:
if key not in common_keys:
- meeting_scp.pop(key)
logging.warning("Pop {} from wav scp".format(key))
- if key not in meeting2rttm:
- meeting2rttm.pop(key)
+ else:
+ new_meeting_scp[key] = meeting_scp[key]
+ new_meeting2rttm = OrderedDict()
+ for key in meeting2rttm:
+ if key not in common_keys:
logging.warning("Pop {} from rttm scp".format(key))
+ else:
+ new_meeting2rttm[key] = meeting2rttm[key]
+ meeting_scp, meeting2rttm = new_meeting_scp, new_meeting2rttm
if not os.path.exists(args.out_dir):
os.makedirs(args.out_dir)
@@ -69,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=int)
+ multi_label = np.zeros([n_spk, num_frame], dtype=np.int32)
for _, st, dur, spk in spk_turns:
idx = spk_list.index(spk)
--
Gitblit v1.9.1