From a8701ad5df6255502c431fb749b4661ff4853e57 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期三, 02 八月 2023 15:26:01 +0800
Subject: [PATCH] TOLD/SOND: add utt2num_frame script
---
egs/callhome/diarization/sond/script/calc_real_meeting_frame_labels.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/egs/callhome/diarization/sond/script/calc_real_meeting_frame_labels.py b/egs/callhome/diarization/sond/script/calc_real_meeting_frame_labels.py
index 58232d4..f295c0c 100644
--- a/egs/callhome/diarization/sond/script/calc_real_meeting_frame_labels.py
+++ b/egs/callhome/diarization/sond/script/calc_real_meeting_frame_labels.py
@@ -1,6 +1,6 @@
import numpy as np
-from opennmt.utils.job_runner import MultiProcessRunnerV3
-from opennmt.utils.misc import load_scp_as_list, load_scp_as_dict
+from funasr.utils.job_runner import MultiProcessRunnerV3
+from funasr.utils.misc import load_scp_as_list, load_scp_as_dict
import os
import librosa
import scipy.io as sio
@@ -90,7 +90,7 @@
for mid, wav_path, rttms in task_list:
meeting_labels, spk_list = build_labels(wav_path, rttms, args.n_spk, args.remove_sil,
args.sr, args.frame_shift)
- save_path = os.path.join(args.out_dir, "{}.lbl".format(mid))
+ save_path = os.path.join(args.out_dir, "{}.lbl.mat".format(mid))
sio.savemat(save_path, {"labels": meeting_labels.astype(bool), "spk_list": spk_list})
# print mid
return None
--
Gitblit v1.9.1