mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 10:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update preprocess_mdcc.py

Browse Source
This commit is contained in:
jinzr 2024-03-08 19:18:19 +08:00
parent 1c509a24f8
commit ba01616103
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
egs/mdcc/ASR/local/preprocess_mdcc.py
View File

@ -26,6 +26,7 @@ files in the directory "data/lang_char":
"""
import argparse
import logging
from pathlib import Path
from typing import List
@ -86,18 +87,21 @@ def get_word_segments(lines: List[str]) -> List[str]:
new_lines = []
for line in tqdm(lines, desc="Segmenting lines"):
# code switching
if len(line.split(" ")) > 1:
segments = []
for segment in line.split(" "):
if not is_cjk(segment[0]): # en segment
segments.append(segment)
else: # zh segment
segments.extend(pycantonese.segment(segment))
new_lines.append(" ".join(segments) + "\n")
# not code switching
else:
new_lines.append(" ".join(pycantonese.segment(line)) + "\n")
try:
# code switching
if len(line.split(" ")) > 1:
segments = []
for segment in line.split(" "):
if not is_cjk(segment[0]): # en segment
segments.append(segment)
else: # zh segment
segments.extend(pycantonese.segment(segment))
new_lines.append(" ".join(segments) + "\n")
# not code switching
else:
new_lines.append(" ".join(pycantonese.segment(line)) + "\n")
except:
logging.error(f"Failed to process line: {line}")
return new_lines