Была разобрана c сеть по статье Choi Hyeong-Seok и др. “Phase-Aware Speech Enhancement with Deep Complex U-Net”.
Установка зависимостей (conda):
conda env create -f dcunet.yml
conda activate dcunet
Загрузка данных:
python3 -m src.dataloader download
Разархивирование данных:
python3 -m src.dataloader clean-data
notebooks/report.ipynb
– отчет к задачеnotebooks/Experiment 1.ipynb
– Код для эксперементаsrc/
– код проектаsrc/train_utils.py
– содержит загрузчик для данных.src/model/dcenet.py
– содержит общую реализацию DCUNetsrc/model/ecnhmodel.py
– поная модель избавление данных от шумаsrc/model/complexnet.py
– реализация слоев над комплексными числамиsrc/loss.py
– реализация wsdr loss функцииsrc/dataloader.py
– загрузчик данных для обучения
config.toml
– конфигурация проектаdcunet.yml
– conda envaroment config