Add contributor

zhuyunfeng

2023-05-09 b15db52e4e67da8a133a67e8ffa415386de48b40

 egs/mars/sd/scripts/dump_rttm_to_labels.py

@@ -31,9 +31,29 @@
        for scp_path in wav_scp_list:
            meeting_scp.update(load_scp_as_dict(scp_path))

        assert len(meeting_scp) == len(meeting2rttm), \
            "Number of wav and rttm mismatch {} != {}".format(len(meeting_scp), len(meeting2rttm))
        if len(meeting_scp) != len(meeting2rttm):
            logging.warning("Number of wav and rttm mismatch {} != {}".format(
                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()
            rm_keys = []
            for key in meeting_scp:
                if key not in common_keys:
                    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:
                    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)

@@ -59,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)