Metadata-Version: 2.1
Name: denoisers
Version: 0.1.6
Summary: A package for training audio denoisers
Author-email: Will Rice <wrice20@gmail.com>
Project-URL: Homepage, https://github.com/will-rice/denoisers
Project-URL: Bug Tracker, https://github.com/will-rice/denoisers/issues
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.9
Description-Content-Type: text/markdown
License-File: LICENSE

# Denoisers

Denoisers is a denoising library for audio with a focus on simplicity and ease of use. There are two major types of architectures available. WaveUNet for waveform denoising and UNet for spectrogram denoising.

## Demo

Gradio demo [here](https://wrice-denoisers.hf.space)

## Usage/Examples

```sh
pip install denoisers
```

```python
import torch
import torchaudio
from denoisers import WaveUNetModel
from tqdm import tqdm

model = WaveUNetModel.from_pretrained("wrice/waveunet-vctk-24khz")

audio, sr = torchaudio.load("noisy_audio.wav")

if sr != model.config.sample_rate:
    audio = torchaudio.functional.resample(audio, sr, model.config.sample_rate)

chunk_size = model.config.max_length

padding = abs(audio.size(-1) % chunk_size - chunk_size)
padded = torch.nn.functional.pad(audio, (0, padding))

clean = []
for i in tqdm(range(0, padded.shape[-1], chunk_size)):
    audio_chunk = padded[:, i:i+chunk_size]
    with torch.no_grad():
        clean_chunk = model(audio_chunk[None]).logits
    clean.append(clean_chunk.squeeze(0))

denoised = torch.concat(clean, 1)[:, :audio.shape[-1]]
```
