update:加入测试速度功能

This commit is contained in:
hrz
2025-02-09 16:44:57 +08:00
parent 99f634f929
commit ba2cc8c447
7 changed files with 143 additions and 27 deletions
+9 -4
View File
@@ -20,7 +20,6 @@ class TTS(ABC):
def __init__(self, config, delete_audio_file):
self.delete_audio_file = delete_audio_file
self.output_file = config.get("output_file")
self.delete_audio_file = delete_audio_file
@abstractmethod
def generate_filename(self):
@@ -164,6 +163,9 @@ def create_instance(class_name, *args, **kwargs):
if __name__ == "__main__":
"""
响应速度测试
"""
config = read_config(get_project_dir() + "config.yaml")
tts = create_instance(
config["selected_module"]["TTS"],
@@ -171,6 +173,9 @@ if __name__ == "__main__":
config["delete_audio"]
)
tts.output_file = get_project_dir() + tts.output_file
file_path = tts.to_tts("你好,测试")
print(file_path)
print(tts.wav_to_opus_data(file_path))
start = datetime.now()
file_path = tts.to_tts("你好,测试,我是人工智能小智")
print("语音合成耗时:" + str(datetime.now() - start))
start = datetime.now()
tts.wav_to_opus_data(file_path)
print("语音opus耗时:" + str(datetime.now() - start))