auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-19 06:44:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add phoneme map to phonemize script

Browse Source
This commit is contained in:
Michael Hansen
2023-05-12 14:30:34 -05:00
parent 5bff1c3da7
commit 1ee3d7c7c5
3 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/python/piper_train/preprocess.py
View File

@@ -24,6 +24,7 @@ from .phonemize import (
PhonemeType,
phonemes_to_ids,
phonemize,
PHONEME_MAPS,
)
_LOGGER = logging.getLogger("preprocess")
@@ -250,6 +251,7 @@ def phonemize_batch_espeak(
casing = get_text_casing(args.text_casing)
silence_detector = make_silence_detector()
phonemizer = Phonemizer(default_voice=args.language)
phoneme_map = PHONEME_MAPS.get(args.langauge)
while True:
utt_batch = queue_in.get()
@@ -259,7 +261,9 @@ def phonemize_batch_espeak(
for utt in utt_batch:
try:
_LOGGER.debug(utt)
utt.phonemes = phonemize(casing(utt.text), phonemizer)
utt.phonemes = phonemize(
casing(utt.text), phonemizer, phoneme_map=phoneme_map
)
utt.phoneme_ids = phonemes_to_ids(
utt.phonemes,
missing_phonemes=utt.missing_phonemes,