You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
PaddleSpeech/third_party/python-pinyin/pypinyin/style/bopomofo.py

75 lines
2.5 KiB

E2E/Streaming Transformer/Conformer ASR (#578) * add cmvn and label smoothing loss layer * add layer for transformer * add glu and conformer conv * add torch compatiable hack, mask funcs * not hack size since it exists * add test; attention * add attention, common utils, hack paddle * add audio utils * conformer batch padding mask bug fix #223 * fix typo, python infer fix rnn mem opt name error and batchnorm1d, will be available at 2.0.2 * fix ci * fix ci * add encoder * refactor egs * add decoder * refactor ctc, add ctc align, refactor ckpt, add warmup lr scheduler, cmvn utils * refactor docs * add fix * fix readme * fix bugs, refactor collator, add pad_sequence, fix ckpt bugs * fix docstring * refactor data feed order * add u2 model * refactor cmvn, test * add utils * add u2 config * fix bugs * fix bugs * fix autograd maybe has problem when using inplace operation * refactor data, build vocab; add format data * fix text featurizer * refactor build vocab * add fbank, refactor feature of speech * refactor audio feat * refactor data preprare * refactor data * model init from config * add u2 bins * flake8 * can train * fix bugs, add coverage, add scripts * test can run * fix data * speed perturb with sox * add spec aug * fix for train * fix train logitc * fix logger * log valid loss, time dataset process * using np for speed perturb, remove some debug log of grad clip * fix logger * fix build vocab * fix logger name * using module logger as default * fix * fix install * reorder imports * fix board logger * fix logger * kaldi fbank and mfcc * fix cmvn and print prarams * fix add_eos_sos and cmvn * fix cmvn compute * fix logger and cmvn * fix subsampling, label smoothing loss, remove useless * add notebook test * fix log * fix tb logger * multi gpu valid * fix log * fix log * fix config * fix compute cmvn, need paddle 2.1 * add cmvn notebook * fix layer tools * fix compute cmvn * add rtf * fix decoding * fix layer tools * fix log, add avg script * more avg and test info * fix dataset pickle problem; using 2.1 paddle; num_workers can > 0; ckpt save in exp dir;fix setup.sh; * add vimrc * refactor tiny script, add transformer and stream conf * spm demo; librisppech scripts and confs * fix log * add librispeech scripts * refactor data pipe; fix conf; fix u2 default params * fix bugs * refactor aishell scripts * fix test * fix cmvn * fix s0 scripts * fix ds2 scripts and bugs * fix dev & test dataset filter * fix dataset filter * filter dev * fix ckpt path * filter test, since librispeech will cause OOM, but all test wer will be worse, since mismatch train with test * add comment * add syllable doc * fix ds2 configs * add doc * add pypinyin tools * fix decoder using blank_id=0 * mmseg with pybind11 * format code
3 years ago
"""BOPOMOFO 相关的几个拼音风格实现:
Style.BOPOMOFO
Style.BOPOMOFO_FIRST
"""
import re
from typing import Any
from typing import Dict
from typing import Text
from typing import Tuple
from pypinyin.constants import Style
from pypinyin.style import register
from pypinyin.style._constants import RE_TONE3
from pypinyin.style._utils import replace_symbol_to_number
# 注音转换表
BOPOMOFO_REPLACE = (
(re.compile(r'^m(\d)$'), 'mu\\1'), # 呣
(re.compile(r'^n(\d)$'), 'N\\1'), # 嗯
(re.compile(r'^r5$'), 'er5'), # 〜兒
(re.compile(r'iu'), 'iou'),
(re.compile(r'ui'), 'uei'),
(re.compile(r'ong'), 'ung'),
(re.compile(r'^yi?'), 'i'),
(re.compile(r'^wu?'), 'u'),
(re.compile(r'iu'), 'v'),
(re.compile(r'^([jqx])u'), '\\1v'),
(re.compile(r'([iuv])n'), '\\1en'),
(re.compile(r'^zhi?'), 'Z'),
(re.compile(r'^chi?'), 'C'),
(re.compile(r'^shi?'), 'S'),
(re.compile(r'^([zcsr])i'), '\\1'),
(re.compile(r'ai'), 'A'),
(re.compile(r'ei'), 'I'),
(re.compile(r'ao'), 'O'),
(re.compile(r'ou'), 'U'),
(re.compile(r'ang'), 'K'),
(re.compile(r'eng'), 'G'),
(re.compile(r'an'), 'M'),
(re.compile(r'en'), 'N'),
(re.compile(r'er'), 'R'),
(re.compile(r'eh'), 'E'),
(re.compile(r'([iv])e'), '\\1E'),
(re.compile(r'([^0-4])$'), '\\g<1>0'),
(re.compile(r'1$'), ''), ) # type: Tuple[Tuple[Any]]
BOPOMOFO_TABLE = dict(
zip('bpmfdtnlgkhjqxZCSrzcsiuvaoeEAIOUMNKGR2340',
'ㄅㄆㄇㄈㄉㄊㄋㄌㄍㄎㄏㄐㄑㄒㄓㄔㄕㄖㄗㄘㄙㄧㄨㄩㄚㄛㄜㄝㄞㄟㄠㄡㄢㄣㄤㄥㄦˊˇˋ˙')) # type: Dict[Text, Text]
class BopomofoConverter():
def _pre_convert(self, pinyin: Text) -> Text:
# 用数字表示声调
pinyin = replace_symbol_to_number(pinyin)
# 将声调数字移动到最后
return RE_TONE3.sub(r'\1\3\2', pinyin)
def to_bopomofo(self, pinyin: Text, **kwargs: Any) -> Text:
pinyin = self._pre_convert(pinyin)
# 查表替换成注音
for find_re, replace in BOPOMOFO_REPLACE:
pinyin = find_re.sub(replace, pinyin)
pinyin = ''.join(BOPOMOFO_TABLE.get(x, x) for x in pinyin)
return pinyin
def to_bopomofo_first(self, pinyin: Text, **kwargs: Any) -> Text:
pinyin = self.to_bopomofo(pinyin, **kwargs)
return pinyin[0]
converter = BopomofoConverter()
register(Style.BOPOMOFO, func=converter.to_bopomofo)
register(Style.BOPOMOFO_FIRST, func=converter.to_bopomofo_first)