- 配置环境
pip install -r requirements.txt 或者
conda activate server_dev
- 训练
python trainECAPAModel.py --save_path <save_path>
python testECAPAModel.py --eval
-
trainECAPAModel.py训练脚本 -
ECAPAModel.py调用模型 -
model.py模型- ECAPA_TDNN -> AAMsoftmax
-
testECAPAModel.py测试脚本修改参数:
train_list、train_path、eval_list、eval_path、musan_path噪声、rir_path混响、save_pathinitial_model需要测试的模型(required)ECAPAMModel.py调用模型脚本中test_network里modelname每次要更新
dataloader.py数据加载脚本- 实现数据增强,加入噪声、混响,注意匹配所有wav文件时的路径问题
get_label(label)提取标签fake-0 genuine-1