ASEGAN: Speech Enhancement Generative Adversarial Network Based on Asymmetric AutoEncoder
中文版简介
Readme with English Version
介绍
基于SEGAN模型的改进版本,使用自主设计的非对称自编码结构替换原有的全卷积结构,使得模型在保持原有性能的条件下更加轻量化。
(本模型未在任何期刊发布)
软件架构
安装教程
- 安装必要库
Anaconda
cuda
cudnn - 创建python环境
conda create -n ASEGAN python=3.8
conda activate ASEGAN
- 安装必要环境
pip install -r requirements.txt
或者conda install --yes --file requirements.txt
使用说明
使用前检查config/config.yaml中参数配置是否正确,数据集文件结构参考data文件夹中结构
- 数据预处理
python data_preprocess.py
- 模型训练
python train.py
- 模型测试
python test.py
预训练模型下载
百度网盘,提取码6793
GoogleDrive
预训练模型命名方式为‘数据集-训练周期-数据量.pkl’