Data Augmentation Python Packages

webdataset

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

torch-audiomentations

Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.

audiomentations

A Python library for audio data augmentation. Useful for making audio ML models work well in the real world, not just in the lab.

albumentationsx

Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.

pilgram

A python library for instagram filters

torchio

Medical imaging processing for AI applications.

numpy-audio-limiter

Simple audio limiter. Made for use in audiomentations.

fast-mp3-augment

Fast Python library for MP3 audio data augmentation (encode + decode for intentional audio quality degradation). Made for use in audiomentations.

snorkel

A system for quickly generating training data with weak supervision

nvidia-dali-cuda120

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

augraphy

Augmentation pipeline for rendering synthetic paper printing, faxing, scanning and copy machine processes

wids

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

imagecorruptions

Python package to corrupt arbitrary images.

datarax

Differentiable data pipelines for JAX/Flax NNX: HuggingFace, TFDS and ArrayRecord sources, augmentation stages, scan-based epochs, exact mid-epoch resume.

polyagamma

An efficient and flexible sampler of the Pólya-Gamma distribution with a NumPy/SciPy compatible interface.

textattack

TextAttack 🐙 is a Python framework for adversarial attacks, data augmentation, and model training in NLP https://textattack.readthedocs.io/en/master/

elasticdeform

Differentiable elastic deformations for N-dimensional images (Python, SciPy, NumPy, TensorFlow, PyTorch).

nvidia-dali-nightly-cuda120

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

fastdup

fastdup is a powerful, free tool designed to rapidly generate valuable insights from image and video datasets. It helps enhance the quality of both images and labels, while significantly reducing data operation costs, all with unmatched scalability.

nvidia-dali-cuda110

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

bnnr

XAI-driven augmentation & diagnostics for PyTorch vision - find model failures, fix with saliency-guided augmentation (ICD/AICD), prove with auditable reports.

nvidia-dali-tf-plugin-nightly-cuda120

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

nvidia-dali-cuda130

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

nvidia-dali-nightly-cuda110

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.