Skip to main content

NSNet2 Deep Noise Suppression (DNS) package

Project description

Noise Suppression Net 2 (NSNet2) baseline inference script

  • As a baseline for ICASSP 2021 Deep Noise Suppression challenge, we will use the recently developed SE method based on Recurrent Neural Network (RNN). For ease of reference, we will call this method as Noise Suppression Net 2 (NSNet 2). More details about this method can be found in here.

Installation

pip install nsnet2-denoiser

Usage:

From the NSNet2-baseline directory, run run_nsnet2.py with the following required arguments:

  • -i "Specify the path to noisy speech files that you want to enhance"
  • -o "Specify the path to a directory where you want to store the enhanced clips"
  • -fs "Sampling rate of the input audio. (48000/16000)"

python -m nsnet2_denoiser.denoise -i audio/

Use default values for the rest. Run to enhance the clips.

Python

from nsnet2_denoiser import NSnet2Enhancer
enhancer = NSnet2Enhancer(fs=48000)

# numpy
import soundfile as sf
sigIn, fs = sf.read("audio.wav")
outSig = enhancer(sigIn, fs)

# pcm_16le
from pydub import AudioSegment
audioIn = AudioSegment.from_wav("audio.wav")
audioOut = enhancer.pcm_16le(audioIn.raw_data)

Attribution:

Pretrained model NSNet2 by Microsoft is licensed under CC BY 4.0

Citation:

The baseline NSNet noise suppression:

@misc{braun2020data,
    title={Data augmentation and loss normalization for deep noise suppression},
    author={Sebastian Braun and Ivan Tashev},
    year={2020},
    eprint={2008.06412},
    archivePrefix={arXiv},
    primaryClass={eess.AS}
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nsnet2-denoiser-0.2.2.tar.gz (33.0 MB view hashes)

Uploaded Source

Built Distribution

nsnet2_denoiser-0.2.2-py3-none-any.whl (33.0 MB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page