NBSS项目安装和配置指南

NBSS项目安装和配置指南

NBSS 项目地址: https://gitcode.com/gh_mirrors/nb/NBSS

1. 项目基础介绍和主要编程语言

NBSS（Narrow-band and Broadband SpatialNet）是一个用于多通道语音分离、降噪和去混响的开源项目。该项目由Audio-WestlakeU团队开发，主要用于处理多通道音频数据，旨在提高语音信号的质量。NBSS项目主要使用Python语言进行开发，依赖于PyTorch深度学习框架。

2. 项目使用的关键技术和框架

NBSS项目主要使用了以下关键技术和框架：

• PyTorch：一个开源的深度学习框架，用于构建和训练神经网络模型。
• PyTorch Lightning：一个轻量级的PyTorch包装器，简化了训练和测试过程。
• CUDA：用于GPU加速的并行计算平台和API。
• GPURIR：一个用于生成房间脉冲响应（RIR）的库，用于模拟声学环境。

3. 项目安装和配置的准备工作和详细安装步骤

准备工作

在开始安装和配置NBSS项目之前，请确保您的系统满足以下要求：

• Python 3.7或更高版本：项目依赖于Python 3.7或更高版本。
• CUDA支持的GPU：项目需要GPU加速，因此请确保您的系统安装了支持CUDA的GPU。
• Git：用于克隆项目代码库。

安装步骤

1. 克隆项目代码库

   首先，使用Git克隆NBSS项目的代码库到本地：

   git clone https://github.com/Audio-WestlakeU/NBSS.git
   cd NBSS
   

2. 创建虚拟环境

   建议在虚拟环境中安装项目的依赖项，以避免与其他项目的依赖冲突。可以使用venv或conda创建虚拟环境：

   python -m venv nbss_env
   source nbss_env/bin/activate  # 对于Windows系统，使用 `nbss_env\Scripts\activate`
   

3. 安装依赖项

   进入项目目录后，安装所需的Python依赖项：

   pip install -r requirements.txt
   

4. 安装GPURIR库

   NBSS项目依赖于GPURIR库，用于生成房间脉冲响应。您可以通过以下命令安装GPURIR：

   pip install git+https://github.com/DavidDiazGuerra/gpuRIR.git
   

5. 配置数据集

   项目需要特定的数据集进行训练和测试。您需要下载并配置数据集，例如SMS-WSJ-Plus数据集。具体步骤可以参考项目文档中的说明。

6. 训练模型

   使用以下命令启动训练过程：

   python SharedTrainer.py fit \
       --config=configs/SpatialNet.yaml \
       --config=configs/datasets/sms_wsj_plus.yaml \
       --model.channels=[0,1,2,3,4,5] \
       --model.arch.dim_input=12 \
       --model.arch.dim_output=4 \
       --model.arch.num_freqs=129 \
       --trainer.precision=bf16-mixed \
       --model.compile=true \
       --data.batch_size=[2,4] \
       --trainer.devices=0 \
       --trainer.max_epochs=100
   

   请根据您的硬件配置和需求调整参数。

7. 测试模型

   训练完成后，可以使用以下命令测试模型：

   python SharedTrainer.py test \
       --config=logs/SpatialNet/version_x/config.yaml \
       --ckpt_path=logs/SpatialNet/version_x/checkpoints/epochY_neg_si_sdrZ.ckpt \
       --trainer.devices=0
   

   请将version_x和epochY_neg_si_sdrZ替换为实际的版本号和检查点文件名。

通过以上步骤，您应该能够成功安装和配置NBSS项目，并开始训练和测试模型。如果在安装过程中遇到任何问题，请参考项目的GitHub页面或相关文档获取更多帮助。

NBSS 项目地址: https://gitcode.com/gh_mirrors/nb/NBSS