Skip to content
Audio Denoising with Deep Network Priors
Branch: master
Clone or download
Latest commit 357023c Apr 18, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
wav_files new samples Apr 6, 2019
DNP.py added jupyter notebook and moved accumulator functions to a class Apr 18, 2019
README.md Update README.md Apr 18, 2019
accumulator.png added image Apr 18, 2019
demo.wav added jupyter notebook and moved accumulator functions to a class Apr 18, 2019
dnp.ipynb added jupyter notebook and moved accumulator functions to a class Apr 18, 2019
environment.yml added conda environment file Apr 6, 2019
index.html new samples Apr 6, 2019
pyramidnet.py Initialize repository Apr 6, 2019
unet.py
utils.py added jupyter notebook and moved accumulator functions to a class Apr 18, 2019

README.md

DNP

Audio Denoising with Deep Network Priors

This repository provides a PyTorch implementation of "Audio Denoising with Deep Network Priors" (paper)

The method trains on noisy audio signal and provides a the clean underlying signal. Comparison with traditional unsupervised methods can be found here

The method is completely unsupervised and only trains on the specific audio clip that is being denoised.

The algorithm is based on the observation that modeling noise in the signal is harder than a clean signal. During the fitting we observe flactuations in different stages of the train. By calculating the amount of difference between outputs in the time-frequency domain we create a robust spectral mask used for denoising the noisy output. Accumulator

Dependencies

A conda environment file is available in the repository.

  • Python 3.6 +
  • Pytorch 1.0
  • Torchvision
  • librosa
  • tqdm
  • scipy
  • soundfile

Usage

1. Cloning the repository & setting up conda environment

$ git clone https://github.com/mosheman5/DNP.git
$ cd DNP/

For creating and activating the conda environment:

$ conda env create -f environment.yml
$ conda activate DNP

2. Testing

To test on the demo speech file:

$ python DNP.py --run_name demo --noisy_file demo.wav --samples_dir samples --save_every 50 --num_iter 5000 --LR 0.001

To test on any other audio file insert the file path after the --noisy_file option.

A jupyter notebook with visualization is available: dnp.ipynb

Reference

If you found this code useful, please cite the following paper:

@article{michelashvili2019DNP,
  title={Audio Denoising with Deep Network Priors},
  author={Michael Michelashvili and Lior Wolf},
  journal={arXiv preprint arXiv:1904.07612},
  year={2019}
}

Acknowledgement

The implemantation of the network architecture is taken from Wave-U-Net

You can’t perform that action at this time.