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.

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.

snorkel

A system for quickly generating training data with weak supervision

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.

textattack

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

albumentationsx

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

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.

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.

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.

deeptrack

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

elasticdeform

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

vkit-nightly

Boosting Document Intelligence

nvidia-dali-nightly-cuda110

NVIDIA DALI nightly for CUDA 11.0. Git SHA: 2be08c56f2be9ec8055256256039eb534ab7a080

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