Augmentation Python Packages

torch-audiomentations

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

torch-pitch-shift

Pitch-shift audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

imgaug

Image augmentation for machine learning experiments.

nlpaug

Data augmentation for NLP

color-matcher

automatic color-grading and image data augmentation

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.

ttach

Image Test Time Augmentation with PyTorch!

python-stretch

Simple python library for pitch shifting and time stretching. Wrapper of Signalsmith Stretch C++ Library

torchio

Medical imaging processing for AI applications.

albumentationsx

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

tonic

Event datasets and transforms.

augmentor

Image augmentation library in Python for machine learning.

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

deep-utils

An open-source toolkit which is full of handy functions, including the most used models and utilities for deep-learning practitioners!

caer

High-performance Vision library in Python. Scale your research, not boilerplate.

torch-time-stretch

Time-stretch audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

augmenty

Augmenty is an augmentation library based on spaCy for augmenting texts.

volumentations-3d

Library for 3D augmentations

imagebaker

Let's bake an image.

textaugment

TextAugment: Text Augmentation Library

amt-augmentor

Python augmentation toolkit for Automatic Music Transcription datasets

auglib

Data augmentation for audio

audioaugmentor

This library is designed to augment audio data for machine learning purposes. It combines several tools and libraries for audio data augmentation and provides a unified interface that can be used to apply a large set of audio augmentations in one place.

duplipy

DupliPy is a quick and easy-to-use package that can handle text formatting and data augmentation tasks for NLP in Python, with added support for image augmentation.