update report file name
Browse files
tests/test_app_models/test_funasr.py
CHANGED
|
@@ -13,7 +13,7 @@ def asr(get_platform):
|
|
| 13 |
return FunAsrOv()
|
| 14 |
|
| 15 |
|
| 16 |
-
def
|
| 17 |
#TODO: 测试CER
|
| 18 |
report = []
|
| 19 |
for audio_file, text, duration in read_emilia(count_limit=20):
|
|
@@ -21,6 +21,6 @@ def test_inference(asr):
|
|
| 21 |
audio = read_audio(audio_file)
|
| 22 |
asr_text, time_cost = asr.transcribe(audio)
|
| 23 |
report.append([audio_file,duration, text, asr_text, time_cost])
|
| 24 |
-
save_csv(REPORTS_DIR/"
|
| 25 |
speed=[i[4] for i in report]
|
| 26 |
print("Funasr average speed: ", sum(speed)/len(speed))
|
|
|
|
| 13 |
return FunAsrOv()
|
| 14 |
|
| 15 |
|
| 16 |
+
def test_funasr(asr):
|
| 17 |
#TODO: 测试CER
|
| 18 |
report = []
|
| 19 |
for audio_file, text, duration in read_emilia(count_limit=20):
|
|
|
|
| 21 |
audio = read_audio(audio_file)
|
| 22 |
asr_text, time_cost = asr.transcribe(audio)
|
| 23 |
report.append([audio_file,duration, text, asr_text, time_cost])
|
| 24 |
+
save_csv(REPORTS_DIR/"test_funasr.csv", ["audio", "duration", "ref", "asr", "time"], report)
|
| 25 |
speed=[i[4] for i in report]
|
| 26 |
print("Funasr average speed: ", sum(speed)/len(speed))
|
tests/test_app_models/test_llm.py
CHANGED
|
@@ -17,7 +17,7 @@ def test_llm_zh(llm):
|
|
| 17 |
for src in read_translation("zh", count_limit=None):
|
| 18 |
dst, time_cost = llm.translate(src, src_lang="zh", dst_lang="en")
|
| 19 |
report.append([src, dst, time_cost])
|
| 20 |
-
save_csv(REPORTS_DIR/"
|
| 21 |
speed = [i[2] for i in report]
|
| 22 |
print("llm zh average speed: ", sum(speed) / len(speed))
|
| 23 |
|
|
@@ -26,6 +26,6 @@ def test_llm_en(llm):
|
|
| 26 |
for src in read_translation("en", count_limit=None):
|
| 27 |
dst, time_cost = llm.translate(src, src_lang="en", dst_lang="zh")
|
| 28 |
report.append([src, dst, time_cost])
|
| 29 |
-
save_csv(REPORTS_DIR/"
|
| 30 |
speed = [i[2] for i in report]
|
| 31 |
print("llm en average speed: ", sum(speed) / len(speed))
|
|
|
|
| 17 |
for src in read_translation("zh", count_limit=None):
|
| 18 |
dst, time_cost = llm.translate(src, src_lang="zh", dst_lang="en")
|
| 19 |
report.append([src, dst, time_cost])
|
| 20 |
+
save_csv(REPORTS_DIR/"test_llm_zh.csv", ["src", "dst", "time"], report)
|
| 21 |
speed = [i[2] for i in report]
|
| 22 |
print("llm zh average speed: ", sum(speed) / len(speed))
|
| 23 |
|
|
|
|
| 26 |
for src in read_translation("en", count_limit=None):
|
| 27 |
dst, time_cost = llm.translate(src, src_lang="en", dst_lang="zh")
|
| 28 |
report.append([src, dst, time_cost])
|
| 29 |
+
save_csv(REPORTS_DIR/"test_llm_en.csv", ["src", "dst", "time"], report)
|
| 30 |
speed = [i[2] for i in report]
|
| 31 |
print("llm en average speed: ", sum(speed) / len(speed))
|
tests/test_app_models/test_tts.py
CHANGED
|
@@ -20,7 +20,7 @@ def test_tts_zh(tts):
|
|
| 20 |
for text in read_translation("zh", count_limit=20):
|
| 21 |
samples, sr, time_cost = tts.generate(text)
|
| 22 |
report.append([text, time_cost])
|
| 23 |
-
save_csv(REPORTS_DIR / "
|
| 24 |
speed = [i[1] for i in report]
|
| 25 |
print("tts zh average speed: ", sum(speed) / len(speed))
|
| 26 |
|
|
@@ -31,6 +31,6 @@ def test_tts_en(tts):
|
|
| 31 |
for text in read_translation("en",count_limit=20):
|
| 32 |
samples, sr, time_cost = tts.generate(text, speed=1.4)
|
| 33 |
report.append([text, time_cost])
|
| 34 |
-
save_csv(REPORTS_DIR / "
|
| 35 |
speed = [i[1] for i in report]
|
| 36 |
print("tts en average speed: ", sum(speed) / len(speed))
|
|
|
|
| 20 |
for text in read_translation("zh", count_limit=20):
|
| 21 |
samples, sr, time_cost = tts.generate(text)
|
| 22 |
report.append([text, time_cost])
|
| 23 |
+
save_csv(REPORTS_DIR / "test_tts_zh.csv", ["text", "time"], report)
|
| 24 |
speed = [i[1] for i in report]
|
| 25 |
print("tts zh average speed: ", sum(speed) / len(speed))
|
| 26 |
|
|
|
|
| 31 |
for text in read_translation("en",count_limit=20):
|
| 32 |
samples, sr, time_cost = tts.generate(text, speed=1.4)
|
| 33 |
report.append([text, time_cost])
|
| 34 |
+
save_csv(REPORTS_DIR / "test_tts_en.csv", ["text", "time"], report)
|
| 35 |
speed = [i[1] for i in report]
|
| 36 |
print("tts en average speed: ", sum(speed) / len(speed))
|
tests/test_app_models/test_whisper.py
CHANGED
|
@@ -12,7 +12,7 @@ def whisper(get_platform):
|
|
| 12 |
from lib.models.intel.whisper import WhisperOv
|
| 13 |
return WhisperOv()
|
| 14 |
|
| 15 |
-
def
|
| 16 |
#TODO: 测试CER
|
| 17 |
report = []
|
| 18 |
for audio_file, text, duration in read_emilia(count_limit=20):
|
|
@@ -20,6 +20,6 @@ def test_inference(whisper):
|
|
| 20 |
audio = read_audio(audio_file)
|
| 21 |
asr_text, time_cost = whisper.transcribe(audio, "zh")
|
| 22 |
report.append([audio_file,duration, text, asr_text, time_cost])
|
| 23 |
-
save_csv(REPORTS_DIR/"
|
| 24 |
speed = [i[4] for i in report]
|
| 25 |
print("Whisper average speed: ", sum(speed) / len(speed))
|
|
|
|
| 12 |
from lib.models.intel.whisper import WhisperOv
|
| 13 |
return WhisperOv()
|
| 14 |
|
| 15 |
+
def test_whisper(whisper):
|
| 16 |
#TODO: 测试CER
|
| 17 |
report = []
|
| 18 |
for audio_file, text, duration in read_emilia(count_limit=20):
|
|
|
|
| 20 |
audio = read_audio(audio_file)
|
| 21 |
asr_text, time_cost = whisper.transcribe(audio, "zh")
|
| 22 |
report.append([audio_file,duration, text, asr_text, time_cost])
|
| 23 |
+
save_csv(REPORTS_DIR/"test_whisper.csv", ["audio", "duration", "ref", "asr", "time"], report)
|
| 24 |
speed = [i[4] for i in report]
|
| 25 |
print("Whisper average speed: ", sum(speed) / len(speed))
|