diff --git a/runtime/python/onnxruntime/demo_paraformer_online.py b/runtime/python/onnxruntime/demo_paraformer_online.py index bb72b41a15..cca5357ab8 100644 --- a/runtime/python/onnxruntime/demo_paraformer_online.py +++ b/runtime/python/onnxruntime/demo_paraformer_online.py @@ -3,7 +3,8 @@ from pathlib import Path model_dir = "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online" -wav_path = ["{}/.cache/modelscope/hub/{}/example/asr_example.wav".format(Path.home(), model_dir)] +#wav_path = ["{}/.cache/modelscope/hub/{}/example/asr_example.wav".format(Path.home(), model_dir)] +wav_path = "{}/.cache/modelscope/hub/{}/example/asr_example.wav".format(Path.home(), model_dir) chunk_size = [5, 10, 5] model = Paraformer( diff --git a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py index 4f35fcc864..bde7f442e4 100644 --- a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py @@ -175,7 +175,23 @@ def plot_wave_timestamp(self, wav, text_timestamp, dest): plt.savefig(plotname, bbox_inches="tight") def load_data(self, wav_content: Union[str, np.ndarray, List[str]], fs: int = None) -> List: + def convert_to_wav(input_path, output_path): + from pydub import AudioSegment + try: + audio = AudioSegment.from_mp3(input_path) + audio.export(output_path, format="wav") + print("音频文件为mp3格式,已转换为wav格式") + + except Exception as e: + print(f"转换失败:{e}") + def load_wav(path: str) -> np.ndarray: + if not path.lower().endswith('.wav'): + import os + input_path = path + path = os.path.splitext(path)[0]+'.wav' + convert_to_wav(input_path,path) #将mp3格式转换成wav格式 + waveform, _ = librosa.load(path, sr=fs) return waveform diff --git a/runtime/python/onnxruntime/funasr_onnx/paraformer_online_bin.py b/runtime/python/onnxruntime/funasr_onnx/paraformer_online_bin.py index ddd857d378..3f63ea020f 100644 --- a/runtime/python/onnxruntime/funasr_onnx/paraformer_online_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/paraformer_online_bin.py @@ -208,7 +208,24 @@ def infer(self, feats: np.ndarray, feats_len: np.ndarray, cache): return asr_res def load_data(self, wav_content: Union[str, np.ndarray, List[str]], fs: int = None) -> List: + + def convert_to_wav(input_path, output_path): + from pydub import AudioSegment + try: + audio = AudioSegment.from_mp3(input_path) + audio.export(output_path, format="wav") + print("音频文件为mp3格式,已转换为wav格式") + + except Exception as e: + print(f"转换失败:{e}") + def load_wav(path: str) -> np.ndarray: + if not path.lower().endswith('.wav'): + import os + input_path = path + path = os.path.splitext(path)[0]+'.wav' + convert_to_wav(input_path,path) #将mp3格式转换成wav格式 + waveform, _ = librosa.load(path, sr=fs) return waveform diff --git a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py index 92928a8403..3f21004bfc 100644 --- a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py @@ -143,7 +143,24 @@ def __call__(self, audio_in: Union[str, np.ndarray, List[str]], **kwargs) -> Lis return segments def load_data(self, wav_content: Union[str, np.ndarray, List[str]], fs: int = None) -> List: + + def convert_to_wav(input_path, output_path): + from pydub import AudioSegment + try: + audio = AudioSegment.from_mp3(input_path) + audio.export(output_path, format="wav") + print("音频文件为mp3格式,已转换为wav格式") + + except Exception as e: + print(f"转换失败:{e}") + def load_wav(path: str) -> np.ndarray: + if not path.lower().endswith('.wav'): + import os + input_path = path + path = os.path.splitext(path)[0]+'.wav' + convert_to_wav(input_path,path) #将mp3格式转换成wav格式 + waveform, _ = librosa.load(path, sr=fs) return waveform