---
id: batchgenerators
version: "0.25.3"
license: Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and conditions for… (full text in the JSON record)
license_treatment: permissive
maintenance: active
---

# batchgenerators — Data augmentation toolkit

License: permissive · Maintenance: active · Downloads: 149.4K/mo

## What it is and what it does

batchgenerators is a data augmentation framework developed by the Medical Image Computing division at DKFZ for preparing training data in deep learning pipelines. It specializes in augmentations that work uniformly on both 2D and 3D image data—a capability the documentation notes was missing in most competing frameworks. The package provides spatial transforms (mirroring, elastic deformation, rotation, scaling, resampling), color adjustments (brightness, contrast, gamma), noise injection (Gaussian, Rician), and cropping operations, plus specialized augmentations for medical imaging such as anatomy-informed deformations and multi-modal misalignment simulation.

The typical workflow involves subclassing a DataLoaderBase, implementing a generate_train_batch method to stack transforms using the Compose utility, and then feeding the result into MultiThreadedAugmenter for parallel augmentation across multiple processes. Data flows through the pipeline as a simple Python dictionary with required 'data' and optional 'seg' keys; spatial transforms automatically apply to segmentation masks when present. The package depends on numpy, scipy, scikit-image, scikit-learn, pillow, and pandas for its core operations.

Use it for:

- Augment 3D medical imaging datasets for segmentation tasks while ensuring spatial transforms apply to both images and segmentation masks.
- Build training pipelines that apply per-sample augmentation decisions with configurable probabilities rather than batch-wide decisions.
- Preprocess multi-modal medical images with controlled misalignments between channels to improve model robustness.
- Combine spatial and color augmentations in a composable pipeline and execute them in parallel across multiple worker processes.
- Simulate soft-tissue deformations in medical imaging workflows using anatomy-informed augmentation for realistic training data.

## Worth the install?

AI-flagged interpretation of the facts above — verify before relying.

batchgenerators provides data augmentation transforms for 2D and 3D images, including spatial transformations, color adjustments, noise injection, and cropping, designed for medical image analysis and deep learning pipelines.

Yes. batchgenerators is actively maintained, has no known vulnerabilities, and installs with low friction. It is well-suited for medical image analysis and deep learning projects that need 2D/3D augmentation with per-sample control and multithreaded execution. The Apache 2.0 license is permissive. Use it if you need specialized augmentation for medical or scientific imaging.

## Install

pip install batchgenerators
uv add batchgenerators
poetry add batchgenerators

## Installing batchgenerators

Before you install: Low install friction with a pure Python wheel and stable maintenance. Last release 59 days ago, active repository with 1172 stars, and no known vulnerabilities.

License in practice: Apache License 2.0 is permissive; you may use, modify, and distribute batchgenerators freely in commercial and private projects, provided you include the license and attribute the original authors.

Quickstart:

pip install batchgenerators

from batchgenerators.transforms.color_transforms import ContrastAugmentationTransform
from batchgenerators.dataloading.multi_threaded_augmenter import MultiThreadedAugmenter

# Stack transforms and pass to MultiThreadedAugmenter for multithreaded augmentation

Windows requires guarding code with `if __name__ == '__main__'` and calling `multiprocessing.freeze_support()` due to multiprocessing differences; Linux has no such requirement.

Verify before relying:

- Whether anatomy-informed and misalignment augmentations (added in 0.23.1 and 0.23.2) are production-ready or still experimental.
- Specific Python version compatibility (requires_python is unspecified in metadata).
- Performance characteristics and memory overhead when processing large 3D datasets.

## Package facts

- License: Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and conditions for… (full text in the JSON record) (permissive)
- Python support: unspecified
- Install friction: low
- Maintenance: active
- Downloads: 149.4K/month (top 15,000 on PyPI)
- Known vulnerabilities: none known

## Tags

data augmentation medical images, 2D 3D image transforms, elastic deformation rotation scaling, batch augmentation deep learning, image segmentation augmentation, spatial color noise augmentation, medical-imaging, data-augmentation, 3d-support

[View on SkillFed](https://skillfed.io/packages/batchgenerators) · [View on PyPI](https://pypi.org/project/batchgenerators/)