此项目针对 CN-Celeb 数据集所开发，框架采用 ECAPA-TDNN + AAM-Softmax。

• CN-Celeb 1 [点此下载]
• CN-Celeb 2 [点此下载]

CN-Celeb 原始数据是 flac 格式，考虑到转换格式又要占磁盘空间，就直接读取 flac 格式进行训练了。

• 用build_datalist.py生成train.csv
• 数据集中提供的trial.lst以wav作为后缀，疑似存在问题，用dataset.py中的create_cnceleb_trails函数生成新的trial.lst

conda create -n cnceleb python=3.12.3
conda activate cnceleb
pip install -r requirements.txt -i https://pypi.mirrors.ustc.edu.cn/simple/

pip 安装不了可以试试下面其他的源

• 清华：https://pypi.tuna.tsinghua.edu.cn/simple/
• 阿里云：http://mirrors.aliyun.com/pypi/simple/
• **科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
• 华中理工大学：http://pypi.hustunique.com/
• 山东理工大学：http://pypi.sdutlinux.org/
• 豆瓣：http://pypi.douban.com/simple/

1. 在 trainECAPAModel.py 配置好对应路径
2. 激活 conda 环境 conda activate cnceleb
3. 运行 python trainECAPAModel.py

1. 在主程序中设置定义 initial_model 路径
2. 运行python trainECAPAModel.py --eval

1. 从release中下载预训练权重
2. 使用命令gradio demo_with_gradio.py启动脚本，在浏览器中打开对应链接

本项目基于 PunkMale/ECAPA-TDNN-CNCeleb和TaoRuijie/ECAPA-TDNN 修改，并参考了 Lantian Li/Sunine。