志浩
2023-02-15 de034ca5d232178f97a0836e05a7b99e36f67050
egs/mars/sd/scripts/dump_rttm_to_labels.py
@@ -44,7 +44,7 @@
                    new_meeting_scp[key] = meeting_scp[key]
            new_meeting2rttm = OrderedDict()
            for key in meeting2rttm:
                if key not in meeting2rttm:
                if key not in common_keys:
                    logging.warning("Pop {} from rttm scp".format(key))
                else:
                    new_meeting2rttm[key] = meeting2rttm[key]
@@ -75,7 +75,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)