From 7d23704b9aebe8b6113b0708545be36a2298ff69 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= Date: Wed, 15 Feb 2023 20:04:08 +0800 Subject: [PATCH] add scripts for simu data --- egs/mars/sd/scripts/dump_rttm_to_labels.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py index cf76b7545..d582eb63a 100644 --- a/egs/mars/sd/scripts/dump_rttm_to_labels.py +++ b/egs/mars/sd/scripts/dump_rttm_to_labels.py @@ -36,15 +36,20 @@ class MyRunner(MultiProcessRunnerV3): 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)) + else: + new_meeting_scp[key] = meeting_scp[key] + new_meeting2rttm = OrderedDict() for key in meeting2rttm: if key not in meeting2rttm: - meeting2rttm.pop(key) 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)