From fa56f36921c6bcb4608a28ab76686822033b728e Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期四, 25 一月 2024 18:48:48 +0800
Subject: [PATCH] Update demo.py
---
funasr/datasets/audio_datasets/datasets.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/datasets/audio_datasets/datasets.py b/funasr/datasets/audio_datasets/datasets.py
index edf127f..ebb72a3 100644
--- a/funasr/datasets/audio_datasets/datasets.py
+++ b/funasr/datasets/audio_datasets/datasets.py
@@ -1,7 +1,7 @@
import torch
from funasr.register import tables
-from funasr.utils.load_utils import extract_fbank
+from funasr.utils.load_utils import extract_fbank, load_audio_text_image_video
@tables.register("dataset_classes", "AudioDataset")
@@ -55,10 +55,10 @@
# import pdb;
# pdb.set_trace()
source = item["source"]
- data_src = load_audio(source, fs=self.fs)
+ data_src = load_audio_text_image_video(source, fs=self.fs)
if self.preprocessor_speech:
data_src = self.preprocessor_speech(data_src)
- speech, speech_lengths = extract_fbank(data_src, data_type=self.data_type, frontend=self.frontend) # speech: [b, T, d]
+ speech, speech_lengths = extract_fbank(data_src, data_type=self.data_type, frontend=self.frontend, is_final=True) # speech: [b, T, d]
target = item["target"]
if self.preprocessor_text:
--
Gitblit v1.9.1