فهرست منبع

add scripts for simu data

志浩 3 سال پیش
والد
کامیت
56a32ce5e3
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      egs/mars/sd/scripts/dump_rttm_to_labels.py

+ 1 - 1
egs/mars/sd/scripts/dump_rttm_to_labels.py

@@ -75,7 +75,7 @@ def calc_labels(spk_turns, spk_list, length, n_spk, remove_sil=False, max_overla
                 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.int32)
     for _, st, dur, spk in spk_turns:
         idx = spk_list.index(spk)