Data Augmentation Techniques ML Frameworks

Libraries and methods for artificially expanding datasets through transformations, synthetic generation, and policy optimization across images, text, tabular, and time-series data. Does NOT include general data preprocessing, feature engineering, or dataset collection tools.

There are 56 data augmentation techniques frameworks tracked. 1 score above 70 (verified tier). The highest-rated is TorchIO-project/torchio at 77/100 with 2,370 stars. 1 of the top 10 are actively maintained.

Get all 56 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=data-augmentation-techniques&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	TorchIO-project/torchio Medical imaging processing for AI applications.	77	Verified	2,370	Python
2	aleju/imgaug Image augmentation for machine learning experiments.	63	Established	14,732	Python
3	makcedward/nlpaug Data augmentation for NLP	60	Established	4,650	Jupyter Notebook
4	mdbloice/Augmentor Image augmentation library in Python for machine learning.	59	Established	5,145	Python
5	BloodAxe/pytorch-toolbelt PyTorch extensions for fast R&D prototyping and Kaggle farming	56	Established	1,568	Python
6	qubvel/ttach Image Test Time Augmentation with PyTorch!	52	Established	1,029	Python
7	Paperspace/DataAugmentationForObjectDetection Data Augmentation For Object Detection	51	Established	1,151	Jupyter Notebook
8	imedslab/solt Streaming over lightweight data transformations	50	Established	266	Jupyter Notebook
9	albumentations-team/albucore A high-performance image processing library designed to optimize and extend...	50	Established	23	Python
10	stratospark/keras-multiprocess-image-data-generator Accelerating Deep Learning with Multiprocess Image Augmentation in Keras	49	Emerging	319	Jupyter Notebook
11	ildoonet/pytorch-randaugment Unofficial PyTorch Reimplementation of RandAugment.	48	Emerging	638	Python
12	Evizero/Augmentor.jl A fast image augmentation library in Julia for machine learning.	47	Emerging	144	Julia
13	albumentations-team/albumentations Fast and flexible image augmentation library. Paper about the library:...	47	Emerging	15,272	Python
14	kakaobrain/fast-autoaugment Official Implementation of 'Fast AutoAugment' in PyTorch.	47	Emerging	1,611	Python
15	arcelien/pba Efficient Learning of Augmentation Policy Schedules	47	Emerging	508	Jupyter Notebook
16	conradry/copy-paste-aug Copy-paste augmentation for segmentation and detection tasks	45	Emerging	571	Jupyter Notebook
17	mb010/AstroAugmentations A package with various custom augmentations implemented which are...	45	Emerging	9	Jupyter Notebook
18	firmai/deltapy DeltaPy - Tabular Data Augmentation (by @firmai)	44	Emerging	556	Jupyter Notebook
19	codebox/image_augmentor Data augmentation tool for images	43	Emerging	458	Python
20	rostepifanov/ecgmentations A Python library for ecg data augmentation. Useful for machine learning.	42	Emerging	22	Python
21	chenmingxiang110/AugNet The AugNet Python module contains functions for the fast computation of...	41	Emerging	95	Jupyter Notebook
22	AgaMiko/data-augmentation-review List of useful data augmentation resources. You will find here some not...	39	Emerging	1,636	—
23	bhky/targetran Python library for data augmentation in object detection or image...	39	Emerging	21	Python
24	rostepifanov/voxelmentations A Python library for 3d image data augmentation. Useful for machine learning.	39	Emerging	6	Python
25	MSD-IRIMAS/Augmenting-TSC-Elastic-Averaging Augmenting Time Series Datasets with Weighted Elastic Barycenter Averaging	37	Emerging	10	Python
26	DIAGNijmegen/pathology-he-auto-augment H&E tailored Randaugment: automatic data augmentation policy selection for...	36	Emerging	57	Python
27	cherise215/advchain [Medical Image Analysis] Adversarial Data Augmentation with Chained...	36	Emerging	57	Jupyter Notebook
28	zzdyyy/Patho-GAN Patho-GAN: interpretation + medical data augmentation. Code for paper work...	36	Emerging	81	Python
29	anhtuan85/Data-Augmentation-for-Object-Detection Data Augmentation For Object Detection using Pytorch and PIL	36	Emerging	80	Jupyter Notebook
30	VirtualEmbryo/ZAugNet ZAugNet	34	Emerging	6	Python
31	Clinacuity/CliniDeID CliniDeID automatically de-identifies clinical text notes according to the...	34	Emerging	10	Java
32	kennethleungty/Image-Augmentation-Libraries Sample implementation codes for a variety of popular image augmentation...	33	Emerging	8	Jupyter Notebook
33	KentaItakura/CNN-classification-using-random-erasing-and-cut-out This demo shows how to perform random erasing/cut out augmentation in CNN...	33	Emerging	5	MATLAB
34	affjljoo3581/Differentiable-RandAugment Optimize RandAugment with differentiable operations	32	Emerging	25	Python
35	RooTender/augmentator Ready-to-use tool for image augmentation	31	Emerging	7	Rust
36	juliendenize/torchaug Library to perform efficient vision data augmentations for CPU/GPU...	30	Emerging	26	Python
37	chogovadze/Deep-Illuminator Deep Illuminator is a data augmentation tool designed for image relighting....	28	Experimental	70	Python
38	argilla-io/adept-augmentations A Python library aimed at dissecting and augmenting NER training data.	27	Experimental	61	Python
39	avs-abhishek123/AugStatic AugStatic - A Light-Weight Image Augmentation Library	27	Experimental	7	Python
40	lartpang/tta.pytorch Test-Time Augmentation library for Pytorch	27	Experimental	83	Python
41	hkqiu/DataAugmentation4SmallData Data enhancement scheme for addressing the dilemma of small data on machine...	27	Experimental	10	Jupyter Notebook
42	AmirhosseinHonardoust/Noise-Injection-Techniques Noise Injection Techniques provides a comprehensive exploration of methods...	25	Experimental	22	—
43	rasbt/comparing-automatic-augmentation-blog Comparing four automatic image augmentation techniques in PyTorch:...	23	Experimental	31	Jupyter Notebook
44	Nazmul7989/python-image-augmentation A versatile Python-based image augmentation tool that helps generate diverse...	22	Experimental	—	Python
45	AIoT-MLSys-Lab/DeepAA [ICLR 2022] "Deep AutoAugment" by Yu Zheng, Zhi Zhang, Shen Yan, Mi Zhang	22	Experimental	65	Python
46	thejvdev/dl-aug-methods Repository for Analyzing the Impact of Data Augmentation Techniques	22	Experimental	—	Jupyter Notebook
47	adobe-research/beacon-aug Cross-library augmentation toolbox supporting 300 operators over 8 libraries...	21	Experimental	12	Jupyter Notebook
48	WeiChihChern/Copy-Paste-Semantic-Segmentation Simple Copy and Paste Implementation for Semantic Segmentation (Data Augmentation)	20	Experimental	7	Jupyter Notebook
49	Linear-Fox-Labs/AISynthDataAug AI-Driven Image Synthesis for Data Augmentation	20	Experimental	6	TeX
50	MSD-IRIMAS/Data-Augmentation-4-TSC Re-framing Time Series Augmentation Through the Lens of Generative Models	20	Experimental	3	Python
51	matin-ghorbani/Albumentations-Package Using Albumentations Package for some image augmentation in different vision tasks	19	Experimental	4	Jupyter Notebook
52	aflah02/Easy-Data-Augmentation-Implementation My Implementation for the paper EDA: Easy Data Augmentation Techniques for...	13	Experimental	12	Jupyter Notebook
53	Alyssonmach/artificial-augmentation-gans Generation of Artificial Images for Data Augmentation Using Generative...	11	Experimental	—	Jupyter Notebook
54	beenotung/augment-image Image augmentation library and cli for machine learning tasks	11	Experimental	—	TypeScript
55	ege621/data-augmentation-for-timeseries Data augmentation with randomly distributed noise for timeseries data	10	Experimental	2	—
56	LuisHBeck/tensorflow-data-augmentation learnig synthetic dataset generation (data augmentation) with tensorflow lib...	10	Experimental	2	Jupyter Notebook

Comparisons in this category

Augmentor and albumentations (59 vs 47) imgaug and Augmentor (63 vs 59) albucore and albumentations (50 vs 47) imgaug and image_augmentor (63 vs 43) Augmentor and image_augmentor (59 vs 43) Augmentor and Augmentor.jl (59 vs 47) torchio and ttach (77 vs 52)