mirror of
https://github.com/index-tts/index-tts.git
synced 2025-11-28 02:10:23 +08:00
72c09ec0b7* indextts2 * update lfs for audio files --------- Co-authored-by: wangyining02 <wangyining02@bilibili.com>
41 lines
No EOL
1.1 KiB
Python
41 lines
No EOL
1.1 KiB
Python
import re
|
|
|
|
from textstat import textstat
|
|
|
|
|
|
def contains_chinese(text):
|
|
# 正则表达式，用于匹配中文字符 + 数字 -> 都认为是 zh
|
|
if re.search(r'[\u4e00-\u9fff0-9]', text):
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_text_syllable_num(text):
|
|
chinese_char_pattern = re.compile(r'[\u4e00-\u9fff]')
|
|
number_char_pattern = re.compile(r'[0-9]')
|
|
syllable_num = 0
|
|
tokens = re.findall(r'[\u4e00-\u9fff]+|[a-zA-Z]+|[0-9]+', text)
|
|
# print(tokens)
|
|
if contains_chinese(text):
|
|
for token in tokens:
|
|
if chinese_char_pattern.search(token) or number_char_pattern.search(token):
|
|
syllable_num += len(token)
|
|
else:
|
|
syllable_num += textstat.syllable_count(token)
|
|
else:
|
|
syllable_num = textstat.syllable_count(text)
|
|
|
|
return syllable_num
|
|
|
|
|
|
def get_text_tts_dur(text):
|
|
min_speed = 3 # 2.18 #
|
|
max_speed = 5.50
|
|
|
|
ratio = 0.8517 if contains_chinese(text) else 1.0
|
|
|
|
syllable_num = get_text_syllable_num(text)
|
|
max_dur = syllable_num * ratio / max_speed
|
|
min_dur = syllable_num * ratio / min_speed
|
|
|
|
return max_dur, min_dur |