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.

pilgram

A python library for instagram filters

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.

torchio

Medical imaging processing for AI applications.

snorkel

A system for quickly generating training data with weak supervision

textattack

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

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.

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.

albumentationsx

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

wids

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

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.

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-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.

deeptrack

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

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.

vkit-nightly

Boosting Document Intelligence

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.

scaper

A library for soundscape synthesis and augmentation

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.