neverlib 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- neverlib/.history/Docs/audio_aug/test_snr_20250806011311.py +0 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011331.py +75 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011342.py +57 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011352.py +57 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011403.py +57 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011413.py +57 -0
- neverlib/.history/Docs/audio_aug/test_snr_20250806011435.py +55 -0
- neverlib/.history/Docs/vad/1_20250810032405.py +0 -0
- neverlib/.history/Docs/vad/1_20250810032417.py +39 -0
- neverlib/.history/audio_aug/audio_aug_20250806010451.py +125 -0
- neverlib/.history/audio_aug/audio_aug_20250806010750.py +138 -0
- neverlib/.history/audio_aug/audio_aug_20250806010759.py +140 -0
- neverlib/.history/audio_aug/audio_aug_20250806010803.py +140 -0
- neverlib/.history/audio_aug/audio_aug_20250806010809.py +140 -0
- neverlib/.history/audio_aug/audio_aug_20250806011108.py +140 -0
- neverlib/.history/dataAnalyze/__init___20250806204125.py +14 -0
- neverlib/.history/dataAnalyze/__init___20250806204139.py +14 -0
- neverlib/.history/dataAnalyze/__init___20250806204159.py +14 -0
- neverlib/.history/filter/__init___20250820103351.py +70 -0
- neverlib/.history/filter/__init___20250821102348.py +70 -0
- neverlib/.history/filter/__init___20250821102405.py +14 -0
- neverlib/.history/filter/auto_eq/__init___20250819213121.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102241.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102259.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102307.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102310.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102318.py +36 -0
- neverlib/.history/filter/auto_eq/__init___20250821102507.py +36 -0
- neverlib/{filter/AudoEQ/auto_eq_de.py → .history/filter/auto_eq/de_eq_20250820103848.py} +1 -1
- neverlib/.history/filter/auto_eq/de_eq_20250821102422.py +360 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820140732.py +75 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820140745.py +75 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820140816.py +75 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820140938.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141003.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141006.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141019.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141049.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141211.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141227.py +77 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141311.py +78 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141340.py +78 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141712.py +78 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141733.py +78 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250820141755.py +78 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250821102434.py +76 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250821102500.py +76 -0
- neverlib/.history/filter/auto_eq/freq_eq_20250821102502.py +76 -0
- neverlib/{filter/AudoEQ/auto_eq_ga_basic.py → .history/filter/auto_eq/ga_eq_basic_20250820102957.py} +1 -1
- neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113054.py +380 -0
- neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113150.py +380 -0
- neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113520.py +385 -0
- neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113525.py +385 -0
- neverlib/.history/filter/auto_eq/ga_eq_basic_20250821102212.py +385 -0
- neverlib/.history/metrics/dnsmos_20250806001612.py +160 -0
- neverlib/.history/metrics/dnsmos_20250815180659.py +160 -0
- neverlib/.history/metrics/dnsmos_20250815180701.py +158 -0
- neverlib/.history/metrics/dnsmos_20250815181321.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181327.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181331.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181620.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181631.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181742.py +154 -0
- neverlib/.history/metrics/dnsmos_20250815181824.py +153 -0
- neverlib/.history/metrics/dnsmos_20250815181834.py +153 -0
- neverlib/.history/metrics/dnsmos_20250815181922.py +153 -0
- neverlib/.history/metrics/dnsmos_20250815182011.py +147 -0
- neverlib/.history/metrics/dnsmos_20250815182036.py +144 -0
- neverlib/.history/metrics/dnsmos_20250815182936.py +143 -0
- neverlib/.history/metrics/dnsmos_20250815182942.py +143 -0
- neverlib/.history/metrics/dnsmos_20250815183032.py +137 -0
- neverlib/.history/metrics/dnsmos_20250815183101.py +144 -0
- neverlib/.history/metrics/dnsmos_20250815183121.py +144 -0
- neverlib/.history/metrics/dnsmos_20250815183123.py +143 -0
- neverlib/.history/metrics/dnsmos_20250815183214.py +143 -0
- neverlib/.history/metrics/dnsmos_20250815183240.py +143 -0
- neverlib/.history/metrics/dnsmos_20250815183248.py +144 -0
- neverlib/.history/metrics/dnsmos_20250815183407.py +142 -0
- neverlib/.history/metrics/dnsmos_20250815183409.py +142 -0
- neverlib/.history/metrics/dnsmos_20250815183431.py +142 -0
- neverlib/.history/metrics/dnsmos_20250815183507.py +140 -0
- neverlib/.history/metrics/dnsmos_20250815183513.py +139 -0
- neverlib/.history/metrics/dnsmos_20250815183618.py +139 -0
- neverlib/.history/metrics/dnsmos_20250815183709.py +140 -0
- neverlib/.history/metrics/dnsmos_20250815183756.py +137 -0
- neverlib/.history/metrics/dnsmos_20250815183815.py +128 -0
- neverlib/.history/metrics/dnsmos_20250815183827.py +129 -0
- neverlib/.history/metrics/dnsmos_20250815183913.py +117 -0
- neverlib/.history/metrics/dnsmos_20250815183914.py +117 -0
- neverlib/.history/metrics/dnsmos_20250815184003.py +118 -0
- neverlib/.history/metrics/dnsmos_20250815184040.py +118 -0
- neverlib/.history/metrics/dnsmos_20250815184049.py +118 -0
- neverlib/.history/metrics/dnsmos_20250815184104.py +117 -0
- neverlib/.history/metrics/dnsmos_20250815184200.py +117 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816015944.py +128 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020142.py +128 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020156.py +128 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020554.py +130 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020600.py +125 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020631.py +120 -0
- neverlib/.history/metrics/lpc_lsp_metric_20250816020746.py +118 -0
- neverlib/.history/metrics/lpc_me_20250816013111.py +0 -0
- neverlib/.history/metrics/lpc_me_20250816013129.py +121 -0
- neverlib/.history/metrics/lpc_me_20250816015430.py +103 -0
- neverlib/.history/metrics/lpc_me_20250816015535.py +96 -0
- neverlib/.history/metrics/lpc_me_20250816015542.py +96 -0
- neverlib/.history/metrics/lpc_me_20250816015636.py +97 -0
- neverlib/.history/metrics/lpc_me_20250816015658.py +104 -0
- neverlib/.history/metrics/lpc_me_20250816015703.py +100 -0
- neverlib/.history/metrics/lpc_me_20250816015945.py +128 -0
- neverlib/.history/metrics/snr_20250806010538.py +177 -0
- neverlib/.history/metrics/snr_20250806211634.py +184 -0
- neverlib/.history/metrics/spec_20250805234209.py +45 -0
- neverlib/.history/metrics/spec_20250816135530.py +11 -0
- neverlib/.history/metrics/spec_20250816135654.py +16 -0
- neverlib/.history/metrics/spec_20250816135736.py +68 -0
- neverlib/.history/metrics/spec_20250816135904.py +75 -0
- neverlib/.history/metrics/spec_20250816135921.py +82 -0
- neverlib/.history/metrics/spec_20250816140111.py +82 -0
- neverlib/.history/metrics/spec_20250816140543.py +136 -0
- neverlib/.history/metrics/spec_20250816140559.py +172 -0
- neverlib/.history/metrics/spec_20250816140602.py +172 -0
- neverlib/.history/metrics/spec_20250816140608.py +172 -0
- neverlib/.history/metrics/spec_20250816140654.py +148 -0
- neverlib/.history/metrics/spec_20250816140705.py +144 -0
- neverlib/.history/metrics/spec_20250816140755.py +138 -0
- neverlib/.history/metrics/spec_20250816140823.py +170 -0
- neverlib/.history/metrics/spec_20250816140832.py +170 -0
- neverlib/.history/metrics/spec_20250816140833.py +170 -0
- neverlib/.history/metrics/spec_20250816140922.py +147 -0
- neverlib/.history/metrics/spec_20250816141148.py +107 -0
- neverlib/.history/metrics/spec_20250816141219.py +123 -0
- neverlib/.history/metrics/spec_20250816141732.py +178 -0
- neverlib/.history/metrics/spec_20250816141740.py +178 -0
- neverlib/.history/metrics/spec_20250816142030.py +178 -0
- neverlib/.history/metrics/spec_20250816142107.py +135 -0
- neverlib/.history/metrics/spec_20250816142126.py +135 -0
- neverlib/.history/metrics/spec_20250816142410.py +135 -0
- neverlib/.history/metrics/spec_20250816142415.py +136 -0
- neverlib/.history/metrics/spec_metric_20250816135156.py +0 -0
- neverlib/.history/metrics/spec_metric_20250816135226.py +5 -0
- neverlib/.history/metrics/spec_metric_20250816135227.py +10 -0
- neverlib/.history/metrics/spec_metric_20250816135306.py +15 -0
- neverlib/.history/metrics/spec_metric_20250816135442.py +31 -0
- neverlib/.history/metrics/spec_metric_20250816135448.py +31 -0
- neverlib/.history/metrics/spec_metric_20250816135520.py +29 -0
- neverlib/.history/metrics/spec_metric_20250816135537.py +63 -0
- neverlib/.history/metrics/spec_metric_20250816135653.py +65 -0
- neverlib/.history/vad/PreProcess_20250805234211.py +63 -0
- neverlib/.history/vad/PreProcess_20250809232455.py +63 -0
- neverlib/.history/vad/PreProcess_20250816020725.py +66 -0
- neverlib/.history/vad/VAD_Silero_20250805234211.py +50 -0
- neverlib/.history/vad/VAD_Silero_20250809232456.py +50 -0
- neverlib/.history/vad/VAD_WebRTC_20250805234211.py +61 -0
- neverlib/.history/vad/VAD_WebRTC_20250809232456.py +61 -0
- neverlib/.history/vad/VAD_funasr_20250805234211.py +54 -0
- neverlib/.history/vad/VAD_funasr_20250809232456.py +54 -0
- neverlib/.history/vad/VAD_vadlib_20250805234211.py +70 -0
- neverlib/.history/vad/VAD_vadlib_20250809232455.py +70 -0
- neverlib/.history/vad/VAD_whisper_20250805234211.py +55 -0
- neverlib/.history/vad/VAD_whisper_20250809232456.py +55 -0
- neverlib/.specstory/.what-is-this.md +69 -0
- neverlib/.specstory/history/2025-08-05_17-06Z-/350/277/231/344/270/200/346/255/245/347/232/204/347/233/256/347/232/204/346/230/257/344/273/200/344/271/210.md +424 -0
- neverlib/Docs/audio_aug/test_snr.py +55 -0
- neverlib/audio_aug/HarmonicDistortion.py +79 -0
- neverlib/audio_aug/TFDrop.py +41 -0
- neverlib/audio_aug/TFMask.py +56 -0
- neverlib/audio_aug/audio_aug.py +16 -1
- neverlib/audio_aug/clip_aug.py +41 -0
- neverlib/audio_aug/coder_aug.py +209 -0
- neverlib/audio_aug/coder_aug2.py +118 -0
- neverlib/audio_aug/loss_packet_aug.py +103 -0
- neverlib/audio_aug/quant_aug.py +78 -0
- neverlib/data_analyze/__init__.py +14 -0
- neverlib/filter/auto_eq/__init__.py +36 -0
- neverlib/filter/auto_eq/de_eq.py +360 -0
- neverlib/filter/auto_eq/freq_eq.py +76 -0
- neverlib/filter/{AudoEQ/auto_eq_ga_advanced.py → auto_eq/ga_eq_advanced.py} +1 -1
- neverlib/filter/auto_eq/ga_eq_basic.py +385 -0
- neverlib/metrics/dnsmos.py +58 -101
- neverlib/metrics/lpc_lsp.py +118 -0
- neverlib/metrics/snr.py +11 -4
- neverlib/metrics/spec.py +136 -45
- neverlib/utils/utils.py +17 -14
- neverlib/vad/PreProcess.py +5 -2
- neverlib/vad/VAD_Silero.py +1 -1
- neverlib/vad/VAD_WebRTC.py +1 -1
- neverlib/vad/VAD_funasr.py +1 -1
- neverlib/vad/VAD_vadlib.py +1 -1
- neverlib/vad/VAD_whisper.py +1 -1
- {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/METADATA +1 -1
- neverlib-0.2.4.dist-info/RECORD +229 -0
- neverlib-0.2.3.dist-info/RECORD +0 -53
- /neverlib/{dataAnalyze/__init__.py → .history/dataAnalyze/__init___20250805234204.py} +0 -0
- /neverlib/{filter/AudoEQ/auto_eq_spectral_direct.py → .history/filter/auto_eq/freq_eq_20250805234206.py} +0 -0
- /neverlib/{dataAnalyze → data_analyze}/README.md +0 -0
- /neverlib/{dataAnalyze → data_analyze}/dataset_analyzer.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/quality_metrics.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/rms_distrubution.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/spectral_analysis.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/statistics.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/temporal_features.py +0 -0
- /neverlib/{dataAnalyze → data_analyze}/visualization.py +0 -0
- /neverlib/filter/{AudoEQ → auto_eq}/README.md +0 -0
- {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/WHEEL +0 -0
- {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/licenses/LICENSE +0 -0
- {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,148 @@
|
|
|
1
|
+
'''
|
|
2
|
+
Author: 凌逆战 | Never
|
|
3
|
+
Date: 2025-08-16 13:51:57
|
|
4
|
+
Description: 频域客观度量指标
|
|
5
|
+
'''
|
|
6
|
+
|
|
7
|
+
import librosa
|
|
8
|
+
import numpy as np
|
|
9
|
+
import soundfile as sf
|
|
10
|
+
from neverlib.utils import EPS
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
def sd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
14
|
+
"""
|
|
15
|
+
计算两个音频信号之间的频谱距离 (Spectral Distance)。
|
|
16
|
+
该指标衡量两个信号在频域上的差异程度。
|
|
17
|
+
|
|
18
|
+
Args:
|
|
19
|
+
ref_wav (np.ndarray): 参考音频信号 (一维数组)
|
|
20
|
+
test_wav (np.ndarray): 测试音频信号 (一维数组)
|
|
21
|
+
n_fft (int): FFT点数,决定频率分辨率,默认为2048
|
|
22
|
+
hop_length (int): 帧移,决定时间分辨率,默认为512
|
|
23
|
+
win_length (int, optional): 窗长,如果为None则默认为n_fft
|
|
24
|
+
|
|
25
|
+
Returns:
|
|
26
|
+
float: 频谱距离值,值越小表示两个信号越相似
|
|
27
|
+
|
|
28
|
+
Raises:
|
|
29
|
+
ValueError: 当输入参数无效时
|
|
30
|
+
AssertionError: 当输入信号维度不正确时
|
|
31
|
+
"""
|
|
32
|
+
assert len(ref_wav) == len(test_wav), "输入信号长度必须相同"
|
|
33
|
+
|
|
34
|
+
# 设置默认窗长
|
|
35
|
+
if win_length is None:
|
|
36
|
+
win_length = n_fft
|
|
37
|
+
|
|
38
|
+
# 计算短时傅里叶变换
|
|
39
|
+
ref_spec = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
40
|
+
test_spec = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
41
|
+
|
|
42
|
+
# 计算频谱距离:均方根误差
|
|
43
|
+
spec_diff = ref_spec - test_spec
|
|
44
|
+
squared_diff = np.abs(spec_diff) ** 2
|
|
45
|
+
mean_squared_diff = np.mean(squared_diff)
|
|
46
|
+
sd_value = np.sqrt(mean_squared_diff)
|
|
47
|
+
|
|
48
|
+
return sd_value
|
|
49
|
+
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
54
|
+
"""
|
|
55
|
+
计算两个一维音频信号之间的对数谱距离 (Log-Spectral Distance, LSD)。
|
|
56
|
+
该实现遵循标准的LSD定义: 整体均方根误差。
|
|
57
|
+
|
|
58
|
+
Args:
|
|
59
|
+
ref_wav (np.ndarray): 原始的、干净的参考信号 (一维数组)。
|
|
60
|
+
test_wav (np.ndarray): 模型估计或处理后的信号 (一维数组)。
|
|
61
|
+
n_fft (int): FFT点数, 决定了频率分辨率。
|
|
62
|
+
hop_length (int): 帧移, 决定了时间分辨率。
|
|
63
|
+
win_length (int, optional): 窗长。如果为None, 则默认为n_fft。
|
|
64
|
+
epsilon (float): 一个非常小的数值, 用于防止对零取对数, 保证数值稳定性。
|
|
65
|
+
|
|
66
|
+
Returns:
|
|
67
|
+
float: 对数谱距离值, 单位为分贝 (dB)。
|
|
68
|
+
"""
|
|
69
|
+
assert ref_wav.ndim == 1 and test_wav.ndim == 1, "输入信号必须是一维数组。"
|
|
70
|
+
|
|
71
|
+
if win_length is None:
|
|
72
|
+
win_length = n_fft
|
|
73
|
+
|
|
74
|
+
ref_stft = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
75
|
+
test_stft = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
76
|
+
|
|
77
|
+
ref_power_spec = np.abs(ref_stft) ** 2 # (F,T)
|
|
78
|
+
test_power_spec = np.abs(test_stft) ** 2 # (F,T)
|
|
79
|
+
|
|
80
|
+
ref_log_power_spec = 10 * np.log10(ref_power_spec + EPS)
|
|
81
|
+
test_log_power_spec = 10 * np.log10(test_power_spec + EPS)
|
|
82
|
+
|
|
83
|
+
squared_error = (ref_log_power_spec - test_log_power_spec) ** 2
|
|
84
|
+
lsd_val = np.sqrt(np.mean(squared_error))
|
|
85
|
+
|
|
86
|
+
return lsd_val
|
|
87
|
+
|
|
88
|
+
def mcd(ref_wav, test_wav, sr=16000):
|
|
89
|
+
"""
|
|
90
|
+
梅尔倒谱距离 Mel-Cepstral Distance
|
|
91
|
+
ref_spec: 参考频谱
|
|
92
|
+
test_spec: 测试频谱
|
|
93
|
+
"""
|
|
94
|
+
ref_wav, ref_sr = sf.read(ref_wav)
|
|
95
|
+
test_wav, test_sr = sf.read(test_wav)
|
|
96
|
+
assert ref_sr == test_sr == sr, "采样率必须为16000Hz"
|
|
97
|
+
assert len(ref_wav) == len(test_wav), "音频长度必须相同"
|
|
98
|
+
|
|
99
|
+
ref_mfcc = librosa.feature.mfcc(y=ref_wav, sr=sr)
|
|
100
|
+
test_mfcc = librosa.feature.mfcc(y=test_wav, sr=sr)
|
|
101
|
+
|
|
102
|
+
# 计算 MCD (跳过 0 阶)
|
|
103
|
+
diff = ref_mfcc[1:] - test_mfcc[1:]
|
|
104
|
+
mcd = (10.0 / np.log(10)) * np.sqrt(2 * np.mean(np.sum(diff ** 2, axis=0)))
|
|
105
|
+
return mcd
|
|
106
|
+
|
|
107
|
+
if __name__ == "__main__":
|
|
108
|
+
# 测试sd函数
|
|
109
|
+
print("=== 测试频谱距离 (SD) 函数 ===")
|
|
110
|
+
|
|
111
|
+
# 创建测试音频数据
|
|
112
|
+
sr = 16000
|
|
113
|
+
duration = 1.0 # 1秒
|
|
114
|
+
t = np.linspace(0, duration, int(sr * duration), False)
|
|
115
|
+
|
|
116
|
+
# 参考信号:正弦波
|
|
117
|
+
ref_wav = np.sin(2 * np.pi * 440 * t) # 440Hz
|
|
118
|
+
|
|
119
|
+
# 测试信号1:相同的正弦波(应该距离为0)
|
|
120
|
+
test_wav1 = np.sin(2 * np.pi * 440 * t)
|
|
121
|
+
|
|
122
|
+
# 测试信号2:不同频率的正弦波
|
|
123
|
+
test_wav2 = np.sin(2 * np.pi * 880 * t) # 880Hz
|
|
124
|
+
|
|
125
|
+
# 测试信号3:添加噪声
|
|
126
|
+
test_wav3 = ref_wav + 0.1 * np.random.randn(len(ref_wav))
|
|
127
|
+
|
|
128
|
+
try:
|
|
129
|
+
# 测试相同信号
|
|
130
|
+
sd1 = sd(ref_wav, test_wav1)
|
|
131
|
+
print(f"相同信号之间的SD: {sd1:.6f} (应该接近0)")
|
|
132
|
+
|
|
133
|
+
# 测试不同频率信号
|
|
134
|
+
sd2 = sd(ref_wav, test_wav2)
|
|
135
|
+
print(f"不同频率信号之间的SD: {sd2:.6f}")
|
|
136
|
+
|
|
137
|
+
# 测试带噪声信号
|
|
138
|
+
sd3 = sd(ref_wav, test_wav3)
|
|
139
|
+
print(f"带噪声信号之间的SD: {sd3:.6f}")
|
|
140
|
+
|
|
141
|
+
# 测试不同参数
|
|
142
|
+
sd4 = sd(ref_wav, test_wav3, n_fft=1024, hop_length=256)
|
|
143
|
+
print(f"使用不同参数的SD: {sd4:.6f}")
|
|
144
|
+
|
|
145
|
+
except Exception as e:
|
|
146
|
+
print(f"测试过程中发生错误: {e}")
|
|
147
|
+
|
|
148
|
+
print("\n=== 测试完成 ===")
|
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
'''
|
|
2
|
+
Author: 凌逆战 | Never
|
|
3
|
+
Date: 2025-08-16 13:51:57
|
|
4
|
+
Description: 频域客观度量指标
|
|
5
|
+
'''
|
|
6
|
+
|
|
7
|
+
import librosa
|
|
8
|
+
import numpy as np
|
|
9
|
+
import soundfile as sf
|
|
10
|
+
from neverlib.utils import EPS
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
def sd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
14
|
+
"""
|
|
15
|
+
计算两个音频信号之间的频谱距离 (Spectral Distance)。
|
|
16
|
+
该指标衡量两个信号在频域上的差异程度。
|
|
17
|
+
|
|
18
|
+
Args:
|
|
19
|
+
ref_wav (np.ndarray): 参考音频信号 (一维数组)
|
|
20
|
+
test_wav (np.ndarray): 测试音频信号 (一维数组)
|
|
21
|
+
n_fft (int): FFT点数,决定频率分辨率,默认为2048
|
|
22
|
+
hop_length (int): 帧移,决定时间分辨率,默认为512
|
|
23
|
+
win_length (int, optional): 窗长,如果为None则默认为n_fft
|
|
24
|
+
|
|
25
|
+
Returns:
|
|
26
|
+
float: 频谱距离值,值越小表示两个信号越相似
|
|
27
|
+
|
|
28
|
+
Raises:
|
|
29
|
+
ValueError: 当输入参数无效时
|
|
30
|
+
AssertionError: 当输入信号维度不正确时
|
|
31
|
+
"""
|
|
32
|
+
assert len(ref_wav) == len(test_wav), "输入信号长度必须相同"
|
|
33
|
+
|
|
34
|
+
# 计算短时傅里叶变换
|
|
35
|
+
ref_spec = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
36
|
+
test_spec = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
37
|
+
|
|
38
|
+
# 计算频谱距离:均方根误差
|
|
39
|
+
spec_diff = ref_spec - test_spec
|
|
40
|
+
squared_diff = np.abs(spec_diff) ** 2
|
|
41
|
+
mean_squared_diff = np.mean(squared_diff)
|
|
42
|
+
sd_value = np.sqrt(mean_squared_diff)
|
|
43
|
+
|
|
44
|
+
return sd_value
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
50
|
+
"""
|
|
51
|
+
计算两个一维音频信号之间的对数谱距离 (Log-Spectral Distance, LSD)。
|
|
52
|
+
该实现遵循标准的LSD定义: 整体均方根误差。
|
|
53
|
+
|
|
54
|
+
Args:
|
|
55
|
+
ref_wav (np.ndarray): 原始的、干净的参考信号 (一维数组)。
|
|
56
|
+
test_wav (np.ndarray): 模型估计或处理后的信号 (一维数组)。
|
|
57
|
+
n_fft (int): FFT点数, 决定了频率分辨率。
|
|
58
|
+
hop_length (int): 帧移, 决定了时间分辨率。
|
|
59
|
+
win_length (int, optional): 窗长。如果为None, 则默认为n_fft。
|
|
60
|
+
epsilon (float): 一个非常小的数值, 用于防止对零取对数, 保证数值稳定性。
|
|
61
|
+
|
|
62
|
+
Returns:
|
|
63
|
+
float: 对数谱距离值, 单位为分贝 (dB)。
|
|
64
|
+
"""
|
|
65
|
+
assert ref_wav.ndim == 1 and test_wav.ndim == 1, "输入信号必须是一维数组。"
|
|
66
|
+
|
|
67
|
+
if win_length is None:
|
|
68
|
+
win_length = n_fft
|
|
69
|
+
|
|
70
|
+
ref_stft = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
71
|
+
test_stft = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
72
|
+
|
|
73
|
+
ref_power_spec = np.abs(ref_stft) ** 2 # (F,T)
|
|
74
|
+
test_power_spec = np.abs(test_stft) ** 2 # (F,T)
|
|
75
|
+
|
|
76
|
+
ref_log_power_spec = 10 * np.log10(ref_power_spec + EPS)
|
|
77
|
+
test_log_power_spec = 10 * np.log10(test_power_spec + EPS)
|
|
78
|
+
|
|
79
|
+
squared_error = (ref_log_power_spec - test_log_power_spec) ** 2
|
|
80
|
+
lsd_val = np.sqrt(np.mean(squared_error))
|
|
81
|
+
|
|
82
|
+
return lsd_val
|
|
83
|
+
|
|
84
|
+
def mcd(ref_wav, test_wav, sr=16000):
|
|
85
|
+
"""
|
|
86
|
+
梅尔倒谱距离 Mel-Cepstral Distance
|
|
87
|
+
ref_spec: 参考频谱
|
|
88
|
+
test_spec: 测试频谱
|
|
89
|
+
"""
|
|
90
|
+
ref_wav, ref_sr = sf.read(ref_wav)
|
|
91
|
+
test_wav, test_sr = sf.read(test_wav)
|
|
92
|
+
assert ref_sr == test_sr == sr, "采样率必须为16000Hz"
|
|
93
|
+
assert len(ref_wav) == len(test_wav), "音频长度必须相同"
|
|
94
|
+
|
|
95
|
+
ref_mfcc = librosa.feature.mfcc(y=ref_wav, sr=sr)
|
|
96
|
+
test_mfcc = librosa.feature.mfcc(y=test_wav, sr=sr)
|
|
97
|
+
|
|
98
|
+
# 计算 MCD (跳过 0 阶)
|
|
99
|
+
diff = ref_mfcc[1:] - test_mfcc[1:]
|
|
100
|
+
mcd = (10.0 / np.log(10)) * np.sqrt(2 * np.mean(np.sum(diff ** 2, axis=0)))
|
|
101
|
+
return mcd
|
|
102
|
+
|
|
103
|
+
if __name__ == "__main__":
|
|
104
|
+
# 测试sd函数
|
|
105
|
+
print("=== 测试频谱距离 (SD) 函数 ===")
|
|
106
|
+
|
|
107
|
+
# 创建测试音频数据
|
|
108
|
+
sr = 16000
|
|
109
|
+
duration = 1.0 # 1秒
|
|
110
|
+
t = np.linspace(0, duration, int(sr * duration), False)
|
|
111
|
+
|
|
112
|
+
# 参考信号:正弦波
|
|
113
|
+
ref_wav = np.sin(2 * np.pi * 440 * t) # 440Hz
|
|
114
|
+
|
|
115
|
+
# 测试信号1:相同的正弦波(应该距离为0)
|
|
116
|
+
test_wav1 = np.sin(2 * np.pi * 440 * t)
|
|
117
|
+
|
|
118
|
+
# 测试信号2:不同频率的正弦波
|
|
119
|
+
test_wav2 = np.sin(2 * np.pi * 880 * t) # 880Hz
|
|
120
|
+
|
|
121
|
+
# 测试信号3:添加噪声
|
|
122
|
+
test_wav3 = ref_wav + 0.1 * np.random.randn(len(ref_wav))
|
|
123
|
+
|
|
124
|
+
try:
|
|
125
|
+
# 测试相同信号
|
|
126
|
+
sd1 = sd(ref_wav, test_wav1)
|
|
127
|
+
print(f"相同信号之间的SD: {sd1:.6f} (应该接近0)")
|
|
128
|
+
|
|
129
|
+
# 测试不同频率信号
|
|
130
|
+
sd2 = sd(ref_wav, test_wav2)
|
|
131
|
+
print(f"不同频率信号之间的SD: {sd2:.6f}")
|
|
132
|
+
|
|
133
|
+
# 测试带噪声信号
|
|
134
|
+
sd3 = sd(ref_wav, test_wav3)
|
|
135
|
+
print(f"带噪声信号之间的SD: {sd3:.6f}")
|
|
136
|
+
|
|
137
|
+
# 测试不同参数
|
|
138
|
+
sd4 = sd(ref_wav, test_wav3, n_fft=1024, hop_length=256)
|
|
139
|
+
print(f"使用不同参数的SD: {sd4:.6f}")
|
|
140
|
+
|
|
141
|
+
except Exception as e:
|
|
142
|
+
print(f"测试过程中发生错误: {e}")
|
|
143
|
+
|
|
144
|
+
print("\n=== 测试完成 ===")
|
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
'''
|
|
2
|
+
Author: 凌逆战 | Never
|
|
3
|
+
Date: 2025-08-16 13:51:57
|
|
4
|
+
Description: 频域客观度量指标
|
|
5
|
+
'''
|
|
6
|
+
|
|
7
|
+
import librosa
|
|
8
|
+
import numpy as np
|
|
9
|
+
import soundfile as sf
|
|
10
|
+
from neverlib.utils import EPS
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
def sd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
14
|
+
"""
|
|
15
|
+
计算两个音频信号之间的频谱距离 (Spectral Distance)。
|
|
16
|
+
该指标衡量两个信号在频域上的差异程度。
|
|
17
|
+
Args:
|
|
18
|
+
ref_wav (np.ndarray): 参考音频信号 (一维数组)
|
|
19
|
+
test_wav (np.ndarray): 测试音频信号 (一维数组)
|
|
20
|
+
n_fft (int): FFT点数,决定频率分辨率,默认为2048
|
|
21
|
+
hop_length (int): 帧移,决定时间分辨率,默认为512
|
|
22
|
+
win_length (int, optional): 窗长,如果为None则默认为n_fft
|
|
23
|
+
Returns:
|
|
24
|
+
float: 频谱距离值,值越小表示两个信号越相似
|
|
25
|
+
"""
|
|
26
|
+
assert len(ref_wav) == len(test_wav), "输入信号长度必须相同"
|
|
27
|
+
|
|
28
|
+
# 计算短时傅里叶变换
|
|
29
|
+
ref_spec = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
30
|
+
test_spec = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
31
|
+
|
|
32
|
+
# 计算频谱距离:均方根误差
|
|
33
|
+
spec_diff = ref_spec - test_spec
|
|
34
|
+
squared_diff = np.abs(spec_diff) ** 2
|
|
35
|
+
mean_squared_diff = np.mean(squared_diff)
|
|
36
|
+
sd_value = np.sqrt(mean_squared_diff)
|
|
37
|
+
|
|
38
|
+
return sd_value
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
44
|
+
"""
|
|
45
|
+
计算两个一维音频信号之间的对数谱距离 (Log-Spectral Distance, LSD)。
|
|
46
|
+
该实现遵循标准的LSD定义: 整体均方根误差。
|
|
47
|
+
|
|
48
|
+
Args:
|
|
49
|
+
ref_wav (np.ndarray): 原始的、干净的参考信号 (一维数组)。
|
|
50
|
+
test_wav (np.ndarray): 模型估计或处理后的信号 (一维数组)。
|
|
51
|
+
n_fft (int): FFT点数, 决定了频率分辨率。
|
|
52
|
+
hop_length (int): 帧移, 决定了时间分辨率。
|
|
53
|
+
win_length (int, optional): 窗长。如果为None, 则默认为n_fft。
|
|
54
|
+
epsilon (float): 一个非常小的数值, 用于防止对零取对数, 保证数值稳定性。
|
|
55
|
+
|
|
56
|
+
Returns:
|
|
57
|
+
float: 对数谱距离值, 单位为分贝 (dB)。
|
|
58
|
+
"""
|
|
59
|
+
assert ref_wav.ndim == 1 and test_wav.ndim == 1, "输入信号必须是一维数组。"
|
|
60
|
+
|
|
61
|
+
if win_length is None:
|
|
62
|
+
win_length = n_fft
|
|
63
|
+
|
|
64
|
+
ref_stft = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
65
|
+
test_stft = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
66
|
+
|
|
67
|
+
ref_power_spec = np.abs(ref_stft) ** 2 # (F,T)
|
|
68
|
+
test_power_spec = np.abs(test_stft) ** 2 # (F,T)
|
|
69
|
+
|
|
70
|
+
ref_log_power_spec = 10 * np.log10(ref_power_spec + EPS)
|
|
71
|
+
test_log_power_spec = 10 * np.log10(test_power_spec + EPS)
|
|
72
|
+
|
|
73
|
+
squared_error = (ref_log_power_spec - test_log_power_spec) ** 2
|
|
74
|
+
lsd_val = np.sqrt(np.mean(squared_error))
|
|
75
|
+
|
|
76
|
+
return lsd_val
|
|
77
|
+
|
|
78
|
+
def mcd(ref_wav, test_wav, sr=16000):
|
|
79
|
+
"""
|
|
80
|
+
梅尔倒谱距离 Mel-Cepstral Distance
|
|
81
|
+
ref_spec: 参考频谱
|
|
82
|
+
test_spec: 测试频谱
|
|
83
|
+
"""
|
|
84
|
+
ref_wav, ref_sr = sf.read(ref_wav)
|
|
85
|
+
test_wav, test_sr = sf.read(test_wav)
|
|
86
|
+
assert ref_sr == test_sr == sr, "采样率必须为16000Hz"
|
|
87
|
+
assert len(ref_wav) == len(test_wav), "音频长度必须相同"
|
|
88
|
+
|
|
89
|
+
ref_mfcc = librosa.feature.mfcc(y=ref_wav, sr=sr)
|
|
90
|
+
test_mfcc = librosa.feature.mfcc(y=test_wav, sr=sr)
|
|
91
|
+
|
|
92
|
+
# 计算 MCD (跳过 0 阶)
|
|
93
|
+
diff = ref_mfcc[1:] - test_mfcc[1:]
|
|
94
|
+
mcd = (10.0 / np.log(10)) * np.sqrt(2 * np.mean(np.sum(diff ** 2, axis=0)))
|
|
95
|
+
return mcd
|
|
96
|
+
|
|
97
|
+
if __name__ == "__main__":
|
|
98
|
+
# 测试sd函数
|
|
99
|
+
print("=== 测试频谱距离 (SD) 函数 ===")
|
|
100
|
+
|
|
101
|
+
# 创建测试音频数据
|
|
102
|
+
sr = 16000
|
|
103
|
+
duration = 1.0 # 1秒
|
|
104
|
+
t = np.linspace(0, duration, int(sr * duration), False)
|
|
105
|
+
|
|
106
|
+
# 参考信号:正弦波
|
|
107
|
+
ref_wav = np.sin(2 * np.pi * 440 * t) # 440Hz
|
|
108
|
+
|
|
109
|
+
# 测试信号1:相同的正弦波(应该距离为0)
|
|
110
|
+
test_wav1 = np.sin(2 * np.pi * 440 * t)
|
|
111
|
+
|
|
112
|
+
# 测试信号2:不同频率的正弦波
|
|
113
|
+
test_wav2 = np.sin(2 * np.pi * 880 * t) # 880Hz
|
|
114
|
+
|
|
115
|
+
# 测试信号3:添加噪声
|
|
116
|
+
test_wav3 = ref_wav + 0.1 * np.random.randn(len(ref_wav))
|
|
117
|
+
|
|
118
|
+
try:
|
|
119
|
+
# 测试相同信号
|
|
120
|
+
sd1 = sd(ref_wav, test_wav1)
|
|
121
|
+
print(f"相同信号之间的SD: {sd1:.6f} (应该接近0)")
|
|
122
|
+
|
|
123
|
+
# 测试不同频率信号
|
|
124
|
+
sd2 = sd(ref_wav, test_wav2)
|
|
125
|
+
print(f"不同频率信号之间的SD: {sd2:.6f}")
|
|
126
|
+
|
|
127
|
+
# 测试带噪声信号
|
|
128
|
+
sd3 = sd(ref_wav, test_wav3)
|
|
129
|
+
print(f"带噪声信号之间的SD: {sd3:.6f}")
|
|
130
|
+
|
|
131
|
+
# 测试不同参数
|
|
132
|
+
sd4 = sd(ref_wav, test_wav3, n_fft=1024, hop_length=256)
|
|
133
|
+
print(f"使用不同参数的SD: {sd4:.6f}")
|
|
134
|
+
|
|
135
|
+
except Exception as e:
|
|
136
|
+
print(f"测试过程中发生错误: {e}")
|
|
137
|
+
|
|
138
|
+
print("\n=== 测试完成 ===")
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
'''
|
|
2
|
+
Author: 凌逆战 | Never
|
|
3
|
+
Date: 2025-08-16 13:51:57
|
|
4
|
+
Description: 频域客观度量指标
|
|
5
|
+
'''
|
|
6
|
+
|
|
7
|
+
import librosa
|
|
8
|
+
import numpy as np
|
|
9
|
+
import soundfile as sf
|
|
10
|
+
from neverlib.utils import EPS
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
def sd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
14
|
+
"""
|
|
15
|
+
计算两个音频信号之间的频谱距离 (Spectral Distance)。
|
|
16
|
+
该指标衡量两个信号在频域上的差异程度。
|
|
17
|
+
Args:
|
|
18
|
+
ref_wav (np.ndarray): 参考音频信号 (一维数组)
|
|
19
|
+
test_wav (np.ndarray): 测试音频信号 (一维数组)
|
|
20
|
+
n_fft (int): FFT点数,决定频率分辨率,默认为2048
|
|
21
|
+
hop_length (int): 帧移,决定时间分辨率,默认为512
|
|
22
|
+
win_length (int, optional): 窗长,如果为None则默认为n_fft
|
|
23
|
+
Returns:
|
|
24
|
+
float: 频谱距离值,值越小表示两个信号越相似
|
|
25
|
+
"""
|
|
26
|
+
assert len(ref_wav) == len(test_wav), "输入信号长度必须相同"
|
|
27
|
+
|
|
28
|
+
# 计算短时傅里叶变换
|
|
29
|
+
ref_spec = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
30
|
+
test_spec = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
|
|
31
|
+
|
|
32
|
+
# 计算频谱距离:均方根误差
|
|
33
|
+
spec_diff = ref_spec - test_spec
|
|
34
|
+
squared_diff = np.abs(spec_diff) ** 2
|
|
35
|
+
mean_squared_diff = np.mean(squared_diff)
|
|
36
|
+
sd_value = np.sqrt(mean_squared_diff)
|
|
37
|
+
|
|
38
|
+
return sd_value
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
|
|
44
|
+
"""
|
|
45
|
+
计算两个一维音频信号之间的对数谱距离 (Log-Spectral Distance, LSD)。
|
|
46
|
+
该实现遵循标准的LSD定义: 整体均方根误差。
|
|
47
|
+
|
|
48
|
+
Args:
|
|
49
|
+
ref_wav (np.ndarray): 原始的、干净的参考信号 (一维数组)。
|
|
50
|
+
test_wav (np.ndarray): 模型估计或处理后的信号 (一维数组)。
|
|
51
|
+
n_fft (int): FFT点数, 决定了频率分辨率。
|
|
52
|
+
hop_length (int): 帧移, 决定了时间分辨率。
|
|
53
|
+
win_length (int, optional): 窗长。如果为None, 则默认为n_fft。
|
|
54
|
+
epsilon (float): 一个非常小的数值, 用于防止对零取对数, 保证数值稳定性。
|
|
55
|
+
|
|
56
|
+
Returns:
|
|
57
|
+
float: 对数谱距离值, 单位为分贝 (dB)。
|
|
58
|
+
"""
|
|
59
|
+
assert ref_wav.ndim == 1 and test_wav.ndim == 1, "输入信号必须是一维数组。"
|
|
60
|
+
|
|
61
|
+
if win_length is None:
|
|
62
|
+
win_length = n_fft
|
|
63
|
+
|
|
64
|
+
ref_stft = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
65
|
+
test_stft = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
|
|
66
|
+
|
|
67
|
+
ref_power_spec = np.abs(ref_stft) ** 2 # (F,T)
|
|
68
|
+
test_power_spec = np.abs(test_stft) ** 2 # (F,T)
|
|
69
|
+
|
|
70
|
+
ref_log_power_spec = 10 * np.log10(ref_power_spec + EPS)
|
|
71
|
+
test_log_power_spec = 10 * np.log10(test_power_spec + EPS)
|
|
72
|
+
|
|
73
|
+
squared_error = (ref_log_power_spec - test_log_power_spec) ** 2
|
|
74
|
+
lsd_val = np.sqrt(np.mean(squared_error))
|
|
75
|
+
|
|
76
|
+
return lsd_val
|
|
77
|
+
|
|
78
|
+
def mcd(ref_wav, test_wav, sr=16000, n_mfcc=13):
|
|
79
|
+
"""
|
|
80
|
+
计算两个音频信号之间的梅尔倒谱距离 (Mel-Cepstral Distance, MCD)。
|
|
81
|
+
该指标常用于语音合成质量评估,值越小表示两个信号越相似。
|
|
82
|
+
|
|
83
|
+
Args:
|
|
84
|
+
ref_wav (np.ndarray): 参考音频信号 (一维数组)
|
|
85
|
+
test_wav (np.ndarray): 测试音频信号 (一维数组)
|
|
86
|
+
sr (int): 采样率,默认为16000Hz
|
|
87
|
+
n_mfcc (int): MFCC系数个数,默认为13
|
|
88
|
+
|
|
89
|
+
Returns:
|
|
90
|
+
float: 梅尔倒谱距离值,值越小表示两个信号越相似
|
|
91
|
+
|
|
92
|
+
Raises:
|
|
93
|
+
ValueError: 当输入参数无效时
|
|
94
|
+
AssertionError: 当输入信号不满足要求时
|
|
95
|
+
"""
|
|
96
|
+
# 输入验证
|
|
97
|
+
if not isinstance(ref_wav, np.ndarray) or not isinstance(test_wav, np.ndarray):
|
|
98
|
+
raise ValueError("输入信号必须是numpy数组")
|
|
99
|
+
|
|
100
|
+
if ref_wav.ndim != 1 or test_wav.ndim != 1:
|
|
101
|
+
raise AssertionError("输入信号必须是一维数组")
|
|
102
|
+
|
|
103
|
+
if len(ref_wav) == 0 or len(test_wav) == 0:
|
|
104
|
+
raise ValueError("输入信号不能为空")
|
|
105
|
+
|
|
106
|
+
if len(ref_wav) != len(test_wav):
|
|
107
|
+
raise ValueError("参考信号和测试信号长度必须相同")
|
|
108
|
+
|
|
109
|
+
if sr <= 0:
|
|
110
|
+
raise ValueError("采样率必须为正数")
|
|
111
|
+
|
|
112
|
+
if n_mfcc <= 0:
|
|
113
|
+
raise ValueError("MFCC系数个数必须为正数")
|
|
114
|
+
|
|
115
|
+
try:
|
|
116
|
+
# 计算MFCC特征
|
|
117
|
+
ref_mfcc = librosa.feature.mfcc(y=ref_wav, sr=sr, n_mfcc=n_mfcc)
|
|
118
|
+
test_mfcc = librosa.feature.mfcc(y=test_wav, sr=sr, n_mfcc=n_mfcc)
|
|
119
|
+
|
|
120
|
+
# 计算MCD (跳过0阶系数,因为0阶主要表示能量)
|
|
121
|
+
diff = ref_mfcc[1:] - test_mfcc[1:]
|
|
122
|
+
mcd_value = (10.0 / np.log(10)) * np.sqrt(2 * np.mean(np.sum(diff ** 2, axis=0)))
|
|
123
|
+
|
|
124
|
+
return mcd_value
|
|
125
|
+
|
|
126
|
+
except Exception as e:
|
|
127
|
+
raise RuntimeError(f"计算梅尔倒谱距离时发生错误: {str(e)}")
|
|
128
|
+
|
|
129
|
+
if __name__ == "__main__":
|
|
130
|
+
# 测试sd函数
|
|
131
|
+
print("=== 测试频谱距离 (SD) 函数 ===")
|
|
132
|
+
|
|
133
|
+
# 创建测试音频数据
|
|
134
|
+
sr = 16000
|
|
135
|
+
duration = 1.0 # 1秒
|
|
136
|
+
t = np.linspace(0, duration, int(sr * duration), False)
|
|
137
|
+
|
|
138
|
+
# 参考信号:正弦波
|
|
139
|
+
ref_wav = np.sin(2 * np.pi * 440 * t) # 440Hz
|
|
140
|
+
|
|
141
|
+
# 测试信号1:相同的正弦波(应该距离为0)
|
|
142
|
+
test_wav1 = np.sin(2 * np.pi * 440 * t)
|
|
143
|
+
|
|
144
|
+
# 测试信号2:不同频率的正弦波
|
|
145
|
+
test_wav2 = np.sin(2 * np.pi * 880 * t) # 880Hz
|
|
146
|
+
|
|
147
|
+
# 测试信号3:添加噪声
|
|
148
|
+
test_wav3 = ref_wav + 0.1 * np.random.randn(len(ref_wav))
|
|
149
|
+
|
|
150
|
+
try:
|
|
151
|
+
# 测试相同信号
|
|
152
|
+
sd1 = sd(ref_wav, test_wav1)
|
|
153
|
+
print(f"相同信号之间的SD: {sd1:.6f} (应该接近0)")
|
|
154
|
+
|
|
155
|
+
# 测试不同频率信号
|
|
156
|
+
sd2 = sd(ref_wav, test_wav2)
|
|
157
|
+
print(f"不同频率信号之间的SD: {sd2:.6f}")
|
|
158
|
+
|
|
159
|
+
# 测试带噪声信号
|
|
160
|
+
sd3 = sd(ref_wav, test_wav3)
|
|
161
|
+
print(f"带噪声信号之间的SD: {sd3:.6f}")
|
|
162
|
+
|
|
163
|
+
# 测试不同参数
|
|
164
|
+
sd4 = sd(ref_wav, test_wav3, n_fft=1024, hop_length=256)
|
|
165
|
+
print(f"使用不同参数的SD: {sd4:.6f}")
|
|
166
|
+
|
|
167
|
+
except Exception as e:
|
|
168
|
+
print(f"测试过程中发生错误: {e}")
|
|
169
|
+
|
|
170
|
+
print("\n=== 测试完成 ===")
|