Была разобрана 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 – содержит общую реализацию DCUNet
  • src/model/ecnhmodel.py – поная модель избавление данных от шума
  • src/model/complexnet.py – реализация слоев над комплексными числами
  • src/loss.py – реализация wsdr loss функции
  • src/dataloader.py – загрузчик данных для обучения
• config.toml – конфигурация проекта
• dcunet.yml – conda envaroment config