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

neverlib/filter/__init__.py

@@ -1,20 +1,40 @@
-'''
-Author: 凌逆战 | Never
-Date: 2025-03-17 19:23:33
-Description: 
-'''
-"""
-节省路径
-from neverlib.filter import common
-如果没有用户必须完整路径
-from neverlib.filter.common import *
-"""
-from lazy_loader import attach
-
-__getattr__, __dir__, __all__ = attach(
-    __name__,
-    submodules=["common", "core", "biquad"],
-    submod_attrs={
-        "common": ["HPFilter", "LPFilter", "HPFilter_torch"],
-    },
-)
+# 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 .biquad import BiquadFilter
+    from .common import HPFilter, HPFilter_torch, LPFilter
+    from .core import EQFilter, EQ_test, eq_process, eq_process_test
+
+# 运行时使用懒加载
+from lazy_loader import attach
+
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "biquad",
+        "common",
+        "core",
+    ],
+    submod_attrs={
+        "biquad": ['BiquadFilter'],
+        "common": ['HPFilter', 'HPFilter_torch', 'LPFilter'],
+        "core": ['EQFilter', 'EQ_test', 'eq_process', 'eq_process_test'],
+    }
+)
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'BiquadFilter',
+        'HPFilter',
+        'HPFilter_torch',
+        'LPFilter',
+        'EQFilter',
+        'EQ_test',
+        'eq_process',
+        'eq_process_test',
+    ]

neverlib/filter/auto_eq/__init__.py

@@ -1,36 +1,55 @@
-'''
-Author: 凌逆战 | Never
-Date: 2025-08-19 21:26:54
-Description: 
-AudoEQ - 自动EQ补偿模块
-Author: 凌逆战 | Never
+# This file is auto-generated. Do NOT edit manually.
+# Generated by neverlib.QA.gen_init
 
-该模块提供多种自动EQ补偿方法:
-- 频谱直接补偿 (auto_eq_spectral_direct)
-- 差分进化优化 (auto_eq_de)  
-- 遗传算法基础版 (auto_eq_ga_basic)
-- 遗传算法高级版 (auto_eq_ga_advanced)
-'''
+from typing import TYPE_CHECKING, Any
 
-# 频谱直接补偿方法
-from .freq_eq import get_freq_eq
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .de_eq import get_filter_function, match_frequency_response, plot_spectra_comparison
+    from .freq_eq import get_freq_eq
+    from .ga_eq_advanced import EQConfig, EQOptimizer, load_config_from_yaml, main
+    from .ga_eq_basic import custom_mutate, evaluate_individual, generate_active_gene, generate_dbgain_gene, generate_fc_gene, generate_q_gene, generate_type_gene, get_combined_eq_response_db, get_magnitude_spectrum_db, get_single_filter_freq_response_db_from_coeffs, individual_creator, main_ga
 
-# 差分进化优化方法
-# from .de_eq import (
-#     get_filter_function,
-#     match_frequency_response,
-#     plot_spectra_comparison
-# )
+# 运行时使用懒加载
+from lazy_loader import attach
 
-# 遗传算法基础版
-# from .ga_eq_basic import (
-#     individual_creator,
-#     get_magnitude_spectrum_db,
-#     get_single_filter_freq_response_db_from_coeffs,
-#     get_combined_eq_response_db,
-#     evaluate_individual,
-#     custom_mutate,
-# )
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "de_eq",
+        "freq_eq",
+        "ga_eq_advanced",
+        "ga_eq_basic",
+    ],
+    submod_attrs={
+        "de_eq": ['get_filter_function', 'match_frequency_response', 'plot_spectra_comparison'],
+        "freq_eq": ['get_freq_eq'],
+        "ga_eq_advanced": ['EQConfig', 'EQOptimizer', 'load_config_from_yaml', 'main'],
+        "ga_eq_basic": ['custom_mutate', 'evaluate_individual', 'generate_active_gene', 'generate_dbgain_gene', 'generate_fc_gene', 'generate_q_gene', 'generate_type_gene', 'get_combined_eq_response_db', 'get_magnitude_spectrum_db', 'get_single_filter_freq_response_db_from_coeffs', 'individual_creator', 'main_ga'],
+    }
+)
 
-# 遗传算法高级版
-# from .ga_eq_advanced import EQOptimizer
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'get_filter_function',
+        'match_frequency_response',
+        'plot_spectra_comparison',
+        'get_freq_eq',
+        'EQConfig',
+        'EQOptimizer',
+        'load_config_from_yaml',
+        'main',
+        'custom_mutate',
+        'evaluate_individual',
+        'generate_active_gene',
+        'generate_dbgain_gene',
+        'generate_fc_gene',
+        'generate_q_gene',
+        'generate_type_gene',
+        'get_combined_eq_response_db',
+        'get_magnitude_spectrum_db',
+        'get_single_filter_freq_response_db_from_coeffs',
+        'individual_creator',
+        'main_ga',
+    ]

neverlib/filter/auto_eq/de_eq.py

@@ -2,8 +2,6 @@
 # Author: AI Assistant based on User's Demand
 # Date: 2023-10-27 (Using Differential Evolution)
 # Modified: 2025-01-05 (Adapted for new filters.py structure)
-import sys
-sys.path.append("..")
 import numpy as np
 import librosa
 import soundfile as sf

neverlib/filter/common.py

@@ -3,10 +3,8 @@ Author: 凌逆战 | Never
 Date: 2025-08-05 23:42:08
 Description: 一些基础和通用的滤波器
 '''
-import torch
 import numpy as np
 from scipy import signal
-import torchaudio.functional as F
 
 
 def HPFilter(wav, sr=16000, order=6, cutoff=100):
@@ -17,7 +15,12 @@ def HPFilter(wav, sr=16000, order=6, cutoff=100):
     :param cutoff: 截止频率
     :return:
     """
-    b, a = signal.butter(order, cutoff, btype='highpass', analog=False, output='ba', fs=sr)
+    b, a = signal.butter(order,
+                         cutoff,
+                         btype='highpass',
+                         analog=False,
+                         output='ba',
+                         fs=sr)
     wav = signal.lfilter(b, a, wav, axis=0)
     return wav.astype(np.float32)
 
@@ -30,16 +33,32 @@ def LPFilter(wav, sr=16000, order=6, cutoff=100):
     :param cutoff: 截止频率
     :return:
     """
-    b, a = signal.butter(order, cutoff, btype='lowpass', analog=False, output='ba', fs=sr)
+    b, a = signal.butter(order,
+                         cutoff,
+                         btype='lowpass',
+                         analog=False,
+                         output='ba',
+                         fs=sr)
     wav = signal.lfilter(b, a, wav, axis=0)
     return wav.astype(np.float32)
 
+
 def HPFilter_torch(wav, sr=16000, order=6, cutoff=100):
     """
     Args:
         wav: (B,T)
     """
-    b, a = signal.butter(order, cutoff, btype='highpass', analog=False, output='ba', fs=sr)
+    try:
+        import torch
+        import torchaudio.functional as F
+    except Exception as e:
+        raise ImportError("需要安装 torch 和 torchaudio 才能使用 HPFilter_torch") from e
+    b, a = signal.butter(order,
+                         cutoff,
+                         btype='highpass',
+                         analog=False,
+                         output='ba',
+                         fs=sr)
 
     # 将滤波器系数转换为 torch 张量
     b = torch.tensor(b, dtype=torch.float32)

neverlib/metrics/__init__.py

@@ -0,0 +1,59 @@
+# 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 .dnsmos import ComputeScore
+    from .lpc_lsp import framing, lpc_lsp_distance, lpc_to_lsp
+    from .snr import get_snr, get_snr_from_noisy, psnr, seg_snr, si_sdr
+    from .spec import lsd, mcd, sd
+    from .test_pesq import mos2pesq, pesq2mos
+    from .time import dc_offset, mean_rms_amplitude, peak_amplitude, rms_amplitude
+
+# 运行时使用懒加载
+from lazy_loader import attach
+
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "dnsmos",
+        "lpc_lsp",
+        "snr",
+        "spec",
+        "test_pesq",
+        "time",
+    ],
+    submod_attrs={
+        "dnsmos": ['ComputeScore'],
+        "lpc_lsp": ['framing', 'lpc_lsp_distance', 'lpc_to_lsp'],
+        "snr": ['get_snr', 'get_snr_from_noisy', 'psnr', 'seg_snr', 'si_sdr'],
+        "spec": ['lsd', 'mcd', 'sd'],
+        "test_pesq": ['mos2pesq', 'pesq2mos'],
+        "time": ['dc_offset', 'mean_rms_amplitude', 'peak_amplitude', 'rms_amplitude'],
+    }
+)
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'ComputeScore',
+        'framing',
+        'lpc_lsp_distance',
+        'lpc_to_lsp',
+        'get_snr',
+        'get_snr_from_noisy',
+        'psnr',
+        'seg_snr',
+        'si_sdr',
+        'lsd',
+        'mcd',
+        'sd',
+        'mos2pesq',
+        'pesq2mos',
+        'dc_offset',
+        'mean_rms_amplitude',
+        'peak_amplitude',
+        'rms_amplitude',
+    ]

neverlib/metrics/dnsmos.py

@@ -5,17 +5,10 @@ Description:
 要计算个性化 MOS 分数（干扰说话者受到惩罚），请提供“-p”参数，例如：python dnsmos.py -t ./SampleClips -o sample.csv -p
 要计算常规 MOS 分数，请省略“-p”参数。例如：python dnsmos.py -t ./SampleClips -o sample.csv
 '''
-import argparse
-import concurrent.futures
-import glob
-import os
 import librosa
 import numpy as np
-import onnxruntime as ort
-import pandas as pd
+import onnxruntime
 import soundfile as sf
-from tqdm import tqdm
-from neverlib.utils import get_path_list
 
 
 class ComputeScore:
@@ -29,8 +22,8 @@ class ComputeScore:
         else:
             primary_model_path = "./DNSMOS/sig_bak_ovr.onnx"
 
-        self.onnx_sess = ort.InferenceSession(primary_model_path)
-        self.p808_onnx_sess = ort.InferenceSession(p808_model_path)
+        self.onnx_sess = onnxruntime.InferenceSession(primary_model_path)
+        self.p808_onnx_sess = onnxruntime.InferenceSession(p808_model_path)
 
     def audio_melspec(self, audio, n_mels=120, frame_size=320, hop_length=160, to_db=True):
         mel_spec = librosa.feature.melspectrogram(y=audio, sr=self.sr, n_fft=frame_size + 1, hop_length=hop_length, n_mels=n_mels)
@@ -102,7 +95,7 @@ class ComputeScore:
         return out_dict
 
 
-def main():
+if __name__ == "__main__":
     SAMPLING_RATE = 16000
     INPUT_LENGTH = 9.01
     is_personalized_MOS = False
@@ -111,7 +104,3 @@ def main():
     compute_score = ComputeScore(is_personalized_MOS, SAMPLING_RATE, INPUT_LENGTH)
     data = compute_score(testset_dir)
     print(data)
-
-
-if __name__ == "__main__":
-    main()