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.

pilgram

A python library for instagram filters

torchio

Medical imaging processing for AI applications.

wids

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

textattack

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

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.

imagecorruptions

Python package to corrupt arbitrary images.

albumentationsx

Next-generation Albumentations: dual-licensed for open-source and commercial use

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.

augraphy

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

polyagamma

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

bnnr

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

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.

elasticdeform

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

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.

deeptrack

DeepTrack2 is a modular Python library for generating, manipulating, and analyzing image data pipelines for machine learning and experimental imaging.

webdatasetng

High performance storage and I/O for deep learning and data processing.

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.

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.

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.