Upload tools.py

tools.py

@@ -2,6 +2,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib
 import librosa
+from scipy.io import wavfile
 from scipy.io.wavfile import write
 import torch
 
@@ -342,3 +343,22 @@ def decode_stft(encoded_D):
 
     D = magnitude * (np.cos(phase) + 1j * np.sin(phase))
     return D
+
+
+def read_wav_to_numpy(file_path):
+    """
+    读取 WAV 文件并返回采样率和音频数据 (NumPy 数组)
+
+    参数:
+    file_path (str): WAV 文件的路径
+
+    返回:
+    tuple: 采样率 (int) 和音频数据 (NumPy array)
+    """
+    # 使用 scipy.io.wavfile 读取 wav 文件
+    sample_rate, data = wavfile.read(file_path)
+    data = data/np.max(np.abs(data))
+
+    return sample_rate, data
+
+