File size: 2,087 Bytes
3478b67 f3ecff1 3478b67 f3ecff1 d1ae526 3478b67 f3ecff1 d1ae526 3478b67 f3ecff1 3478b67 f3ecff1 d1ae526 3478b67 d1ae526 3478b67 f3ecff1 d1ae526 f3ecff1 d1ae526 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import os
import argparse
from SenseVoiceAx import SenseVoiceAx
import librosa
import numpy as np
import time
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--input", "-i", required=True, type=str, help="Input audio file"
)
parser.add_argument(
"--language",
"-l",
required=False,
type=str,
default="auto",
choices=["auto", "zh", "en", "yue", "ja", "ko"],
)
parser.add_argument("--streaming", action="store_true")
return parser.parse_args()
def main():
args = get_args()
input_audio = args.input
language = args.language
use_itn = True # 标点符号预测
if not args.streaming:
max_len = 256
model_path = os.path.join("sensevoice_ax650", "sensevoice.axmodel")
else:
max_len = 26
model_path = os.path.join("sensevoice_ax650", "streaming_sensevoice.axmodel")
assert os.path.exists(model_path), f"model {model_path} not exist"
print(f"input_audio: {input_audio}")
print(f"language: {language}")
print(f"use_itn: {use_itn}")
print(f"model_path: {model_path}")
print(f"streaming: {args.streaming}")
pipeline = SenseVoiceAx(
model_path,
max_len=max_len,
beam_size=3,
language="auto",
hot_words=None,
use_itn=True,
streaming=args.streaming,
)
if not args.streaming:
asr_res = pipeline.infer(input_audio, print_rtf=True)
print("ASR result: " + asr_res)
else:
samples, sr = librosa.load(input_audio, sr=16000)
samples = (samples * 32768).tolist()
duration = len(samples) / 16000
start = time.time()
step = int(0.1 * sr)
for i in range(0, len(samples), step):
is_last = i + step >= len(samples)
for res in pipeline.stream_infer(samples[i : i + step], is_last):
print(res)
end = time.time()
cost_time = end - start
print(f"RTF: {cost_time / duration}")
if __name__ == "__main__":
main()
|