From 0e622e694e6cb4459955f1e5942a7c53349ce640 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 19 十二月 2023 21:58:14 +0800
Subject: [PATCH] funasr2

---
 funasr/datasets/audio_datasets/load_audio_extract_fbank.py |   11 ++++++-----
 1 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/funasr/datasets/fun_datasets/load_audio_extract_fbank.py b/funasr/datasets/audio_datasets/load_audio_extract_fbank.py
similarity index 93%
rename from funasr/datasets/fun_datasets/load_audio_extract_fbank.py
rename to funasr/datasets/audio_datasets/load_audio_extract_fbank.py
index c76f346..c8883ee 100644
--- a/funasr/datasets/fun_datasets/load_audio_extract_fbank.py
+++ b/funasr/datasets/audio_datasets/load_audio_extract_fbank.py
@@ -46,15 +46,16 @@
 	array = np.frombuffer((middle_data.astype(dtype) - offset) / abs_max, dtype=np.float32)
 	return array
 
-def extract_fbank(data, data_len = None, date_type: str="sound", frontend=None):
-	
+def extract_fbank(data, data_len = None, data_type: str="sound", frontend=None):
+	# import pdb;
+	# pdb.set_trace()
 	if isinstance(data, np.ndarray):
 		data = torch.from_numpy(data)
-		if len(data) < 2:
+		if len(data.shape) < 2:
 			data = data[None, :] # data: [batch, N]
 		data_len = [data.shape[1]] if data_len is None else data_len
 	elif isinstance(data, torch.Tensor):
-		if len(data) < 2:
+		if len(data.shape) < 2:
 			data = data[None, :] # data: [batch, N]
 		data_len = [data.shape[1]] if data_len is None else data_len
 	elif isinstance(data, (list, tuple)):
@@ -67,7 +68,7 @@
 		data = pad_sequence(data_list, batch_first=True) # data: [batch, N]
 	# import pdb;
 	# pdb.set_trace()
-	if date_type == "sound":
+	if data_type == "sound":
 		data, data_len = frontend(data, data_len)
 	
 	if isinstance(data_len, (list, tuple)):

--
Gitblit v1.9.1