Augmentation Python Packages

torch-pitch-shift

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

torch-audiomentations

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

imgaug

Image augmentation for machine learning experiments.

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

color-matcher

automatic color-grading and image data augmentation

nlpaug

Data augmentation for NLP

albumentationsx

Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.

ttach

Image Test Time Augmentation with PyTorch!

torchio

Medical imaging processing for AI applications.

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.

volumentations-3d

Library for 3D augmentations

torch-time-stretch

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

albumentationsx-mcp

MCP server for AlbumentationsX image augmentation workflows

bnnr

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

vietquill

VietQuill is a unified framework for Vietnamese paraphrase generation, evaluation, and quality control, supporting both research and production applications. We are committed to advancing Vietnamese paraphrase generation by making state-of-the-art methods accessible, customizable, and easy to integrate into real-world workflows.

imagebaker

Let's bake an image.

variopinta

CPU image-augmentation pipelines compiled from Python and executed by optimized Rust kernels

textaugment

TextAugment: Text Augmentation Library

auglib

Data augmentation for audio