Augmentation Python Packages

torch-audiomentations

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

1.9M 1K 101

torch-pitch-shift

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

1.9M 139 11

imgaug

Image augmentation for machine learning experiments.

957K 15K 2K

color-matcher

automatic color-grading and image data augmentation

478K 651 42

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.

199K 2K 221

python-stretch

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

187K 16 1

nlpaug

Data augmentation for NLP

182K 5K 474

ttach

Image Test Time Augmentation with PyTorch!

165K 1K 69

torchio

Medical imaging processing for AI applications.

102K 2K 270

albumentationsx

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

19K 498 31

albumentationsx-mcp

MCP server for AlbumentationsX image augmentation workflows

17K 5 0

augmentor

Image augmentation library in Python for machine learning.

15K 5K 873

tonic

Event datasets and transforms.

10K 287 53

bnnr

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

9K 25 9

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

7K 2K 126

torch-time-stretch

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

6K 40 3

deep-utils

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

5K 112 17

caer

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

5K 812 107

textaugment

TextAugment: Text Augmentation Library

3K 439 61

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.

2K 59 9