From 9be8a443d74d68f179de88fff13b4e8424579d7b Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 10 三月 2023 18:24:39 +0800
Subject: [PATCH] Merge pull request #207 from alibaba-damo-academy/dev_dzh
---
egs/mars/sd/scripts/dump_rttm_to_labels.py | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py
index d3a58a2..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)
--
Gitblit v1.9.1