Spaces:
Runtime error
Runtime error
File size: 1,015 Bytes
d1b91e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import numpy as np
from data_gen.tts.base_binarizer import BaseBinarizer
class ZhBinarizer(BaseBinarizer):
@staticmethod
def process_align(tg_fn, item):
BaseBinarizer.process_align(tg_fn, item)
# char-level pitch
if 'f0' in item:
ph_list = item['ph'].split(" ")
item['f0_ph'] = np.array([0 for _ in item['f0']], dtype=float)
char_start_idx = 0
f0s_char = []
for idx, (f0_, ph_idx) in enumerate(zip(item['f0'], item['mel2ph'])):
is_pinyin = ph_list[ph_idx - 1][0].isalpha()
if not is_pinyin or ph_idx - item['mel2ph'][idx - 1] > 1:
if len(f0s_char) > 0:
item['f0_ph'][char_start_idx:idx] = sum(f0s_char) / len(f0s_char)
f0s_char = []
char_start_idx = idx
if not is_pinyin:
char_start_idx += 1
if f0_ > 0:
f0s_char.append(f0_)
|