From b5d3df75cf6462aa3bf42fd3c86fa2aa7f1c8a15 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 24 十一月 2023 00:54:44 +0800
Subject: [PATCH] setup jamo
---
setup.py | 2 +-
funasr/datasets/dataset_jsonl.py | 8 +++++---
2 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/funasr/datasets/dataset_jsonl.py b/funasr/datasets/dataset_jsonl.py
index ec2bf07..543b60e 100644
--- a/funasr/datasets/dataset_jsonl.py
+++ b/funasr/datasets/dataset_jsonl.py
@@ -4,8 +4,8 @@
import numpy as np
import kaldiio
import librosa
-
-
+import torchaudio
+import time
def load_audio(audio_path: str, fs: int=16000):
audio = None
@@ -17,7 +17,9 @@
if ".ark:" in audio_path:
audio = kaldiio.load_mat(audio_path)
else:
- audio, fs = librosa.load(audio_path, sr=fs)
+ # audio, fs = librosa.load(audio_path, sr=fs)
+ audio, fs = torchaudio.load(audio_path)
+ audio = audio[0, :]
return audio
def extract_features(data, date_type: str="sound", frontend=None):
diff --git a/setup.py b/setup.py
index 5b7b83c..63beb5b 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
"humanfriendly",
"scipy>=1.4.1",
"librosa",
- # "jamo", # For kss
+ "jamo", # For kss
"PyYAML>=5.1.2",
# "soundfile>=0.12.1",
# "h5py>=3.1.0",
--
Gitblit v1.9.1