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 54 data augmentation techniques frameworks tracked. 2 score above 70 (verified tier). The highest-rated is TorchIO-project/torchio at 80/100 with 2,370 stars. 1 of the top 10 are actively maintained.

Get all 54 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.	80	Verified	2,370	Python
2	aleju/imgaug Image augmentation for machine learning experiments.	73	Verified	14,732	Python
3	BloodAxe/pytorch-toolbelt PyTorch extensions for fast R&D prototyping and Kaggle farming	69	Established	1,568	Python
4	mdbloice/Augmentor Image augmentation library in Python for machine learning.	68	Established	5,145	Python
5	qubvel/ttach Image Test Time Augmentation with PyTorch!	62	Established	1,029	Python
6	makcedward/nlpaug Data augmentation for NLP	60	Established	4,650	Jupyter Notebook
7	imedslab/solt Streaming over lightweight data transformations	56	Established	266	Jupyter Notebook
8	Paperspace/DataAugmentationForObjectDetection Data Augmentation For Object Detection	51	Established	1,151	Jupyter Notebook
9	stratospark/keras-multiprocess-image-data-generator Accelerating Deep Learning with Multiprocess Image Augmentation in Keras	49	Emerging	319	Jupyter Notebook
10	firmai/deltapy DeltaPy - Tabular Data Augmentation (by @firmai)	48	Emerging	556	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	arcelien/pba Efficient Learning of Augmentation Policy Schedules	47	Emerging	508	Jupyter Notebook
14	kakaobrain/fast-autoaugment Official Implementation of 'Fast AutoAugment' in PyTorch.	47	Emerging	1,611	Python
15	albumentations-team/albumentations Fast and flexible image augmentation library. Paper about the library:...	47	Emerging	15,272	Python
16	conradry/copy-paste-aug Copy-paste augmentation for segmentation and detection tasks	45	Emerging	571	Jupyter Notebook
17	albumentations-team/albucore A high-performance image processing library designed to optimize and extend...	43	Emerging	23	Python
18	codebox/image_augmentor Data augmentation tool for images	43	Emerging	458	Python
19	chenmingxiang110/AugNet The AugNet Python module contains functions for the fast computation of...	41	Emerging	95	Jupyter Notebook
20	mb010/AstroAugmentations A package with various custom augmentations implemented which are...	41	Emerging	9	Jupyter Notebook
21	imtishalch/free-augmentcode 🧹 Clean AugmentCode data with this tool, enabling multiple account logins on...	41	Emerging	8	Python
22	rostepifanov/ecgmentations A Python library for ecg data augmentation. Useful for machine learning.	40	Emerging	22	Python
23	AgaMiko/data-augmentation-review List of useful data augmentation resources. You will find here some not...	39	Emerging	1,636	—
24	bhky/targetran Python library for data augmentation in object detection or image...	38	Emerging	21	Python
25	rostepifanov/voxelmentations A Python library for 3d image data augmentation. Useful for machine learning.	37	Emerging	6	Python
26	DIAGNijmegen/pathology-he-auto-augment H&E tailored Randaugment: automatic data augmentation policy selection for...	36	Emerging	57	Python
27	zzdyyy/Patho-GAN Patho-GAN: interpretation + medical data augmentation. Code for paper work...	36	Emerging	81	Python
28	anhtuan85/Data-Augmentation-for-Object-Detection Data Augmentation For Object Detection using Pytorch and PIL	36	Emerging	80	Jupyter Notebook
29	cherise215/advchain [Medical Image Analysis] Adversarial Data Augmentation with Chained...	36	Emerging	57	Jupyter Notebook
30	MSD-IRIMAS/Augmenting-TSC-Elastic-Averaging Augmenting Time Series Datasets with Weighted Elastic Barycenter Averaging	30	Emerging	10	Python
31	affjljoo3581/Differentiable-RandAugment Optimize RandAugment with differentiable operations	28	Experimental	25	Python
32	RooTender/augmentator Ready-to-use tool for image augmentation	28	Experimental	7	Rust
33	chogovadze/Deep-Illuminator Deep Illuminator is a data augmentation tool designed for image relighting....	28	Experimental	70	Python
34	lartpang/tta.pytorch Test-Time Augmentation library for Pytorch	27	Experimental	83	Python
35	VirtualEmbryo/ZAugNet ZAugNet	27	Experimental	6	Python
36	Clinacuity/CliniDeID CliniDeID automatically de-identifies clinical text notes according to the...	27	Experimental	10	Java
37	argilla-io/adept-augmentations A Python library aimed at dissecting and augmenting NER training data.	27	Experimental	61	Python
38	kennethleungty/Image-Augmentation-Libraries Sample implementation codes for a variety of popular image augmentation...	26	Experimental	8	Jupyter Notebook
39	KentaItakura/CNN-classification-using-random-erasing-and-cut-out This demo shows how to perform random erasing/cut out augmentation in CNN...	26	Experimental	5	MATLAB
40	willcagas/synthetic-medical-image-generation [GAISynMeD @ ACCV 2024] Official Implementation of the Paper "Medical...	24	Experimental	3	Jupyter Notebook
41	rasbt/comparing-automatic-augmentation-blog Comparing four automatic image augmentation techniques in PyTorch:...	23	Experimental	31	Jupyter Notebook
42	juliendenize/torchaug Library to perform efficient vision data augmentations for CPU/GPU...	23	Experimental	26	Python
43	thejvdev/dl-aug-methods Repository for Analyzing the Impact of Data Augmentation Techniques	22	Experimental	—	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	AmirhosseinHonardoust/Noise-Injection-Techniques Noise Injection Techniques provides a comprehensive exploration of methods...	21	Experimental	22	—
47	hkqiu/DataAugmentation4SmallData Data enhancement scheme for addressing the dilemma of small data on machine...	20	Experimental	10	Jupyter Notebook
48	avs-abhishek123/AugStatic AugStatic - A Light-Weight Image Augmentation Library	20	Experimental	7	Python
49	MSD-IRIMAS/Data-Augmentation-4-TSC Re-framing Time Series Augmentation Through the Lens of Generative Models	14	Experimental	3	Python
50	adobe-research/beacon-aug Cross-library augmentation toolbox supporting 300 operators over 8 libraries...	14	Experimental	12	Jupyter Notebook
51	Linear-Fox-Labs/AISynthDataAug AI-Driven Image Synthesis for Data Augmentation	13	Experimental	6	TeX
52	WeiChihChern/Copy-Paste-Semantic-Segmentation Simple Copy and Paste Implementation for Semantic Segmentation (Data Augmentation)	13	Experimental	7	Jupyter Notebook
53	matin-ghorbani/Albumentations-Package Using Albumentations Package for some image augmentation in different vision tasks	12	Experimental	4	Jupyter Notebook
54	yuvalrubinil/Robust-ADAS-Synthetic-Augmentation A deep learning framework for enhancing the robustness of Advanced Driver...	11	Experimental	—	Jupyter Notebook

Comparisons in this category

Augmentor and albumentations (68 vs 47) imgaug and Augmentor (73 vs 68) imgaug and fast-autoaugment (73 vs 47) imgaug and DataAugmentationForObjectDetection (73 vs 51) albumentations and albucore (47 vs 43) imgaug and Augmentor.jl (73 vs 47) Augmentor and Augmentor.jl (68 vs 47) torchio and ttach (80 vs 62)