neverlib 0.2.6__py3-none-any.whl → 0.2.8__py3-none-any.whl

neverlib/metrics/pesq_c/pesqpar.h

@@ -0,0 +1,297 @@
+/*****************************************************************************
+
+Perceptual Evaluation of Speech Quality (PESQ)
+ITU-T Recommendation P.862.
+Version 1.2 - 2 August 2002.
+
+              ****************************************
+              PESQ Intellectual Property Rights Notice
+              ****************************************
+
+DEFINITIONS:
+------------
+For the purposes of this Intellectual Property Rights Notice
+the terms �Perceptual Evaluation of Speech Quality Algorithm�
+and �PESQ Algorithm� refer to the objective speech quality
+measurement algorithm defined in ITU-T Recommendation P.862;
+the term �PESQ Software� refers to the C-code component of P.862. 
+
+NOTICE:
+-------
+All copyright, trade marks, trade names, patents, know-how and
+all or any other intellectual rights subsisting in or used in
+connection with including all algorithms, documents and manuals
+relating to the PESQ Algorithm and or PESQ Software are and remain
+the sole property in law, ownership, regulations, treaties and
+patent rights of the Owners identified below. The user may not
+dispute or question the ownership of the PESQ Algorithm and
+or PESQ Software.
+
+OWNERS ARE:
+-----------
+
+1.	British Telecommunications plc (BT), all rights assigned
+      to Psytechnics Limited
+2.	Royal KPN NV, all rights assigned to OPTICOM GmbH
+
+RESTRICTIONS:
+-------------
+
+The user cannot:
+
+1.	alter, duplicate, modify, adapt, or translate in whole or in
+      part any aspect of the PESQ Algorithm and or PESQ Software
+2.	sell, hire, loan, distribute, dispose or put to any commercial
+      use other than those permitted below in whole or in part any
+      aspect of the PESQ Algorithm and or PESQ Software
+
+PERMITTED USE:
+--------------
+
+The user may:
+
+1.	Use the PESQ Software to:
+      i)   understand the PESQ Algorithm; or
+      ii)  evaluate the ability of the PESQ Algorithm to perform
+           its intended function of predicting the speech quality
+           of a system; or
+      iii) evaluate the computational complexity of the PESQ Algorithm,
+           with the limitation that none of said evaluations or its
+           results shall be used for external commercial use.
+
+2.	Use the PESQ Software to test if an implementation of the PESQ
+      Algorithm conforms to ITU-T Recommendation P.862.
+
+3.	With the prior written permission of both Psytechnics Limited
+      and OPTICOM GmbH, use the PESQ Software in accordance with the
+      above Restrictions to perform work that meets all of the following
+      criteria:
+      i)    the work must contribute directly to the maintenance of an
+            existing ITU recommendation or the development of a new ITU
+            recommendation under an approved ITU Study Item; and
+      ii)   the work and its results must be fully described in a
+            written contribution to the ITU that is presented at a formal
+            ITU meeting within one year of the start of the work; and
+      iii)  neither the work nor its results shall be put to any
+            commercial use other than making said contribution to the ITU.
+            Said permission will be provided on a case-by-case basis.
+
+
+ANY OTHER USE OR APPLICATION OF THE PESQ SOFTWARE AND/OR THE PESQ
+ALGORITHM WILL REQUIRE A PESQ LICENCE AGREEMENT, WHICH MAY BE OBTAINED
+FROM EITHER OPTICOM GMBH OR PSYTECHNICS LIMITED. 
+
+EACH COMPANY OFFERS OEM LICENSE AGREEMENTS, WHICH COMBINE OEM
+IMPLEMENTATIONS OF THE PESQ ALGORITHM TOGETHER WITH A PESQ PATENT LICENSE
+AGREEMENT. PESQ PATENT-ONLY LICENSE AGREEMENTS MAY BE OBTAINED FROM OPTICOM.
+
+
+***********************************************************************
+*  OPTICOM GmbH                    *  Psytechnics Limited             *
+*  Am Weichselgarten 7,            *  Fraser House, 23 Museum Street, *
+*  D- 91058 Erlangen, Germany      *  Ipswich IP1 1HN, England        *
+*  Phone: +49 (0) 9131 691 160     *  Phone: +44 (0) 1473 261 800     *
+*  Fax:   +49 (0) 9131 691 325     *  Fax:   +44 (0) 1473 261 880     *
+*  E-mail: info@opticom.de,        *  E-mail: info@psytechnics.com,   *
+*  www.opticom.de                  *  www.psytechnics.com             *
+***********************************************************************
+
+Further information is also available from www.pesq.org
+
+*****************************************************************************/
+
+long Fs = 0L;
+
+long Fs_16k = 16000L;
+
+long Fs_8k = 8000L;
+
+long Downsample;
+
+long Downsample_16k = 64;
+
+long Downsample_8k = 32;
+
+long Align_Nfft;
+
+long Align_Nfft_16k = 1024;
+
+long Align_Nfft_8k = 512;
+
+long InIIR_Nsos;
+
+long InIIR_Nsos_8k = 8L;
+float InIIR_Hsos_8k[LINIIR] =
+ { 0.885535424f,        -0.885535424f,  0.000000000f,   -0.771070709f,  0.000000000f,
+    0.895092588f,       1.292907193f,   0.449260174f,   1.268869037f,   0.442025372f,
+    4.049527940f,       -7.865190042f,  3.815662102f,   -1.746859852f,  0.786305963f,
+    0.500002353f,       -0.500002353f,  0.000000000f,   0.000000000f,   0.000000000f,
+    0.565002834f,       -0.241585934f,  -0.306009671f,  0.259688659f,   0.249979657f,
+    2.115237288f,       0.919935084f,   1.141240051f,   -1.587313419f,  0.665935315f,
+    0.912224584f,       -0.224397719f,  -0.641121413f,  -0.246029464f,  -0.556720590f,
+    0.444617727f,       -0.307589321f,  0.141638062f,   -0.996391149f,  0.502251622f };
+
+long InIIR_Nsos_16k = 12L;
+float InIIR_Hsos_16k[LINIIR] =
+ { 0.325631521f,        -0.086782860f,  -0.238848661f,  -1.079416490f,  0.434583902f,
+   0.403961804f,        -0.556985881f,  0.153024077f,   -0.415115835f,  0.696590244f,
+   4.736162769f,        3.287251046f,   1.753289019f,   -1.859599046f,  0.876284034f,
+   0.365373469f,        0.000000000f,   0.000000000f,   -0.634626531f,  0.000000000f,
+   0.884811506f,        0.000000000f,   0.000000000f,   -0.256725271f,  0.141536777f,
+   0.723593055f,        -1.447186099f,  0.723593044f,   -1.129587469f,  0.657232737f,
+   1.644910855f,        -1.817280902f,  1.249658063f,   -1.778403899f,  0.801724355f,
+   0.633692689f,        -0.284644314f,  -0.319789663f,  0.000000000f,   0.000000000f,
+   1.032763031f,        0.268428979f,   0.602913323f,   0.000000000f,   0.000000000f,
+   1.001616361f,        -0.823749013f,  0.439731942f,   -0.885778255f,  0.000000000f,
+   0.752472096f,        -0.375388990f,  0.188977609f,   -0.077258216f,  0.247230734f,
+   1.023700575f,        0.001661628f,   0.521284240f,   -0.183867259f,  0.354324187f };
+
+
+int nr_of_hz_bands_per_bark_band_8k [42] = { 1,    1,    1,    1,    1,    
+                                                1,    1,    1,    2,    1,    
+                                                1,    1,    1,    1,    2,    
+                                                1,    1,    2,    2,    2,    
+                                                2,    2,    2,    2,    2,    
+                                                3,    3,    3,    3,    4,    
+                                                3,    4,    5,    4,    5,    
+                                                6,    6,    7,    8,    9,    
+                                                9,    11,    };
+
+double centre_of_band_bark_8k [42] = { 0.078672,     0.316341,     0.636559,     0.961246,     1.290450,     
+                                        1.624217,     1.962597,     2.305636,     2.653383,     3.005889,     
+                                        3.363201,     3.725371,     4.092449,     4.464486,     4.841533,     
+                                        5.223642,     5.610866,     6.003256,     6.400869,     6.803755,     
+                                        7.211971,     7.625571,     8.044611,     8.469146,     8.899232,     
+                                        9.334927,     9.776288,     10.223374,     10.676242,     11.134952,     
+                                        11.599563,     12.070135,     12.546731,     13.029408,     13.518232,     
+                                        14.013264,     14.514566,     15.022202,     15.536238,     16.056736,     
+                                        16.583761,     17.117382};
+
+double centre_of_band_hz_8k [42] = { 7.867213,     31.634144,     63.655895,     96.124611,     129.044968,     
+                                    162.421738,     196.259659,     230.563568,     265.338348,     300.588867,     
+                                    336.320129,     372.537140,     409.244934,     446.448578,     484.568604,     
+                                    526.600586,     570.303833,     619.423340,     672.121643,     728.525696,     
+                                    785.675964,     846.835693,     909.691650,     977.063293,     1049.861694,     
+                                    1129.635986,     1217.257568,     1312.109497,     1412.501465,     1517.999390,     
+                                    1628.894165,     1746.194336,     1871.568848,     2008.776123,     2158.979248,     
+                                    2326.743164,     2513.787109,     2722.488770,     2952.586670,     3205.835449,     
+                                    3492.679932,     3820.219238};
+
+double width_of_band_bark_8k [42] = { 0.157344,     0.317994,     0.322441,     0.326934,     0.331474,     
+                                        0.336061,     0.340697,     0.345381,     0.350114,     0.354897,     
+                                        0.359729,     0.364611,     0.369544,     0.374529,     0.379565,     
+                                        0.384653,     0.389794,     0.394989,     0.400236,     0.405538,     
+                                        0.410894,     0.416306,     0.421773,     0.427297,     0.432877,     
+                                        0.438514,     0.444209,     0.449962,     0.455774,     0.461645,     
+                                        0.467577,     0.473569,     0.479621,     0.485736,     0.491912,     
+                                        0.498151,     0.504454,     0.510819,     0.517250,     0.523745,     
+                                        0.530308,     0.536934};
+
+double width_of_band_hz_8k [42] = { 15.734426,     31.799433,     32.244064,     32.693359,     33.147385,     
+                                    33.606140,     34.069702,     34.538116,     35.011429,     35.489655,     
+                                    35.972870,     36.461121,     36.954407,     37.452911,     40.269653,     
+                                    42.311859,     45.992554,     51.348511,     55.040527,     56.775208,     
+                                    58.699402,     62.445862,     64.820923,     69.195374,     76.745667,     
+                                    84.016235,     90.825684,     97.931152,     103.348877,     107.801880,     
+                                    113.552246,     121.490601,     130.420410,     143.431763,     158.486816,     
+                                    176.872803,     198.314697,     219.549561,     240.600098,     268.702393,     
+                                    306.060059,     349.937012};
+
+double pow_dens_correction_factor_8k [42] = { 100.000000,     99.999992,     100.000000,     100.000008,     100.000008, 
+                                                100.000015,     99.999992,     99.999969,     50.000027,     100.000000,     
+                                                99.999969,     100.000015,     99.999947,     100.000061,     53.047077,     
+                                                110.000046,     117.991989,     65.000000,     68.760147,     69.999931,     
+                                                71.428818,     75.000038,     76.843384,     80.968781,     88.646126,     
+                                                63.864388,     68.155350,     72.547775,     75.584831,     58.379192,     
+                                                80.950836,     64.135651,     54.384785,     73.821884,     64.437073,     
+                                                59.176456,     65.521278,     61.399822,     58.144047,     57.004543,     
+                                                64.126297,     59.248363};
+
+double abs_thresh_power_8k [42] = {51286152.000000,     2454709.500000,     70794.593750,     4897.788574,     1174.897705,     
+                                    389.045166,     104.712860,     45.708820,     17.782795,     9.772372,     
+                                    4.897789,     3.090296,     1.905461,     1.258925,     0.977237,     
+                                    0.724436,     0.562341,     0.457088,     0.389045,     0.331131,     
+                                    0.295121,     0.269153,     0.257040,     0.251189,     0.251189,     
+                                    0.251189,     0.251189,     0.263027,     0.288403,     0.309030,     
+                                    0.338844,     0.371535,     0.398107,     0.436516,     0.467735,     
+                                    0.489779,     0.501187,     0.501187,     0.512861,     0.524807,     
+                                    0.524807,     0.524807};
+
+int nr_of_hz_bands_per_bark_band_16k [49] = { 1,    1,    1,    1,    1,    
+                                                1,    1,    1,    2,    1,    
+                                                1,    1,    1,    1,    2,    
+                                                1,    1,    2,    2,    2,    
+                                                2,    2,    2,    2,    2,    
+                                                3,    3,    3,    3,    4,    
+                                                3,    4,    5,    4,    5,    
+                                                6,    6,    7,    8,    9,    
+                                                9,    12,    12,    15,    16,    
+                                                18,    21,    25,    20};
+
+double centre_of_band_bark_16k [49] = { 0.078672,     0.316341,     0.636559,     0.961246,     1.290450,     
+                                        1.624217,     1.962597,     2.305636,     2.653383,     3.005889,     
+                                        3.363201,     3.725371,     4.092449,     4.464486,     4.841533,     
+                                        5.223642,     5.610866,     6.003256,     6.400869,     6.803755,     
+                                        7.211971,     7.625571,     8.044611,     8.469146,     8.899232,     
+                                        9.334927,     9.776288,     10.223374,     10.676242,     11.134952,     
+                                        11.599563,     12.070135,     12.546731,     13.029408,     13.518232,     
+                                        14.013264,     14.514566,     15.022202,     15.536238,     16.056736,     
+                                        16.583761,     17.117382,     17.657663,     18.204674,     18.758478,     
+                                        19.319147,     19.886751,     20.461355,     21.043034};
+
+double centre_of_band_hz_16k [49] = { 7.867213,     31.634144,     63.655895,     96.124611,     129.044968,     
+                                        162.421738,     196.259659,     230.563568,     265.338348,     300.588867,     
+                                        336.320129,     372.537140,     409.244934,     446.448578,     484.568604,     
+                                        526.600586,     570.303833,     619.423340,     672.121643,     728.525696,     
+                                        785.675964,     846.835693,     909.691650,     977.063293,     1049.861694,     
+                                        1129.635986,     1217.257568,     1312.109497,     1412.501465,     1517.999390,     
+                                        1628.894165,     1746.194336,     1871.568848,     2008.776123,     2158.979248,     
+                                        2326.743164,     2513.787109,     2722.488770,     2952.586670,     3205.835449,     
+                                        3492.679932,     3820.219238,     4193.938477,     4619.846191,     5100.437012,     
+                                        5636.199219,     6234.313477,     6946.734863,     7796.473633};
+
+double width_of_band_bark_16k [49] = { 0.157344,     0.317994,     0.322441,     0.326934,     0.331474,     
+                                        0.336061,     0.340697,     0.345381,     0.350114,     0.354897,     
+                                        0.359729,     0.364611,     0.369544,     0.374529,     0.379565,     
+                                        0.384653,     0.389794,     0.394989,     0.400236,     0.405538,     
+                                        0.410894,     0.416306,     0.421773,     0.427297,     0.432877,     
+                                        0.438514,     0.444209,     0.449962,     0.455774,     0.461645,     
+                                        0.467577,     0.473569,     0.479621,     0.485736,     0.491912,     
+                                        0.498151,     0.504454,     0.510819,     0.517250,     0.523745,     
+                                        0.530308,     0.536934,     0.543629,     0.550390,     0.557220,     
+                                        0.564119,     0.571085,     0.578125,     0.585232};
+
+double width_of_band_hz_16k [49] = { 15.734426,     31.799433,     32.244064,     32.693359,     33.147385,     
+                                    33.606140,     34.069702,     34.538116,     35.011429,     35.489655,     
+                                    35.972870,     36.461121,     36.954407,     37.452911,     40.269653,     
+                                    42.311859,     45.992554,     51.348511,     55.040527,     56.775208,     
+                                    58.699402,     62.445862,     64.820923,     69.195374,     76.745667,     
+                                    84.016235,     90.825684,     97.931152,     103.348877,     107.801880,     
+                                    113.552246,     121.490601,     130.420410,     143.431763,     158.486816,     
+                                    176.872803,     198.314697,     219.549561,     240.600098,     268.702393,     
+                                    306.060059,     349.937012,     398.686279,     454.713867,     506.841797,     
+                                    564.863770,     637.261230,     794.717285,     931.068359};
+
+double pow_dens_correction_factor_16k [49] = { 100.000000,     99.999992,     100.000000,     100.000008,     100.000008,     
+                                                100.000015,     99.999992,     99.999969,     50.000027,     100.000000,     
+                                                99.999969,     100.000015,     99.999947,     100.000061,     53.047077,     
+                                                110.000046,     117.991989,     65.000000,     68.760147,     69.999931,     
+                                                71.428818,     75.000038,     76.843384,     80.968781,     88.646126,     
+                                                63.864388,     68.155350,     72.547775,     75.584831,     58.379192,     
+                                                80.950836,     64.135651,     54.384785,     73.821884,     64.437073,     
+                                                59.176456,     65.521278,     61.399822,     58.144047,     57.004543,     
+                                                64.126297,     54.311001,     61.114979,     55.077751,     56.849335,     
+                                                55.628868,     53.137054,     54.985844,     79.546974};
+double abs_thresh_power_16k [49] = {51286152.000000,     2454709.500000,     70794.593750,     4897.788574,     1174.897705,     
+                                    389.045166,     104.712860,     45.708820,     17.782795,     9.772372,     
+                                    4.897789,     3.090296,     1.905461,     1.258925,     0.977237,     
+                                    0.724436,     0.562341,     0.457088,     0.389045,     0.331131,     
+                                    0.295121,     0.269153,     0.257040,     0.251189,     0.251189,     
+                                    0.251189,     0.251189,     0.263027,     0.288403,     0.309030,     
+                                    0.338844,     0.371535,     0.398107,     0.436516,     0.467735,     
+                                    0.489779,     0.501187,     0.501187,     0.512861,     0.524807,     
+                                    0.524807,     0.524807,     0.512861,     0.478630,     0.426580,     
+                                    0.371535,     0.363078,     0.416869,     0.537032};
+
+
+/* END OF FILE */

neverlib/metrics/snr.py

@@ -1,4 +1,3 @@
-import librosa
 import numpy as np
 from neverlib.vad.utils import vad2nad
 from neverlib.filter import HPFilter
@@ -72,6 +71,11 @@ def seg_snr(clean, noisy, frame_length: int, hop_length: int):
     Raises:
         ValueError: 当输入参数不合法时抛出
     """
+    try:
+        import librosa
+    except Exception as e:
+        raise ImportError("需要安装 librosa 才能使用 seg_snr: pip install librosa") from e
+
     assert clean.shape == noisy.shape, "clean和noisy的维度不一样"
 
     # 分帧

neverlib/metrics/spec.py

@@ -36,20 +36,24 @@ def sd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
         float: 频谱距离值，值越小表示两个信号越相似
     """
     assert len(ref_wav) == len(test_wav), "输入信号长度必须相同"
-    
+
     # 计算短时傅里叶变换
-    ref_spec = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
-    test_spec = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length)
-    
+    ref_spec = librosa.stft(ref_wav,
+                            n_fft=n_fft,
+                            hop_length=hop_length,
+                            win_length=win_length)
+    test_spec = librosa.stft(test_wav,
+                             n_fft=n_fft,
+                             hop_length=hop_length,
+                             win_length=win_length)
+
     # 计算频谱距离：均方根误差
     spec_diff = ref_spec - test_spec
-    squared_diff = np.abs(spec_diff) ** 2
+    squared_diff = np.abs(spec_diff)**2
     mean_squared_diff = np.mean(squared_diff)
     sd_value = np.sqrt(mean_squared_diff)
-    
-    return sd_value
-
 
+    return sd_value
 
 
 def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
@@ -69,20 +73,26 @@ def lsd(ref_wav, test_wav, n_fft=2048, hop_length=512, win_length=None):
         float: 对数谱距离值, 单位为分贝 (dB)。
     """
     assert ref_wav.ndim == 1 and test_wav.ndim == 1, "输入信号必须是一维数组。"
-    
+
     if win_length is None:
         win_length = n_fft
 
-    ref_stft = librosa.stft(ref_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
-    test_stft = librosa.stft(test_wav, n_fft=n_fft, hop_length=hop_length, win_length=win_length) # (F,T)
+    ref_stft = librosa.stft(ref_wav,
+                            n_fft=n_fft,
+                            hop_length=hop_length,
+                            win_length=win_length)  # (F,T)
+    test_stft = librosa.stft(test_wav,
+                             n_fft=n_fft,
+                             hop_length=hop_length,
+                             win_length=win_length)  # (F,T)
 
-    ref_power_spec = np.abs(ref_stft) ** 2 # (F,T)
-    test_power_spec = np.abs(test_stft) ** 2 # (F,T)
+    ref_power_spec = np.abs(ref_stft)**2  # (F,T)
+    test_power_spec = np.abs(test_stft)**2  # (F,T)
 
     ref_log_power_spec = 10 * np.log10(ref_power_spec + EPS)
     test_log_power_spec = 10 * np.log10(test_power_spec + EPS)
 
-    squared_error = (ref_log_power_spec - test_log_power_spec) ** 2
+    squared_error = (ref_log_power_spec - test_log_power_spec)**2
     lsd_val = np.sqrt(np.mean(squared_error))
 
     return lsd_val
@@ -108,18 +118,19 @@ def mcd(ref_wav, test_wav, sr=16000, n_mfcc=13):
     # 计算MFCC特征
     ref_mfcc = librosa.feature.mfcc(y=ref_wav, sr=sr, n_mfcc=n_mfcc)
     test_mfcc = librosa.feature.mfcc(y=test_wav, sr=sr, n_mfcc=n_mfcc)
-    
+
     # 计算MCD (跳过0阶系数，因为0阶主要表示能量)
     diff = ref_mfcc[1:] - test_mfcc[1:]
-    mcd_value = (10.0 / np.log(10)) * np.sqrt(2 * np.mean(np.sum(diff ** 2, axis=0)))
-    
+    mcd_value = (10.0 / np.log(10)) * np.sqrt(
+        2 * np.mean(np.sum(diff**2, axis=0)))
+
     return mcd_value
 
 
 if __name__ == "__main__":
-    ref_file = "../data/vad_example.wav"   # 参考语音文件路径
-    test_file = "../data/vad_example.wav" # 测试语音文件路径
-    
+    ref_file = "../data/vad_example.wav"  # 参考语音文件路径
+    test_file = "../data/vad_example.wav"  # 测试语音文件路径
+
     ref_wav, ref_sr = sf.read(ref_file)
     test_wav, test_sr = sf.read(test_file)
     assert ref_sr == test_sr == 16000, "采样率必须为16000Hz"
@@ -133,4 +144,3 @@ if __name__ == "__main__":
 
     sd_value = sd(ref_wav, test_wav)
     print(f"频谱距离: {sd_value:.2f}")
-    

neverlib/metrics/test_pesq.py

@@ -8,7 +8,6 @@ PESQ 包含 3 种类型的值：NB PESQ MOS、NB MOS LQO、WB MOS LQO。此包
 import pesq
 import pypesq
 import librosa
-import os
 import numpy as np
 
 fs = 16000
@@ -30,6 +29,3 @@ def mos2pesq(mos):
     """ 将MOS-LQO得分[1, 4.5]映射到PESQ值[-0.5, 4.5]上，映射函数来源于：P.862.1"""
     inlog = (4.999 - mos) / (mos - 0.999)
     return (4.6607 - np.log(inlog)) / 1.4945
-
-
-# print(mos2pesq(3.518))

neverlib/tests/__init__.py

@@ -1 +1,33 @@
-# 测试包初始化文件 
+# This file is auto-generated. Do NOT edit manually.
+# Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .test_preprocess import test_NS_shape, test_pre_emphasis
+    from .test_vad import test_vad2nad
+
+# 运行时使用懒加载
+from lazy_loader import attach
+
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "test_imports",
+        "test_preprocess",
+        "test_vad",
+    ],
+    submod_attrs={
+        "test_preprocess": ['test_NS_shape', 'test_pre_emphasis'],
+        "test_vad": ['test_vad2nad'],
+    }
+)
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'test_NS_shape',
+        'test_pre_emphasis',
+        'test_vad2nad',
+    ]

neverlib/tests/test_imports.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+'''
+Author: 凌逆战 | Never
+Date: 2025-09-07
+Description: 测试neverlib导入功能
+'''
+import sys
+import os
+import time
+from neverlib.utils import get_path_list
+from neverlib.data_analyze.dataset_analyzer import AudioFileInfo
+
+# 确保当前目录在Python路径中，以便导入neverlib
+# sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+sys.path.append("../..")
+print("开始测试neverlib导入功能...")
+
+from neverlib.audio_aug import limiter

neverlib/utils/__init__.py

@@ -1,15 +1,71 @@
-from lazy_loader import attach
-
-__getattr__, __dir__, __all__ = attach(
-    __name__,
-    submodules=[
-        "utils",
-        "audio_split",
-        "checkGPU",
-        "message",
-    ],
-    submod_attrs={
-        "utils": ["EPS"],
-    },
-)
-
+# This file is auto-generated. Do NOT edit manually.
+# Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .audio_split import audio_split_VADfunasr, audio_split_VADsilero, audio_split_ffmpeg, audio_split_np, audio_split_pydub, audio_split_random, audio_split_sox
+    from .checkGPU import get_gpu_utilization, is_gpu_idle, monitor_gpu_utilization
+    from .lazy_expose import attach_and_expose_all
+    from .lazy_module import LazyModule
+    from .message import send_QQEmail, send_QQEmail_with_images
+    from .utils import DatasetSubfloderSplit, TrainValSplit, TrainValTestSplit, del_empty_folders, get_audio_segments, get_file_time, \
+        get_leaf_folders, get_path_list, pcm2wav, read_pcm, rename_files_and_folders, save_weight_histogram, wav2pcm, EPS
+
+# 运行时使用懒加载
+from lazy_loader import attach
+
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "audio_split",
+        "checkGPU",
+        "lazy_expose",
+        "lazy_module",
+        "message",
+        "utils",
+    ],
+    submod_attrs={
+        "audio_split": ['audio_split_VADfunasr', 'audio_split_VADsilero', 'audio_split_ffmpeg', 'audio_split_np', 'audio_split_pydub', 'audio_split_random', 'audio_split_sox'],
+        "checkGPU": ['get_gpu_utilization', 'is_gpu_idle', 'monitor_gpu_utilization'],
+        "lazy_expose": ['attach_and_expose_all'],
+        "lazy_module": ['LazyModule'],
+        "message": ['send_QQEmail', 'send_QQEmail_with_images'],
+        "utils": ['DatasetSubfloderSplit', 'TrainValSplit', 'TrainValTestSplit', 'del_empty_folders', 'get_audio_segments', 'get_file_time', 'get_leaf_folders',
+                  'get_path_list', 'pcm2wav', 'read_pcm', 'rename_files_and_folders', 'save_weight_histogram', 'wav2pcm', "EPS"],
+    }
+)
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'audio_split_VADfunasr',
+        'audio_split_VADsilero',
+        'audio_split_ffmpeg',
+        'audio_split_np',
+        'audio_split_pydub',
+        'audio_split_random',
+        'audio_split_sox',
+        'get_gpu_utilization',
+        'is_gpu_idle',
+        'monitor_gpu_utilization',
+        'attach_and_expose_all',
+        'LazyModule',
+        'send_QQEmail',
+        'send_QQEmail_with_images',
+        'DatasetSubfloderSplit',
+        'TrainValSplit',
+        'TrainValTestSplit',
+        'del_empty_folders',
+        'get_audio_segments',
+        'get_file_time',
+        'get_leaf_folders',
+        'get_path_list',
+        'pcm2wav',
+        'read_pcm',
+        'rename_files_and_folders',
+        'save_weight_histogram',
+        'wav2pcm',
+        'EPS',
+    ]

neverlib/utils/audio_split.py

@@ -10,7 +10,6 @@ from tqdm import tqdm
 import soundfile as sf
 import numpy as np
 from .utils import get_path_list
-from pydub import AudioSegment
 
 
 def audio_split_ffmpeg(source_path, target_path, sr, channel_num, duration, endwith="*.pcm"):
@@ -54,6 +53,7 @@ def audio_split_sox(source_path, target_path, duration, endwith="*.wav"):
     :param duration: 分割为时长(短音频)(单位:秒)
     :param endwith: 音频格式(只支持wav)
     """
+    assert endwith == "*.wav", "只支持wav格式的音频"
     wav_path_list = get_path_list(source_path, end=endwith)
 
     for wav_path in wav_path_list:
@@ -130,6 +130,11 @@ def audio_split_pydub(source_path, target_path, sr, channel_num, duration, endwi
     :param endwith: 音频格式(支持pcm和wav)
     :param sample_width: 音频的样本宽度(字节数), 默认为2, 表示16位音频
     """
+    try:
+        from pydub import AudioSegment
+    except Exception as e:
+        raise ImportError("需要安装 pydub 才能使用 audio_split_pydub: pip install pydub") from e
+
     assert duration % 1 == 0, "duration必须是1s的整数倍"
     wav_path_list = get_path_list(source_path, end=endwith)  # 获取音频文件列表
     print("待分割的音频数: ", len(wav_path_list))

neverlib/utils/checkGPU.py

@@ -6,14 +6,20 @@
 nohup python -u ./checkGPU.py > ./checkGPU.log 2>&1 &
 pid 5993
 """
-import subprocess
 import time
-import GPUtil
+import subprocess
 import numpy as np
-from neverlib.message import send_QQEmail
+from .message import send_QQEmail
 
 
 def is_gpu_idle():
+    try:
+        import GPUtil
+    except ImportError:
+        raise ImportError(
+            "GPUtil is required for is_gpu_idle(). "
+            "Please install it via `pip install checkGPU`."
+        )
     try:
         # 获取所有可见的GPU设备列表
         gpus = GPUtil.getGPUs()
@@ -49,12 +55,14 @@ def get_gpu_utilization():
 
 # 监控显卡利用率
 def monitor_gpu_utilization(check_interval=5, duration_limit=300, threshold=20,
-                            MonitorGPUs=[0, 1, 2, 3, 4, 5, 6, 7]):
+                            MonitorGPUs=[0, 1, 2, 3, 4, 5, 6, 7],
+                            from_email="xxxxx@qq.com",
+                            from_password="xxxxxxx",
+                            to_email="xxxxx@qq.com"):
     """
     check_interval = 5  每5s检查一次
     duration_limit = 300  检查300/60=5min
     threshold = 20  # 利用率阈值
-
     Returns:
     """
     alarm_times = 0  # 报警次数
@@ -91,10 +99,10 @@ def monitor_gpu_utilization(check_interval=5, duration_limit=300, threshold=20,
                 if t_now - last_alarm_time > 3600.0:
                     send_QQEmail(title=f"GPU利用率警告",
                                  content=f"GPU 利用率低于 {threshold}% 在 {duration_limit} 秒内. "
-                                         f"host ip: {host_ip}, 当前GPU利用率为:{utilization_mean}.",
-                                 from_email="1786088386@qq.com",
-                                 from_password="bgoirzjfkxshcbid",
-                                 to_email="1786088386@qq.com")
+                                 f"host ip: {host_ip}, 当前GPU利用率为:{utilization_mean}.",
+                                 from_email=from_email,
+                                 from_password=from_password,
+                                 to_email=to_email)
 
                     alarm_times += 1
                     last_alarm_time = time.time()