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.

color-matcher

automatic color-grading and image data augmentation

nlpaug

Data augmentation for NLP

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.

python-stretch

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

ttach

Image Test Time Augmentation with PyTorch!

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

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.

deep-utils

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

imagebaker

Let's bake an image.

augmenty

An augmentation library based on SpaCy for joint augmentation of text and labels.

textaugment

TextAugment: Text Augmentation Library

volumentations-3d

Library for 3D augmentations

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.

tools4yolo

Some image augmentation tools for yolov5 (and probably other models)