🎛️ Wav2Aug: Task-Agnostic Waveform Augmentation Beyond ASR

A minimalistic PyTorch-based audio augmentation library for speech augmentation. The goal of this library is to provide a general purpose speech augmentation policy that can be used on any task and perform reasonably well without having to tune augmentation hyperparameters. Just install, and start augmenting. Applies two random augmentations per call. Just install and start augmenting!

📦 Installation

pip

pip install wav2aug

uv

uv add wav2aug

🚀 Quick Start

import torch
from wav2aug.gpu import Wav2Aug

# Initialize the augmenter once
augmenter = Wav2Aug(sample_rate=16000)

# in the forward pass
wavs = torch.randn(3, 50000)
lens = torch.ones((wavs.size(0)))

aug_wavs, aug_lens = augmenter(wavs, lens)

# or just

aug_wavs = augmenter(wavs)

That's it!

🧪 Augmentation Types

• 🔊 Amplitude Scaling/Clipping: Random gain and peak limiting
• 🌫️ Noise Addition: Environmental noise with SNR control
• 📶 Frequency Dropout: Spectral masking with random notch filters
• 🔄 Polarity Inversion: Random phase flip
• 🧩 Chunk Swapping: Temporal segment reordering
• ⏱️ Speed Perturbation: Time-scale modification
• 🕳️ Time Dropout: Random silence insertion
• 👥 Babble Noise: Multi-speaker background (auto-enabled with sufficient buffer)

Randomness: all stochastic ops use PyTorch RNGs. Set a single seed and be done, e.g. torch.manual_seed(0); torch.cuda.manual_seed_all(0)

🛠️ Development Installation

uv

git clone https://github.com/gfdb/wav2aug
cd wav2aug

# create venv and pin Python
uv venv
source .venv/bin/activate
uv python pin 3.10  # or 3.11/3.12

# runtime only
uv sync

# extras
uv sync --extra dev
uv sync --extra test

pip

git clone https://github.com/gfdb/wav2aug
cd wav2aug

# create venv
python -m venv .venv
source .venv/bin/activate

# runtime only
python -m pip install .

# editable + extras for development
python -m pip install -e '.[dev,test]'

✅ Tests

uv

uv run pytest -q tests/

pip

pytest -q tests/

🤝 Contributing

• Issues and PRs are welcome and encouraged!

• Bug reports: please open an issue with a minimal repro (env, dep versions, code snippet, expected vs. actual, traceback, etc.)

• Feature requests: please open an issue with use-case and proposed feature.

• PRs: Add tests for new stuff or when behavior changes. Also, don't forget to run formatters and tests before submitting!