wav-loo 1.1.0

A multi-tool: WAV finder, multi-zone data generator (2/4 zones, multi-channel IR), and CLI aliases

WAV Loo

版本：1.1.0

wav-loo 是一个为音频处理和云原生开发设计的集成工具箱，它提供了三大核心功能：

1. WAV 文件查找器：在本地或远程URL中快速定位WAV音频文件。
2. 命令行快捷别名：集成了数十个常用的 kubectl、atlasctl 和其他开发运维命令，提升效率。
3. Loss 函数库 + 数据生成：提供损失函数与车载多音区数据生成工具。

安装

pip install wav-loo

功能一：WAV 文件查找

你可以在Python代码中或通过命令行来查找WAV文件。

命令行用法

wav-loo find <路径或URL> [--output <文件>] [--verbose]

• 示例:
  • wav-loo find /path/to/audio
  • wav-loo find https://example.com/audio-files/

Python API 用法

from wav_loo import WavFinder
finder = WavFinder()
wavs = finder.find_wav_files('/path/to/audio')
print(wavs)

功能二：快捷命令

通过 wav-loo 执行常用的运维开发命令，无需配置复杂的alias。

命令行用法

只需将别名作为 wav-loo 的子命令即可。

• 示例:

  # 查看pods (等价于 kubectl get pods -o wide)
  wav-loo kg
  
  # 查看signal命名空间日志 (等价于 kubectl logs -n signal)
  wav-loo kln
  
  # 使用unimirror安装numpy (等价于 uv pip install -i ... numpy)
  wav-loo uv numpy
  

支持的快捷命令列表

子命令	等价 bash 命令	说明
kd	kubectl delete pods	删除所有pods
kg	kubectl get pods -o wide	查看pods（详细）
kl	kubectl logs	查看日志
rs	kubectl describe ResourceQuota -n ...	查看资源配额（需补命名空间）
kdn	kubectl delete pods -n signal	删除signal命名空间pods
kgn	kubectl get pods -o wide -n signal	查看signal命名空间pods
kln	kubectl logs -n signal	查看signal命名空间日志
at	atlasctl top node	atlas节点监控
ad	atlasctl delete job	删除atlas作业
atd	atlasctl delete	删除atlas资源
adp	atlasctl delete job pytorchjob	删除pytorch作业
adn	atlasctl delete job -n signal	删除signal命名空间作业
tb	tensorboard --port=3027 --logdir=.	启动tensorboard
ca	conda activate <env>	激活conda环境（需补环境名）
gp	gpustat -i	查看GPU状态
kgg	kubectl get po --all-namespaces -o wide [extra args]	全局查看pods（可附加kubectl过滤参数）
uv	uv pip install -i ...	使用unimirror安装PyPI包（需补包名）

功能三：Loss 函数库

wav-loo 提供了一系列在信号处理和深度学习中常用的损失函数。

multi_channel_separation_consistency_loss

功能:

计算多通道（N>2）音频信号分离后的一致性损失。该损失旨在惩罚模型输出的多个通道之间的幅度谱差异与原始目标信号中对应通道之间的幅度谱差异不一致的情况。它通过计算所有通道对 (pair) 的预测幅度谱差异和目标幅度谱差异，然后最小化这两组差异之间的L1距离来实现。

使用方法:

import torch
from wav_loo.loss import multi_channel_separation_consistency_loss

# 假设 pred_mag_specs 和 target_mag_specs 是你的模型输出和目标真值
# 形状: (B, N, F, T), 类型: torch.float32
# B: batch_size, N: 通道数, F: 频率bins, T: 时间帧
pred_mag_specs = torch.randn(8, 4, 257, 100, dtype=torch.float32)
target_mag_specs = torch.randn(8, 4, 257, 100, dtype=torch.float32)

loss = multi_channel_separation_consistency_loss(pred_mag_specs, target_mag_specs)
print(f"一致性损失: {loss.item()}")

功能四：多通道车载语音数据生成

生成基于车载多位置 IR（房间脉冲响应）和干净语音的多音区数据集，用于语音分离等任务。

• IR 目录结构：ir_root 内部递归查找子目录名包含以下关键词的文件夹，并收集其中的 .wav IR：
  • zhujia, fujia, zhujiahoupai, fujiahoupai（不区分大小写，名称包含即可）
• 干净语音：clean_root 内递归查找 .wav 文件（多通道 clean 会先混为单声道）。
• 音区设置：
  • 4 音区（默认）：上述四个位置分别映射到输出通道 [0,1,2,3]。
  • 2 音区：zone_a=[zhujia, zhujiahoupai] → 通道 0；zone_b=[fujia, fujiahoupai] → 通道 1。
• 生成规则：
  • 每条样本为每个物理位置随机选一个 IR（IR 可为多通道），对 clean 做卷积；同一标签下（位置或音区）的多个子位置贡献相加。
  • 为每个标签得到一个 (T, N) 贡献（当 IR 通道数不足/超出 N 时会自动 pad/截断）。
  • 将所有标签的 (T, N) 堆叠后在“标签维度”求和，得到 mixture (T, N)。
  • target (T, N) 的第 i 列仅保留标签 i 在通道 i 的自通道成分，便于单位置监督。
  • 可配置 presence_prob 控制每个标签出现概率（默认 0.5）。
• 默认长度：每条干净语音默认最多使用 30 秒（--max-clean-seconds）。
• 输出结构：同目录下成对保存：
  • target_00000000.wav # (T, N)
  • mixture_00000000.wav # (T, N)

命令行用法

# 4 音区，默认 60000 条，16kHz，默认输出到 ./generated_car_data
# 默认不归一化，默认每条干净语音最长 30 秒
wav-loo gen --ir-dir /path/to/ir_root --clean-dir /path/to/clean_root

# 2 音区，自定义输出与数量
wav-loo gen \
  --ir-dir /path/to/ir_root \
  --clean-dir /path/to/clean_root \
  --zones 2 \
  --num-samples 2000 \
  --sample-rate 16000 \
  --output /path/to/out_dir

# 开启归一化，限制每条干净语音截断到 8 秒
wav-loo gen --ir-dir IR --clean-dir CLEAN --normalize --max-clean-seconds 8

Python API 用法

from wav_loo.data_gen import CarDataGenerator, GenerationConfig

cfg = GenerationConfig(
    ir_root_dir="/path/to/ir_root",
    clean_root_dir="/path/to/clean_root",
    output_dir="/path/to/out_dir",
    sample_rate=16000,
    num_samples=60000,
    zones=4,  # 或 2
    random_seed=42,
    normalize=False,      # 默认不归一化
    max_clean_seconds=30.0,
    presence_prob=0.5,
)
CarDataGenerator(cfg).generate()

数据生成依赖

• 运行数据生成需要：numpy、soundfile（依赖系统库 libsndfile）、scipy。

依赖

• Python 3.7+
• requests
• beautifulsoup4
• urllib3

许可证

MIT License