AECNN_for_Speech_Enhancement

KO | EN

gitlite — search

#python #ai #typescript #llm #machine-learning #rust #nodejs #macos #windows #kubernetes #security #ios #mcp #open-source

AECNN_for_Speech_Enhancement

(★ 24)

Unofficial Pytorch Lightning Implementation of "A New Framework for CNN-Based Speech Enhancement in the Time Domain"

AECNN_for_Speech_Enhancement 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# A New Framework for CNN-Based Speech Enhancement in the Time Domain

This repository contains the unofficial pytorch lightning implementation of the model described in the paper [A New Framework for CNN-Based Speech Enhancement in the Time Domain](https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=8701652) by Ashutosh Pandey and Deliang Wang.

## Requirements
 
To run this code, you will need:

- pytorch_lightning==2.0.0
- PyYAML==6.0
- torch==2.0.0
- torchaudio==2.0.0

To automatically install these libraries, run the following command:

```pip install -r requirements.txt```

## Usage

To run the code on your own machine, follow these steps:

1. Open the 'config.yaml' file and modify the file paths (and hyperparameters as needed).
2. Run the 'main.py' file to start training the model.

The trained model will be saved as ckpt file in 'logger' directory. You can then use the trained model to perform real-time speech frequency bandwidth extension on your own audio wav file by running the 'inference.py' file as

```python inference.py --mode "wav" --path_ckpt <path of checkpoint file> --path_in <path of wav file>```

This repository also support directory-level inference, where the inference is performed on a directory consisting of wav files. You can use the following example to perform directory-level inference,

```python inference.py --mode "dir" --path_ckpt <path of checkpoint file> --path_in <path of directory that contains input wave files> --path_out <path of directory that output files will be saved>```

## Note
- 2023.5.1 This code now supports Distributed Data Parallel (DDP) training!
- 2023.4.28 The code has been modified to be compatible with PyTorch Lightning 2.0 environment! It includes support for inference as well.
- Feel free to provide issues!

## Citation

```bibtex
@ARTICLE{8701652,
  author={Pandey, Ashutosh and Wang, DeLiang},
  journal={IEEE/ACM Transactions on Audio, Speech, and Language Processing}, 
  title={A New Framework for CNN-Based Speech Enhancement in the Time Domain}, 
  year={2019},
  volume={27},
  number={7},
  pages={1179-1188},
  doi={10.1109/TASLP.2019.2913512}}
```

AECNN_for_Speech_Enhancement

AECNN_for_Speech_Enhancement 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

AECNN_for_Speech_Enhancement

AECNN_for_Speech_Enhancement 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README