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.
Files changed (207) hide show
  1. neverlib/.history/Docs/audio_aug/test_snr_20250806011311.py +0 -0
  2. neverlib/.history/Docs/audio_aug/test_snr_20250806011331.py +75 -0
  3. neverlib/.history/Docs/audio_aug/test_snr_20250806011342.py +57 -0
  4. neverlib/.history/Docs/audio_aug/test_snr_20250806011352.py +57 -0
  5. neverlib/.history/Docs/audio_aug/test_snr_20250806011403.py +57 -0
  6. neverlib/.history/Docs/audio_aug/test_snr_20250806011413.py +57 -0
  7. neverlib/.history/Docs/audio_aug/test_snr_20250806011435.py +55 -0
  8. neverlib/.history/Docs/vad/1_20250810032405.py +0 -0
  9. neverlib/.history/Docs/vad/1_20250810032417.py +39 -0
  10. neverlib/.history/audio_aug/audio_aug_20250806010451.py +125 -0
  11. neverlib/.history/audio_aug/audio_aug_20250806010750.py +138 -0
  12. neverlib/.history/audio_aug/audio_aug_20250806010759.py +140 -0
  13. neverlib/.history/audio_aug/audio_aug_20250806010803.py +140 -0
  14. neverlib/.history/audio_aug/audio_aug_20250806010809.py +140 -0
  15. neverlib/.history/audio_aug/audio_aug_20250806011108.py +140 -0
  16. neverlib/.history/dataAnalyze/__init___20250806204125.py +14 -0
  17. neverlib/.history/dataAnalyze/__init___20250806204139.py +14 -0
  18. neverlib/.history/dataAnalyze/__init___20250806204159.py +14 -0
  19. neverlib/.history/filter/__init___20250820103351.py +70 -0
  20. neverlib/.history/filter/__init___20250821102348.py +70 -0
  21. neverlib/.history/filter/__init___20250821102405.py +14 -0
  22. neverlib/.history/filter/auto_eq/__init___20250819213121.py +36 -0
  23. neverlib/.history/filter/auto_eq/__init___20250821102241.py +36 -0
  24. neverlib/.history/filter/auto_eq/__init___20250821102259.py +36 -0
  25. neverlib/.history/filter/auto_eq/__init___20250821102307.py +36 -0
  26. neverlib/.history/filter/auto_eq/__init___20250821102310.py +36 -0
  27. neverlib/.history/filter/auto_eq/__init___20250821102318.py +36 -0
  28. neverlib/.history/filter/auto_eq/__init___20250821102507.py +36 -0
  29. neverlib/{filter/AudoEQ/auto_eq_de.py → .history/filter/auto_eq/de_eq_20250820103848.py} +1 -1
  30. neverlib/.history/filter/auto_eq/de_eq_20250821102422.py +360 -0
  31. neverlib/.history/filter/auto_eq/freq_eq_20250820140732.py +75 -0
  32. neverlib/.history/filter/auto_eq/freq_eq_20250820140745.py +75 -0
  33. neverlib/.history/filter/auto_eq/freq_eq_20250820140816.py +75 -0
  34. neverlib/.history/filter/auto_eq/freq_eq_20250820140938.py +77 -0
  35. neverlib/.history/filter/auto_eq/freq_eq_20250820141003.py +77 -0
  36. neverlib/.history/filter/auto_eq/freq_eq_20250820141006.py +77 -0
  37. neverlib/.history/filter/auto_eq/freq_eq_20250820141019.py +77 -0
  38. neverlib/.history/filter/auto_eq/freq_eq_20250820141049.py +77 -0
  39. neverlib/.history/filter/auto_eq/freq_eq_20250820141211.py +77 -0
  40. neverlib/.history/filter/auto_eq/freq_eq_20250820141227.py +77 -0
  41. neverlib/.history/filter/auto_eq/freq_eq_20250820141311.py +78 -0
  42. neverlib/.history/filter/auto_eq/freq_eq_20250820141340.py +78 -0
  43. neverlib/.history/filter/auto_eq/freq_eq_20250820141712.py +78 -0
  44. neverlib/.history/filter/auto_eq/freq_eq_20250820141733.py +78 -0
  45. neverlib/.history/filter/auto_eq/freq_eq_20250820141755.py +78 -0
  46. neverlib/.history/filter/auto_eq/freq_eq_20250821102434.py +76 -0
  47. neverlib/.history/filter/auto_eq/freq_eq_20250821102500.py +76 -0
  48. neverlib/.history/filter/auto_eq/freq_eq_20250821102502.py +76 -0
  49. neverlib/{filter/AudoEQ/auto_eq_ga_basic.py → .history/filter/auto_eq/ga_eq_basic_20250820102957.py} +1 -1
  50. neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113054.py +380 -0
  51. neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113150.py +380 -0
  52. neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113520.py +385 -0
  53. neverlib/.history/filter/auto_eq/ga_eq_basic_20250820113525.py +385 -0
  54. neverlib/.history/filter/auto_eq/ga_eq_basic_20250821102212.py +385 -0
  55. neverlib/.history/metrics/dnsmos_20250806001612.py +160 -0
  56. neverlib/.history/metrics/dnsmos_20250815180659.py +160 -0
  57. neverlib/.history/metrics/dnsmos_20250815180701.py +158 -0
  58. neverlib/.history/metrics/dnsmos_20250815181321.py +154 -0
  59. neverlib/.history/metrics/dnsmos_20250815181327.py +154 -0
  60. neverlib/.history/metrics/dnsmos_20250815181331.py +154 -0
  61. neverlib/.history/metrics/dnsmos_20250815181620.py +154 -0
  62. neverlib/.history/metrics/dnsmos_20250815181631.py +154 -0
  63. neverlib/.history/metrics/dnsmos_20250815181742.py +154 -0
  64. neverlib/.history/metrics/dnsmos_20250815181824.py +153 -0
  65. neverlib/.history/metrics/dnsmos_20250815181834.py +153 -0
  66. neverlib/.history/metrics/dnsmos_20250815181922.py +153 -0
  67. neverlib/.history/metrics/dnsmos_20250815182011.py +147 -0
  68. neverlib/.history/metrics/dnsmos_20250815182036.py +144 -0
  69. neverlib/.history/metrics/dnsmos_20250815182936.py +143 -0
  70. neverlib/.history/metrics/dnsmos_20250815182942.py +143 -0
  71. neverlib/.history/metrics/dnsmos_20250815183032.py +137 -0
  72. neverlib/.history/metrics/dnsmos_20250815183101.py +144 -0
  73. neverlib/.history/metrics/dnsmos_20250815183121.py +144 -0
  74. neverlib/.history/metrics/dnsmos_20250815183123.py +143 -0
  75. neverlib/.history/metrics/dnsmos_20250815183214.py +143 -0
  76. neverlib/.history/metrics/dnsmos_20250815183240.py +143 -0
  77. neverlib/.history/metrics/dnsmos_20250815183248.py +144 -0
  78. neverlib/.history/metrics/dnsmos_20250815183407.py +142 -0
  79. neverlib/.history/metrics/dnsmos_20250815183409.py +142 -0
  80. neverlib/.history/metrics/dnsmos_20250815183431.py +142 -0
  81. neverlib/.history/metrics/dnsmos_20250815183507.py +140 -0
  82. neverlib/.history/metrics/dnsmos_20250815183513.py +139 -0
  83. neverlib/.history/metrics/dnsmos_20250815183618.py +139 -0
  84. neverlib/.history/metrics/dnsmos_20250815183709.py +140 -0
  85. neverlib/.history/metrics/dnsmos_20250815183756.py +137 -0
  86. neverlib/.history/metrics/dnsmos_20250815183815.py +128 -0
  87. neverlib/.history/metrics/dnsmos_20250815183827.py +129 -0
  88. neverlib/.history/metrics/dnsmos_20250815183913.py +117 -0
  89. neverlib/.history/metrics/dnsmos_20250815183914.py +117 -0
  90. neverlib/.history/metrics/dnsmos_20250815184003.py +118 -0
  91. neverlib/.history/metrics/dnsmos_20250815184040.py +118 -0
  92. neverlib/.history/metrics/dnsmos_20250815184049.py +118 -0
  93. neverlib/.history/metrics/dnsmos_20250815184104.py +117 -0
  94. neverlib/.history/metrics/dnsmos_20250815184200.py +117 -0
  95. neverlib/.history/metrics/lpc_lsp_metric_20250816015944.py +128 -0
  96. neverlib/.history/metrics/lpc_lsp_metric_20250816020142.py +128 -0
  97. neverlib/.history/metrics/lpc_lsp_metric_20250816020156.py +128 -0
  98. neverlib/.history/metrics/lpc_lsp_metric_20250816020554.py +130 -0
  99. neverlib/.history/metrics/lpc_lsp_metric_20250816020600.py +125 -0
  100. neverlib/.history/metrics/lpc_lsp_metric_20250816020631.py +120 -0
  101. neverlib/.history/metrics/lpc_lsp_metric_20250816020746.py +118 -0
  102. neverlib/.history/metrics/lpc_me_20250816013111.py +0 -0
  103. neverlib/.history/metrics/lpc_me_20250816013129.py +121 -0
  104. neverlib/.history/metrics/lpc_me_20250816015430.py +103 -0
  105. neverlib/.history/metrics/lpc_me_20250816015535.py +96 -0
  106. neverlib/.history/metrics/lpc_me_20250816015542.py +96 -0
  107. neverlib/.history/metrics/lpc_me_20250816015636.py +97 -0
  108. neverlib/.history/metrics/lpc_me_20250816015658.py +104 -0
  109. neverlib/.history/metrics/lpc_me_20250816015703.py +100 -0
  110. neverlib/.history/metrics/lpc_me_20250816015945.py +128 -0
  111. neverlib/.history/metrics/snr_20250806010538.py +177 -0
  112. neverlib/.history/metrics/snr_20250806211634.py +184 -0
  113. neverlib/.history/metrics/spec_20250805234209.py +45 -0
  114. neverlib/.history/metrics/spec_20250816135530.py +11 -0
  115. neverlib/.history/metrics/spec_20250816135654.py +16 -0
  116. neverlib/.history/metrics/spec_20250816135736.py +68 -0
  117. neverlib/.history/metrics/spec_20250816135904.py +75 -0
  118. neverlib/.history/metrics/spec_20250816135921.py +82 -0
  119. neverlib/.history/metrics/spec_20250816140111.py +82 -0
  120. neverlib/.history/metrics/spec_20250816140543.py +136 -0
  121. neverlib/.history/metrics/spec_20250816140559.py +172 -0
  122. neverlib/.history/metrics/spec_20250816140602.py +172 -0
  123. neverlib/.history/metrics/spec_20250816140608.py +172 -0
  124. neverlib/.history/metrics/spec_20250816140654.py +148 -0
  125. neverlib/.history/metrics/spec_20250816140705.py +144 -0
  126. neverlib/.history/metrics/spec_20250816140755.py +138 -0
  127. neverlib/.history/metrics/spec_20250816140823.py +170 -0
  128. neverlib/.history/metrics/spec_20250816140832.py +170 -0
  129. neverlib/.history/metrics/spec_20250816140833.py +170 -0
  130. neverlib/.history/metrics/spec_20250816140922.py +147 -0
  131. neverlib/.history/metrics/spec_20250816141148.py +107 -0
  132. neverlib/.history/metrics/spec_20250816141219.py +123 -0
  133. neverlib/.history/metrics/spec_20250816141732.py +178 -0
  134. neverlib/.history/metrics/spec_20250816141740.py +178 -0
  135. neverlib/.history/metrics/spec_20250816142030.py +178 -0
  136. neverlib/.history/metrics/spec_20250816142107.py +135 -0
  137. neverlib/.history/metrics/spec_20250816142126.py +135 -0
  138. neverlib/.history/metrics/spec_20250816142410.py +135 -0
  139. neverlib/.history/metrics/spec_20250816142415.py +136 -0
  140. neverlib/.history/metrics/spec_metric_20250816135156.py +0 -0
  141. neverlib/.history/metrics/spec_metric_20250816135226.py +5 -0
  142. neverlib/.history/metrics/spec_metric_20250816135227.py +10 -0
  143. neverlib/.history/metrics/spec_metric_20250816135306.py +15 -0
  144. neverlib/.history/metrics/spec_metric_20250816135442.py +31 -0
  145. neverlib/.history/metrics/spec_metric_20250816135448.py +31 -0
  146. neverlib/.history/metrics/spec_metric_20250816135520.py +29 -0
  147. neverlib/.history/metrics/spec_metric_20250816135537.py +63 -0
  148. neverlib/.history/metrics/spec_metric_20250816135653.py +65 -0
  149. neverlib/.history/vad/PreProcess_20250805234211.py +63 -0
  150. neverlib/.history/vad/PreProcess_20250809232455.py +63 -0
  151. neverlib/.history/vad/PreProcess_20250816020725.py +66 -0
  152. neverlib/.history/vad/VAD_Silero_20250805234211.py +50 -0
  153. neverlib/.history/vad/VAD_Silero_20250809232456.py +50 -0
  154. neverlib/.history/vad/VAD_WebRTC_20250805234211.py +61 -0
  155. neverlib/.history/vad/VAD_WebRTC_20250809232456.py +61 -0
  156. neverlib/.history/vad/VAD_funasr_20250805234211.py +54 -0
  157. neverlib/.history/vad/VAD_funasr_20250809232456.py +54 -0
  158. neverlib/.history/vad/VAD_vadlib_20250805234211.py +70 -0
  159. neverlib/.history/vad/VAD_vadlib_20250809232455.py +70 -0
  160. neverlib/.history/vad/VAD_whisper_20250805234211.py +55 -0
  161. neverlib/.history/vad/VAD_whisper_20250809232456.py +55 -0
  162. neverlib/.specstory/.what-is-this.md +69 -0
  163. 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
  164. neverlib/Docs/audio_aug/test_snr.py +55 -0
  165. neverlib/audio_aug/HarmonicDistortion.py +79 -0
  166. neverlib/audio_aug/TFDrop.py +41 -0
  167. neverlib/audio_aug/TFMask.py +56 -0
  168. neverlib/audio_aug/audio_aug.py +16 -1
  169. neverlib/audio_aug/clip_aug.py +41 -0
  170. neverlib/audio_aug/coder_aug.py +209 -0
  171. neverlib/audio_aug/coder_aug2.py +118 -0
  172. neverlib/audio_aug/loss_packet_aug.py +103 -0
  173. neverlib/audio_aug/quant_aug.py +78 -0
  174. neverlib/data_analyze/__init__.py +14 -0
  175. neverlib/filter/auto_eq/__init__.py +36 -0
  176. neverlib/filter/auto_eq/de_eq.py +360 -0
  177. neverlib/filter/auto_eq/freq_eq.py +76 -0
  178. neverlib/filter/{AudoEQ/auto_eq_ga_advanced.py → auto_eq/ga_eq_advanced.py} +1 -1
  179. neverlib/filter/auto_eq/ga_eq_basic.py +385 -0
  180. neverlib/metrics/dnsmos.py +58 -101
  181. neverlib/metrics/lpc_lsp.py +118 -0
  182. neverlib/metrics/snr.py +11 -4
  183. neverlib/metrics/spec.py +136 -45
  184. neverlib/utils/utils.py +17 -14
  185. neverlib/vad/PreProcess.py +5 -2
  186. neverlib/vad/VAD_Silero.py +1 -1
  187. neverlib/vad/VAD_WebRTC.py +1 -1
  188. neverlib/vad/VAD_funasr.py +1 -1
  189. neverlib/vad/VAD_vadlib.py +1 -1
  190. neverlib/vad/VAD_whisper.py +1 -1
  191. {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/METADATA +1 -1
  192. neverlib-0.2.4.dist-info/RECORD +229 -0
  193. neverlib-0.2.3.dist-info/RECORD +0 -53
  194. /neverlib/{dataAnalyze/__init__.py → .history/dataAnalyze/__init___20250805234204.py} +0 -0
  195. /neverlib/{filter/AudoEQ/auto_eq_spectral_direct.py → .history/filter/auto_eq/freq_eq_20250805234206.py} +0 -0
  196. /neverlib/{dataAnalyze → data_analyze}/README.md +0 -0
  197. /neverlib/{dataAnalyze → data_analyze}/dataset_analyzer.py +0 -0
  198. /neverlib/{dataAnalyze → data_analyze}/quality_metrics.py +0 -0
  199. /neverlib/{dataAnalyze → data_analyze}/rms_distrubution.py +0 -0
  200. /neverlib/{dataAnalyze → data_analyze}/spectral_analysis.py +0 -0
  201. /neverlib/{dataAnalyze → data_analyze}/statistics.py +0 -0
  202. /neverlib/{dataAnalyze → data_analyze}/temporal_features.py +0 -0
  203. /neverlib/{dataAnalyze → data_analyze}/visualization.py +0 -0
  204. /neverlib/filter/{AudoEQ → auto_eq}/README.md +0 -0
  205. {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/WHEEL +0 -0
  206. {neverlib-0.2.3.dist-info → neverlib-0.2.4.dist-info}/licenses/LICENSE +0 -0
  207. {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=== 测试完成 ===")