志浩
2023-02-15 7d23704b9aebe8b6113b0708545be36a2298ff69
add scripts for simu data
1个文件已修改
9 ■■■■ 已修改文件
egs/mars/sd/scripts/dump_rttm_to_labels.py 9 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
egs/mars/sd/scripts/dump_rttm_to_labels.py
@@ -36,15 +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))
                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)