31 lines
967 B
Python
31 lines
967 B
Python
from text import chinese, cleaned_text_to_sequence
|
|
|
|
|
|
language_module_map = {
|
|
'ZH': chinese
|
|
}
|
|
|
|
|
|
def clean_text(text, language):
|
|
language_module = language_module_map[language]
|
|
norm_text = language_module.text_normalize(text)
|
|
phones, tones, word2ph = language_module.g2p(norm_text)
|
|
return norm_text, phones, tones, word2ph
|
|
|
|
def clean_text_bert(text, language):
|
|
language_module = language_module_map[language]
|
|
norm_text = language_module.text_normalize(text)
|
|
phones, tones, word2ph = language_module.g2p(norm_text)
|
|
bert = language_module.get_bert_feature(norm_text, word2ph)
|
|
return phones, tones, bert
|
|
|
|
def text_to_sequence(text, language):
|
|
norm_text, phones, tones, word2ph = clean_text(text, language)
|
|
return cleaned_text_to_sequence(phones, tones, language)
|
|
|
|
if __name__ == '__main__':
|
|
#print(text_to_sequence("你好，啊啊啊额、还是到付红四方。", 'ZH'))
|
|
#print(text_to_sequence("hello", 'EN'))
|
|
|
|
|