yujuanqin commited on
Commit
e5b5f3e
·
1 Parent(s): 42742c6

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 test_inference(asr):
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/"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))
 
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/"translation_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,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/"translation_en.csv", ["src", "dst", "time"], report)
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 / "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,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 / "tts_en.csv", ["text", "time"], report)
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 test_inference(whisper):
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/"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))
 
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))