From b15db52e4e67da8a133a67e8ffa415386de48b40 Mon Sep 17 00:00:00 2001
From: zhuyunfeng <10596244@qq.com>
Date: 星期二, 09 五月 2023 23:03:15 +0800
Subject: [PATCH] Add contributor
---
egs/mars/sd/scripts/dump_rttm_to_labels.py | 12 ++++++++----
1 files changed, 8 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 f254e85..ec1c765 100644
--- a/egs/mars/sd/scripts/dump_rttm_to_labels.py
+++ b/egs/mars/sd/scripts/dump_rttm_to_labels.py
@@ -37,18 +37,22 @@
common_keys = set(meeting_scp.keys()) & set(meeting2rttm.keys())
logging.warning("Keep {} records.".format(len(common_keys)))
new_meeting_scp = OrderedDict()
+ rm_keys = []
for key in meeting_scp:
if key not in common_keys:
- logging.warning("Pop {} from wav scp".format(key))
+ rm_keys.append(key)
else:
new_meeting_scp[key] = meeting_scp[key]
+ logging.warning("Keys are removed from wav scp: {}".format(" ".join(rm_keys)))
+
new_meeting2rttm = OrderedDict()
+ rm_keys = []
for key in meeting2rttm:
if key not in common_keys:
- logging.warning("Pop {} from rttm scp".format(key))
+ rm_keys.append(key)
else:
new_meeting2rttm[key] = meeting2rttm[key]
-
+ logging.warning("Keys are removed from rttm scp: {}".format(" ".join(rm_keys)))
meeting_scp, meeting2rttm = new_meeting_scp, new_meeting2rttm
if not os.path.exists(args.out_dir):
os.makedirs(args.out_dir)
@@ -75,7 +79,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.float32)
for _, st, dur, spk in spk_turns:
idx = spk_list.index(spk)
--
Gitblit v1.9.1