Spaces:
Sleeping
Sleeping
| """ from https://github.com/keithito/tacotron """ | |
| from text import cleaners | |
| #from text.symbols import symbols | |
| from viphoneme import syms, vi2IPA_split | |
| symbols = syms | |
| # Mappings from symbol to numeric ID and vice versa: | |
| _symbol_to_id = {s: i for i, s in enumerate(symbols)} | |
| _id_to_symbol = {i: s for i, s in enumerate(symbols)} | |
| def text_to_sequence(text, cleaner_names): | |
| sequence = [] | |
| text = text.replace('\s+',' ').lower() | |
| phon = vi2IPA_split(text,"/") | |
| phon = phon.split("/")[1:] | |
| eol = -1 | |
| for i,p in reversed(list(enumerate(phon))): | |
| if p not in ["..",""," ","."," "]: | |
| eol = i | |
| break | |
| phones = phon[:i+1]+[" ","."] | |
| phones_id =[] | |
| for i in phones: | |
| if i in _symbol_to_id: | |
| phones_id.append(_symbol_to_id[i]) | |
| #phones_id = [_symbol_to_id[i] for i in phones] | |
| sequence.extend(phones_id) | |
| return sequence | |
| def cleaned_text_to_sequence(cleaned_text): | |
| sequence = [] | |
| phon = cleaned_text.split("/")[1:] | |
| eol = -1 | |
| for i,p in reversed(list(enumerate(phon))): | |
| if p not in ["..",""," ","."," "]: | |
| eol = i | |
| break | |
| phones = phon[:i+1]+[" ","."] | |
| phones_id =[] | |
| for i in phones: | |
| if i in _symbol_to_id: | |
| phones_id.append(_symbol_to_id[i]) | |
| #phones_id = [_symbol_to_id[i] for i in phones] | |
| sequence.extend(phones_id) | |
| return sequence | |
| def sequence_to_text(sequence): | |
| result = '' | |
| for symbol_id in sequence: | |
| if symbol_id in _id_to_symbol: | |
| result += _id_to_symbol[symbol_id] | |
| return result | |
| def _clean_text(text, cleaner_names): | |
| for name in cleaner_names: | |
| cleaner = getattr(cleaners, name) | |
| if not cleaner: | |
| raise Exception("Unknown cleaner: %s" % name) | |
| text = cleaner(text) | |
| return text | |