From 28ccfbfc51068a663a80764e14074df5edf2b5ba Mon Sep 17 00:00:00 2001
From: kongdeqiang <kongdeqiang960204@163.com>
Date: 星期五, 13 三月 2026 17:41:41 +0800
Subject: [PATCH] 提交
---
funasr/utils/load_utils.py | 34 +++++++++++++++++++++++++++++-----
1 files changed, 29 insertions(+), 5 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index 9613d1b..d208f7d 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -10,7 +10,6 @@
import time
import logging
from torch.nn.utils.rnn import pad_sequence
-from pydub import AudioSegment
try:
from funasr.download.file import download_from_url
@@ -19,6 +18,11 @@
import pdb
import subprocess
from subprocess import CalledProcessError, run
+
+try:
+ from pydub import AudioSegment
+except:
+ pass
def is_ffmpeg_installed():
@@ -82,8 +86,10 @@
): # download url to local file
data_or_path_or_list = download_from_url(data_or_path_or_list)
- if isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list): # local file
+ if (isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list)) or hasattr(data_or_path_or_list, 'read'): # local file or bytes io
if data_type is None or data_type == "sound":
+ if hasattr(data_or_path_or_list, "read") and hasattr(data_or_path_or_list, "seek"):
+ data_or_path_or_list.seek(0)
# if use_ffmpeg:
# data_or_path_or_list = _load_audio_ffmpeg(data_or_path_or_list, sr=fs)
# data_or_path_or_list = torch.from_numpy(data_or_path_or_list).squeeze() # [n_samples,]
@@ -101,7 +107,8 @@
data_or_path_or_list
).squeeze() # [n_samples,]
elif data_type == "text" and tokenizer is not None:
- data_or_path_or_list = tokenizer.encode(data_or_path_or_list)
+ with open(data_or_path_or_list, "r") as f:
+ data_or_path_or_list = tokenizer.encode(f.read().strip())
elif data_type == "image": # undo
pass
elif data_type == "video": # undo
@@ -138,7 +145,10 @@
def load_bytes(input):
- # input = validate_frame_rate(input)
+ try:
+ input = validate_frame_rate(input)
+ except:
+ pass
middle_data = np.frombuffer(input, dtype=np.int16)
middle_data = np.asarray(middle_data)
if middle_data.dtype.kind not in "iu":
@@ -163,7 +173,12 @@
byte_data = BytesIO(input)
# 浣跨敤 pydub 鍔犺浇闊抽
- audio = AudioSegment.from_file(byte_data)
+ try:
+ audio = AudioSegment.from_file(byte_data)
+ except:
+ raise RuntimeError(
+ "You are decoding the pcm data, please install pydub first. via `pip install pydub`."
+ )
# 纭繚閲囨牱鐜囦负 16000 Hz
if audio.frame_rate != fs:
@@ -226,10 +241,19 @@
# This launches a subprocess to decode audio while down-mixing
# and resampling as necessary. Requires the ffmpeg CLI in PATH.
# fmt: off
+ pcm_params = []
+ if file.lower().endswith('.pcm'):
+ pcm_params = [
+ "-f", "s16le",
+ "-ar", str(sr),
+ "-ac", "1"
+ ]
+
cmd = [
"ffmpeg",
"-nostdin",
"-threads", "0",
+ *pcm_params, # PCM files need input format specified before -i since PCM is raw data without headers
"-i", file,
"-f", "s16le",
"-ac", "1",
--
Gitblit v1.9.1