neverlib 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

neverlib/QA/ImpactNoiseRejection.py

@@ -0,0 +1,119 @@
+"""VPU 冲击声检测 + 可视化（轻量版）。"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import List, Tuple
+
+import matplotlib.pyplot as plt
+import numpy as np
+import soundfile as sf
+
+# 基本参数，可按需修改
+AUDIO_PATH = Path(__file__).resolve().parent / "out.wav"
+CHANNEL_INDEX = 4
+DC_WINDOW_MS = 40.0
+SHORT_MS = 6.0
+LONG_MS = 60.0
+RATIO_THRESHOLD = 3.0
+MIN_DURATION_MS = 8.0
+MIN_GAP_MS = 35.0
+
+
+def moving_average(x: np.ndarray, window: int) -> np.ndarray:
+    if window <= 1:
+        return x
+    kernel = np.ones(window, dtype=np.float64) / window
+    return np.convolve(x, kernel, mode="same")
+
+
+def preprocess(signal: np.ndarray, sr: int) -> np.ndarray:
+    dc_window = max(int(sr * DC_WINDOW_MS / 1000.0), 1)
+    return signal - moving_average(signal, dc_window)
+
+
+def ratio_score(signal: np.ndarray, sr: int) -> np.ndarray:
+    short = max(int(sr * SHORT_MS / 1000.0), 1)
+    long = max(int(sr * LONG_MS / 1000.0), 1)
+    short_env = moving_average(np.abs(signal), short)
+    long_env = moving_average(np.abs(signal), long)
+    return short_env / np.maximum(long_env, 1e-6)
+
+
+def detect_events(mask: np.ndarray, sr: int) -> List[Tuple[float, float]]:
+    min_samples = max(int(sr * MIN_DURATION_MS / 1000.0), 1)
+    gap_samples = max(int(sr * MIN_GAP_MS / 1000.0), 1)
+
+    events: List[Tuple[int, int]] = []
+    start = -1
+    for idx, active in enumerate(mask):
+        if active and start < 0:
+            start = idx
+        elif not active and start >= 0:
+            if idx - start >= min_samples:
+                events = _append_or_merge(events, start, idx, gap_samples)
+            start = -1
+
+    if start >= 0 and len(mask) - start >= min_samples:
+        events = _append_or_merge(events, start, len(mask), gap_samples)
+
+    return [(s / sr, e / sr) for s, e in events]
+
+
+def _append_or_merge(events: List[Tuple[int, int]], start: int, end: int, gap: int) -> List[Tuple[int, int]]:
+    if events and start - events[-1][1] <= gap:
+        previous_start, _ = events[-1]
+        events[-1] = (previous_start, end)
+    else:
+        events.append((start, end))
+    return events
+
+
+def plot_detection(time: np.ndarray, signal: np.ndarray, score: np.ndarray, events: List[Tuple[float, float]]) -> None:
+    fig, axes = plt.subplots(2, 1, figsize=(12, 6), sharex=True)
+
+    axes[0].plot(time, signal, color="steelblue", linewidth=0.8)
+    axes[0].set_ylabel("Amplitude")
+    axes[0].set_title("VPU 信号与冲击事件")
+
+    axes[1].plot(time, score, color="crimson", linewidth=0.8)
+    axes[1].axhline(RATIO_THRESHOLD, color="gray", linestyle="--", linewidth=0.8, label="阈值")
+    axes[1].set_ylabel("短/长包络比")
+    axes[1].set_xlabel("时间 (s)")
+    axes[1].legend(loc="upper right")
+
+    for axis in axes:
+        for start, end in events:
+            axis.axvspan(start, end, color="lime", alpha=0.25)
+        axis.grid(alpha=0.2)
+
+    fig.tight_layout()
+    plt.savefig('./impact_noise_rejection.png')
+
+
+def main() -> None:
+    if not AUDIO_PATH.exists():
+        raise FileNotFoundError(f"找不到音频文件: {AUDIO_PATH}")
+
+    data, sr = sf.read(AUDIO_PATH, always_2d=True)
+    if CHANNEL_INDEX >= data.shape[1]:
+        raise ValueError(f"音频仅有 {data.shape[1]} 通道，无法访问 VPU 通道 {CHANNEL_INDEX}")
+
+    signal = data[:, CHANNEL_INDEX].astype(np.float64)
+    signal = preprocess(signal, sr)
+    score = ratio_score(signal, sr)
+    events = detect_events(score > RATIO_THRESHOLD, sr)
+
+    if events:
+        print("检测到的冲击事件：")
+        for start, end in events:
+            print(f"  起始 {start:.3f}s  结束 {end:.3f}s  持续 {(end-start)*1000:.1f}ms")
+    else:
+        print("未检测到明显的冲击事件。")
+
+    time_axis = np.arange(len(signal)) / sr
+    plot_detection(time_axis, signal, score, events)
+
+
+if __name__ == "__main__":
+    main()

neverlib/QA/gen_init.py

@@ -1,4 +1,5 @@
 # 自动生成指定包目录下的 __init__.py（懒加载格式：lazy_loader.attach）
+# 并且支持IDE友好的 __init__.py 版本
 
 import ast
 from pathlib import Path
@@ -43,7 +44,7 @@ def _extract_exports_from_module(py_file: Path) -> List[str]:
     return sorted(exports)
 
 
-def generate_init_for_directory(package_dir: Path) -> Path:
+def generate_init_for_directory(package_dir: Path, ide_friendly: bool = False) -> Path:
     """
     为指定目录生成懒加载版 __init__.py（覆盖写入）。
     - 仅扫描一级子模块（同级 .py 文件），忽略以下划线开头的模块与 __init__.py 本身。
@@ -55,6 +56,10 @@ def generate_init_for_directory(package_dir: Path) -> Path:
             submodules=["m1", "m2", ...],
             submod_attrs={"m1": ["A", "B"], ...}
         )
+
+    Args:
+        package_dir: 包目录路径
+        ide_friendly: 是否生成IDE友好版本（使用TYPE_CHECKING）
     """
     package_dir = package_dir.resolve()
     if not package_dir.is_dir():
@@ -76,6 +81,24 @@ def generate_init_for_directory(package_dir: Path) -> Path:
     lines: List[str] = []
     lines.append('# This file is auto-generated. Do NOT edit manually.')
     lines.append('# Generated by neverlib.QA.gen_init')
+
+    if ide_friendly:
+        lines.append('')
+        lines.append('from typing import TYPE_CHECKING, Any')
+        lines.append('')
+        lines.append('if TYPE_CHECKING:')
+        lines.append('    # 仅在类型检查时导入，提供IDE补全支持')
+
+        # 生成 TYPE_CHECKING 导入
+        for module_name in sorted(module_to_exports.keys()):
+            exports = module_to_exports[module_name]
+            if exports:
+                imports = ', '.join(exports)
+                lines.append(f'    from .{module_name} import {imports}')
+
+        lines.append('')
+        lines.append('# 运行时使用懒加载')
+
     lines.append('from lazy_loader import attach')
     lines.append('')
     lines.append('__getattr__, __dir__, __all__ = attach(')
@@ -93,24 +116,102 @@ def generate_init_for_directory(package_dir: Path) -> Path:
     lines.append(')')
     lines.append('')
 
+    if ide_friendly:
+        # 生成显式的 __all__ 声明
+        lines.append('# 显式声明 __all__ 以便 IDE 识别')
+        lines.append('if TYPE_CHECKING:')
+        lines.append('    __all__ = [')
+
+        all_exports = []
+        for module_name in sorted(module_to_exports.keys()):
+            exports = module_to_exports[module_name]
+            for export in exports:
+                all_exports.append(f"        '{export}',")
+
+        lines.extend(all_exports)
+        lines.append('    ]')
+
     init_file = package_dir / '__init__.py'
     init_file.write_text('\n'.join(lines), encoding='utf-8')
     return init_file
 
 
+def generate_all_packages(root_dir: Path, ide_friendly: bool = False) -> List[Path]:
+    """
+    为所有子包生成 __init__.py 文件
+
+    Args:
+        root_dir: 根目录路径
+        ide_friendly: 是否生成IDE友好版本
+
+    Returns:
+        生成的文件路径列表
+    """
+    generated_files = []
+    processed_dirs = set()  # 避免重复处理
+
+    # 查找所有包含 __init__.py 的包目录
+    for init_file in root_dir.rglob('__init__.py'):
+        package_dir = init_file.parent
+
+        # 跳过根目录和已处理的目录
+        if package_dir == root_dir or package_dir in processed_dirs:
+            continue
+
+        # 检查是否包含其他Python文件（不是只有__init__.py）
+        has_other_py_files = any(
+            f.name != '__init__.py' and f.suffix == '.py'
+            for f in package_dir.iterdir()
+        )
+
+        if has_other_py_files:
+            processed_dirs.add(package_dir)
+            try:
+                init_path = generate_init_for_directory(package_dir, ide_friendly=ide_friendly)
+                generated_files.append(init_path)
+                print(f'✅ 已生成: {init_path}')
+            except Exception as e:
+                print(f'❌ 生成失败 {package_dir}: {e}')
+
+    return generated_files
+
+
 def main():
     import argparse
 
     parser = argparse.ArgumentParser(description='为指定包目录自动生成 __init__.py')
     parser.add_argument('-p', '--package-dir', dest='package_dir',
-                        default='../../neverlib/vad',
                         help='包目录路径，比如 /path/to/pkg 或 ./neverlib/utils')
+    parser.add_argument('--all', action='store_true',
+                        help='为所有子包生成 __init__.py')
+    parser.add_argument('--ide-friendly', action='store_true',
+                        help='生成IDE友好版本（使用TYPE_CHECKING）')
+    parser.add_argument('--backup', action='store_true',
+                        help='备份原始文件')
     args = parser.parse_args()
 
-    target_dir = Path(args.package_dir)
-    init_path = generate_init_for_directory(target_dir)
-    print(f'已生成: {init_path}')
-# python -m neverlib.QA.gen_init /data03/never/Desktop/neverlib/neverlib/utils
+    if args.all:
+        # 批量生成所有包
+        root_dir = Path(__file__).parent.parent  # neverlib 目录
+        generated_files = generate_all_packages(root_dir, ide_friendly=args.ide_friendly)
+    else:
+        # 生成单个包
+        if not args.package_dir:
+            parser.error('请指定包目录路径 (-p) 或使用 --all 批量生成')
+
+        target_dir = Path(args.package_dir)
+
+        if args.backup and (target_dir / '__init__.py').exists():
+            backup_path = target_dir / '__init__.py.backup'
+            (target_dir / '__init__.py').rename(backup_path)
+            print(f"已备份原始文件到: {backup_path}")
+
+        init_path = generate_init_for_directory(target_dir, ide_friendly=args.ide_friendly)
+        print(f'已生成: {init_path}')
+        if args.ide_friendly:
+            print('✅ 已生成IDE友好版本（支持代码补全和类型检查）')
+        else:
+            print('ℹ️  生成标准懒加载版本，如需IDE支持请使用 --ide-friendly 参数')
 
 
 if __name__ == '__main__':

neverlib/__init__.py

@@ -37,6 +37,13 @@ try:
 except Exception:
     __version__ = "0.1.2"  # 如果出错, 使用默认版本号
 
+from typing import TYPE_CHECKING, Any
+
+
+if TYPE_CHECKING:
+    from .utils import get_path_list
+    from .filter import HPFilter
+    from .audio_aug import volume_norm
 
 # 懒加载子包，减少初始导入开销
 from lazy_loader import attach
@@ -45,4 +52,16 @@ __getattr__, __dir__, __all__ = attach(
     __name__,
     submodules=["audio_aug", "data_analyze", "filter", "metrics", "utils", "vad", ],
     # 只导出子模块，不直接导出函数
+    submod_attrs={
+        "utils": ["get_path_list"],
+        "filter": ["HPFilter"],
+        "audio_aug": ["volume_norm"],
+    }
 )
+
+if TYPE_CHECKING:
+    __all__ = [
+        "get_path_list",
+        "HPFilter",
+        "volume_norm",
+    ]

neverlib/audio_aug/README.md

@@ -0,0 +1,3 @@
+https://github.com/iver56/audiomentations
+https://github.com/gfdb/wav2aug
+https://github.com/documentation-ai/DocumentationAI-Docs

neverlib/audio_aug/__init__.py

@@ -6,6 +6,7 @@ __getattr__, __dir__, __all__ = attach(
     __name__,
     submodules=[
         "HarmonicDistortion",
+        "TFDrop",
         "TFMask",
         "audio_aug",
         "clip_aug",
@@ -17,10 +18,7 @@ __getattr__, __dir__, __all__ = attach(
     submod_attrs={
         "HarmonicDistortion": ['apply_harmonic_distortion', 'apply_pedalboard_distortion'],
         "TFMask": ['FreqMask', 'TimeMask'],
-        "audio_aug": ['add_reverb', 'get_snr_use_vad', 'limiter', 'measure_loudness', 'snr_aug_Interpolation',
-                      'snr_aug_changeClean', 'snr_aug_changeNoise', 'snr_aug_vad_Interpolation',
-                      'snr_diff_changeClean', 'snr_diff_changeNoise', 'volume_aug', 'volume_aug_dbrms',
-                      'volume_aug_linmax', 'volume_aug_lufs', 'volume_convert', 'volume_norm'],
+        "audio_aug": ['add_reverb', 'get_snr_use_vad', 'limiter', 'measure_loudness', 'snr_aug_Interpolation', 'snr_aug_changeClean', 'snr_aug_changeNoise', 'snr_aug_vad_Interpolation', 'snr_diff_changeClean', 'snr_diff_changeNoise', 'volume_aug', 'volume_aug_dbrms', 'volume_aug_linmax', 'volume_aug_lufs', 'volume_convert', 'volume_norm'],
         "clip_aug": ['clipping_aug'],
         "coder_aug": ['aac_aug_save', 'amr_nb_aug', 'amr_wb_aug', 'flac_aug', 'flac_encode_save', 'mp3_aug', 'opus_aug_save', 'vorbis_aug'],
         "coder_aug2": ['apply_codec_distortion', 'check_codec_available'],

neverlib/filter/core.py

@@ -322,9 +322,8 @@ def EQ_test():
     from scipy import signal
 
     fs = 16000
-
-    # b, a = LowshelfFilter(1500, 15, Q=0.5)
-    b, a = PeakingFilter(1500, 5, Q=0.5)
+    eq = EQFilter(fs)
+    b, a = eq.LowshelfFilter(4000, 6, Q=0.7)
     print(b)
     print(a)
 
@@ -335,5 +334,9 @@ def EQ_test():
     plt.ylabel('Amplitude [dB]')
     plt.grid(which='both', axis='both')  # 显示网格
     # 画红色的垂直线, 标记截止频率
-    plt.xscale('log')  # x轴对数化
-    plt.show()
+    # plt.xscale('log')  # x轴对数化
+    plt.savefig('./lowshelf_freq_response.png')
+
+
+if __name__ == "__main__":
+    EQ_test()

neverlib/metrics/README.md

@@ -0,0 +1,35 @@
+https://github.com/ftshijt/speech_evaluation?tab=readme-ov-file
+https://github.com/Amir-Ivry/MAPSS-measures
+https://github.com/espnet/espnet/blob/f517b63fe5defb865e2067b11c2dfc516150868d/utils/eval-source-separation.py#L361
+
+
+
+
+
+F0 Correlation
+
+F0 Root Mean Square Error
+sdr
+isr
+sir
+sar
+perm
+
+
+Convolutional scale-invariant signal-to-distortion ratio (CI-SDR) [x]
+Scale-invariant signal-to-noise ratio (SI-SNR) [x]
+Speech BERT Score
+Discrete Speech BLEU Score [x]
+Discrete Speech Token Edit Distance
+
+
+
+
+STOI
+ESTOI
+PESQ
+
+
+音频特征：
+https://github.com/libAudioFlux/audioFlux
+https://github.com/csteinmetz1/pyloudnorm

neverlib/metrics/__init__.py

@@ -13,7 +13,7 @@ __getattr__, __dir__, __all__ = attach(
         "time",
     ],
     submod_attrs={
-        "dnsmos": ['ComputeScore', 'main'],
+        "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'],

neverlib/signal_gen/babble_noise_generate.py

@@ -0,0 +1,113 @@
+# -*- coding:utf-8 -*-
+# Author:凌逆战 | Never
+# Date: 2024/6/14
+"""
+babble噪声生成
+"""
+import os
+import torch
+import random
+import numpy as np
+import soundfile as sf
+import pickle
+
+from neverlib.utils import get_path_list
+from vad import EnergyVAD
+from utils.audio_aug import volume_aug, add_reverb
+from joblib import Parallel, delayed
+from multiprocessing import Pool
+
+vad = EnergyVAD(
+    sample_rate=16000,
+    frame_length=25,  # in milliseconds
+    frame_shift=20,  # in milliseconds
+    energy_threshold=0.05,  # you may need to adjust this value
+    pre_emphasis=0.95,
+)  # default values are used here
+
+
+def get_one_babble(speech_path_list, rir_path_list):
+    sub_len = 0
+    sub_list = []
+    while sub_len < sentence_len:
+        speech_path = random.choice(speech_path_list)
+        rir_path = random.choice(rir_path_list)
+        # 加rir
+        with open(rir_path, 'rb') as f:
+            # arrayCoor (2,3) (num_mic, 3(x,y,z))
+            # sPos (4,3) (num_source, 3(x,y,z))
+            [arrayCoor, sPos, sAziPth, rir, L, rt, c, mtype, n, fs] = pickle.load(f)
+            rir = np.array(rir)  # (souece, mic, T) (4, 3, 2668)
+            rir_idx = np.random.randint(0, rir.shape[0])  # 随机选择一个source
+            sPos = sPos[rir_idx]  # (3,) (x,y,z)
+            DistMic2Source_1 = np.sqrt(np.sum((arrayCoor[0] - sPos) ** 2))
+            DistMic2Source_2 = np.sqrt(np.sum((arrayCoor[1] - sPos) ** 2))
+            if DistMic2Source_1 < 2 or DistMic2Source_2 < 2:
+                continue
+            rir = rir[rir_idx]
+
+        wav, _ = sf.read(speech_path, dtype='float32', always_2d=True)
+        assert wav.shape[-1] == 1, f"音频应该为单通道"
+        if len(wav) > sentence_len:
+            idx = random.randint(0, len(wav) - sentence_len)
+            wav = wav[idx:idx + sentence_len]
+
+        wav = wav / np.abs(wav).max()  # 归一化
+        wav = vad.apply_vad(wav.T).T
+
+        wav = add_reverb(wav, rir, add_rir_ratio=1)  # 双麦wav
+        # 音量增强
+        rms = np.sqrt(np.mean(wav ** 2))
+        wav *= random.uniform(0.5, 1.0) / rms
+
+        sub_len += len(wav)
+        sub_list.append(wav)
+
+    sub_babble = np.concatenate(sub_list, axis=0)
+    if sub_len >= sentence_len:
+        sub_babble = sub_babble[:sentence_len]
+    return sub_babble
+
+
+def babble_noise_generate(i, speech_path_list, rir_path_list):
+    source_num = random.randint(15, 25)
+    babble_noise = np.zeros((sentence_len, 2))
+    for _ in range(source_num):
+        sub_babble = get_one_babble(speech_path_list, rir_path_list)
+        babble_noise += sub_babble
+
+    babble_path = os.path.join(target_dir, f"babble_noise_{i}.wav")
+    print(babble_path)
+    babble_noise = babble_noise / np.max(np.abs(babble_noise)) * 0.5
+    sf.write(babble_path, babble_noise, sr)
+
+
+if __name__ == "__main__":
+    speech_dir = "/data/never/kws_data/BigDataset/speech"
+    rir_dir = "/data/never/RIR/2mic_Line_r50mm_fs16k_4source_t60mix08_maxdist8m_2w"
+    target_dir = "/data/never/SE_dataset/babble_noise10"
+    # if os.path.exists(target_dir):
+    #     os.system(f"rm -rf {target_dir}")
+    #     os.makedirs(target_dir, exist_ok=True)
+    # else:
+    #     os.makedirs(target_dir, exist_ok=True)
+    speech_path_list = get_path_list(speech_dir, "*.wav")
+    rir_path_list = get_path_list(rir_dir, "*.rir")
+
+    sentence_num = 20000
+    sr = 16000
+    volume_range = [0.5, 1.0]
+    sentence_len = sr * 10
+    # for i in range(10):
+    #     babble_noise_generate(i, speech_path_list, rir_path_list)
+
+    # 多核并行处理
+    # Parallel(n_jobs=-1)(delayed(babble_noise_generate)(i, speech_path_list, rir_path_list) for i in range(sentence_num))
+
+    pool = Pool(32)  # 创建进程池, 一定要放在 事件函数之后, 不然会保错
+
+    for i in range(sentence_num):
+        r = pool.apply_async(func=babble_noise_generate, args=(i, speech_path_list, rir_path_list,))
+
+    pool.close()  # 关闭进程池
+    pool.join()  # 回收进程池

neverlib/tests/__init__.py

@@ -1 +1,16 @@
-# 测试包初始化文件 
+# This file is auto-generated. Do NOT edit manually.
+# Generated by neverlib.QA.gen_init
+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'],
+    }
+)

neverlib/tests/test_imports.py

@@ -8,6 +8,8 @@ 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__))))

neverlib/utils/README.md

@@ -0,0 +1,29 @@
+prsync
+来源：https://github.com/nathanhaigh/parallel-rsync
+用法
+如果您的命令如下所示：rsync
+
+```bash
+rsync \
+  --times --recursive --progress \
+  --exclude "raw_reads" --exclude ".snakemake" \
+  user@example.com:/my_remote_dir/ /my_local_dir/
+```
+只需替换此脚本的可执行文件：rsync
+```bash
+./prsync \
+  --times --recursive --progress \
+  --exclude "raw_reads" --exclude ".snakemake" \
+  user@example.com:/my_remote_dir/ /my_local_dir/
+  ```
+
+并行作业数
+默认情况下，脚本将为计算机上的每个处理器使用 1 个并行作业。 这是由确定的，如果失败，我们将回退到并行作业来传输文件。 
+可以通过使用作为脚本的第一个命令行参数来覆盖此行为：```nproc 10 --parallel```
+```bash
+./prsync \
+  --parallel=20 \
+  --times --recursive --progress \
+  --exclude "raw_reads" --exclude ".snakemake" \
+  user@example.com:/my_remote_dir/ /my_local_dir/
+```

neverlib/utils/__init__.py

@@ -7,20 +7,21 @@ __getattr__, __dir__, __all__ = attach(
     submodules=[
         "audio_split",
         "checkGPU",
+        "floder",
         "lazy_expose",
         "lazy_module",
         "message",
+        "pcm",
         "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'],
+        "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', 'get_gpu_utilization2', 'monitor_gpu_utilization'],
+        "floder": ['change_path', 'del_empty_folders', 'get_leaf_folders', 'rename_files_and_folders'],
         "lazy_expose": ['attach_and_expose_all'],
         "lazy_module": ['LazyModule'],
         "message": ['send_QQEmail', 'send_QQEmail_with_images'],
-        "utils": ['EPS', 'DatasetSubfloderSplit', 'TrainValSplit', 'TrainValTestSplit', 'del_empty_folders',
-                  'get_audio_segments', 'get_file_time', 'get_leaf_folders', 'get_path_list', 'pcm2wav',
-                  'rename_files_and_folders', 'save_weight_histogram', 'wav2pcm'],
+        "pcm": ['pcm2wav', 'read_pcm', 'wav2pcm'],
+        "utils": ['DatasetSubfloderSplit', 'TrainValSplit', 'TrainValTestSplit', 'get_audio_segments', 'get_file_time', 'get_path_list', 'save_weight_histogram'],
     }
 )