speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add scripts for simu data

Browse Source
This commit is contained in:
志浩 2023-02-15 19:58:08 +08:00
parent 0917d4ee0e
commit 3aec61c6ea
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
egs/mars/sd/scripts/dump_rttm_to_labels.py
View File

@ -31,8 +31,18 @@ class MyRunner(MultiProcessRunnerV3):
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)))
for key in meeting_scp:
if key not in common_keys:
meeting_scp.pop(key)
logging.warning("Pop {} from wav scp".format(key))
if key not in meeting2rttm:
meeting2rttm.pop(key)
logging.warning("Pop {} from rttm scp".format(key))
if not os.path.exists(args.out_dir):
os.makedirs(args.out_dir)